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 #include "CommandObjectCommands.h"
11ebc09c36SJim Ingham 
12ebc09c36SJim Ingham // C Includes
13ebc09c36SJim Ingham // C++ Includes
14ebc09c36SJim Ingham // Other libraries and framework includes
150e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h"
160e5e5a79SGreg Clayton 
17ebc09c36SJim Ingham // Project includes
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.",
455a988416SJim Ingham                              NULL),
465a988416SJim Ingham         m_options (interpreter)
475a988416SJim Ingham     {
485a988416SJim Ingham     }
495a988416SJim Ingham 
505a988416SJim Ingham     ~CommandObjectCommandsHistory () {}
515a988416SJim Ingham 
525a988416SJim Ingham     virtual Options *
535a988416SJim Ingham     GetOptions ()
545a988416SJim Ingham     {
555a988416SJim Ingham         return &m_options;
565a988416SJim Ingham     }
575a988416SJim Ingham 
585a988416SJim Ingham protected:
59a5a97ebeSJim Ingham 
60a5a97ebeSJim Ingham     class CommandOptions : public Options
61a5a97ebeSJim Ingham     {
62a5a97ebeSJim Ingham     public:
63a5a97ebeSJim Ingham 
64a5a97ebeSJim Ingham         CommandOptions (CommandInterpreter &interpreter) :
657594f14fSEnrico Granata             Options (interpreter),
667594f14fSEnrico Granata             m_start_idx(0),
677594f14fSEnrico Granata             m_stop_idx(0),
687594f14fSEnrico Granata             m_count(0),
6963123b64SEnrico Granata             m_clear(false)
70a5a97ebeSJim Ingham         {
71a5a97ebeSJim Ingham         }
72a5a97ebeSJim Ingham 
73a5a97ebeSJim Ingham         virtual
74a5a97ebeSJim Ingham         ~CommandOptions (){}
75a5a97ebeSJim Ingham 
76a5a97ebeSJim Ingham         virtual Error
77a5a97ebeSJim Ingham         SetOptionValue (uint32_t option_idx, const char *option_arg)
78a5a97ebeSJim Ingham         {
79a5a97ebeSJim Ingham             Error error;
803bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
81a5a97ebeSJim Ingham 
82a5a97ebeSJim Ingham             switch (short_option)
83a5a97ebeSJim Ingham             {
84a5a97ebeSJim Ingham                 case 'c':
85c95f7e2aSPavel Labath                     error = m_count.SetValueFromString(option_arg,eVarSetOperationAssign);
86a5a97ebeSJim Ingham                     break;
87a5a97ebeSJim Ingham                 case 's':
887594f14fSEnrico Granata                     if (option_arg && strcmp("end", option_arg) == 0)
897594f14fSEnrico Granata                     {
907594f14fSEnrico Granata                         m_start_idx.SetCurrentValue(UINT64_MAX);
917594f14fSEnrico Granata                         m_start_idx.SetOptionWasSet();
927594f14fSEnrico Granata                     }
937594f14fSEnrico Granata                     else
94c95f7e2aSPavel Labath                         error = m_start_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
957594f14fSEnrico Granata                     break;
967594f14fSEnrico Granata                 case 'e':
97c95f7e2aSPavel Labath                     error = m_stop_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
987594f14fSEnrico Granata                     break;
9963123b64SEnrico Granata                 case 'C':
10063123b64SEnrico Granata                     m_clear.SetCurrentValue(true);
10163123b64SEnrico Granata                     m_clear.SetOptionWasSet();
102a5a97ebeSJim Ingham                     break;
103a5a97ebeSJim Ingham                 default:
10486edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
105a5a97ebeSJim Ingham                     break;
106a5a97ebeSJim Ingham             }
107a5a97ebeSJim Ingham 
108a5a97ebeSJim Ingham             return error;
109a5a97ebeSJim Ingham         }
110a5a97ebeSJim Ingham 
111a5a97ebeSJim Ingham         void
112a5a97ebeSJim Ingham         OptionParsingStarting ()
113a5a97ebeSJim Ingham         {
1147594f14fSEnrico Granata             m_start_idx.Clear();
1157594f14fSEnrico Granata             m_stop_idx.Clear();
1167594f14fSEnrico Granata             m_count.Clear();
11763123b64SEnrico Granata             m_clear.Clear();
118a5a97ebeSJim Ingham         }
119a5a97ebeSJim Ingham 
120a5a97ebeSJim Ingham         const OptionDefinition*
121a5a97ebeSJim Ingham         GetDefinitions ()
122a5a97ebeSJim Ingham         {
123a5a97ebeSJim Ingham             return g_option_table;
124a5a97ebeSJim Ingham         }
125a5a97ebeSJim Ingham 
126a5a97ebeSJim Ingham         // Options table: Required for subclasses of Options.
127a5a97ebeSJim Ingham 
128a5a97ebeSJim Ingham         static OptionDefinition g_option_table[];
129a5a97ebeSJim Ingham 
130a5a97ebeSJim Ingham         // Instance variables to hold the values for command options.
131a5a97ebeSJim Ingham 
1327594f14fSEnrico Granata         OptionValueUInt64 m_start_idx;
1337594f14fSEnrico Granata         OptionValueUInt64 m_stop_idx;
1347594f14fSEnrico Granata         OptionValueUInt64 m_count;
13563123b64SEnrico Granata         OptionValueBoolean m_clear;
136a5a97ebeSJim Ingham     };
137a5a97ebeSJim Ingham 
138a5a97ebeSJim Ingham     bool
1395a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
140a5a97ebeSJim Ingham     {
14163123b64SEnrico Granata         if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet())
1427594f14fSEnrico Granata         {
1437594f14fSEnrico Granata             m_interpreter.GetCommandHistory().Clear();
1447594f14fSEnrico Granata             result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
1457594f14fSEnrico Granata         }
1467594f14fSEnrico Granata         else
1477594f14fSEnrico Granata         {
1487594f14fSEnrico Granata             if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet())
1497594f14fSEnrico Granata             {
1507594f14fSEnrico Granata                 result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation");
1517594f14fSEnrico Granata                 result.SetStatus(lldb::eReturnStatusFailed);
1527594f14fSEnrico Granata             }
1537594f14fSEnrico Granata             else
1547594f14fSEnrico Granata             {
15584400ec7SVirgile Bello                 std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue());
15684400ec7SVirgile Bello                 std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue());
15784400ec7SVirgile Bello                 std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue());
158a5a97ebeSJim Ingham 
1597594f14fSEnrico Granata                 const CommandHistory& history(m_interpreter.GetCommandHistory());
1607594f14fSEnrico Granata 
1617594f14fSEnrico Granata                 if (start_idx.first && start_idx.second == UINT64_MAX)
1627594f14fSEnrico Granata                 {
1637594f14fSEnrico Granata                     if (count.first)
1647594f14fSEnrico Granata                     {
1657594f14fSEnrico Granata                         start_idx.second = history.GetSize() - count.second;
1667594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1677594f14fSEnrico Granata                     }
1687594f14fSEnrico Granata                     else if (stop_idx.first)
1697594f14fSEnrico Granata                     {
1707594f14fSEnrico Granata                         start_idx.second = stop_idx.second;
1717594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1727594f14fSEnrico Granata                     }
1737594f14fSEnrico Granata                     else
1747594f14fSEnrico Granata                     {
1757594f14fSEnrico Granata                         start_idx.second = 0;
1767594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1777594f14fSEnrico Granata                     }
1787594f14fSEnrico Granata                 }
1797594f14fSEnrico Granata                 else
1807594f14fSEnrico Granata                 {
1817594f14fSEnrico Granata                     if (!start_idx.first && !stop_idx.first && !count.first)
1827594f14fSEnrico Granata                     {
1837594f14fSEnrico Granata                         start_idx.second = 0;
1847594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1857594f14fSEnrico Granata                     }
1867594f14fSEnrico Granata                     else if (start_idx.first)
1877594f14fSEnrico Granata                     {
1887594f14fSEnrico Granata                         if (count.first)
1897594f14fSEnrico Granata                         {
1907594f14fSEnrico Granata                             stop_idx.second = start_idx.second + count.second - 1;
1917594f14fSEnrico Granata                         }
1927594f14fSEnrico Granata                         else if (!stop_idx.first)
1937594f14fSEnrico Granata                         {
1947594f14fSEnrico Granata                             stop_idx.second = history.GetSize() - 1;
1957594f14fSEnrico Granata                         }
1967594f14fSEnrico Granata                     }
1977594f14fSEnrico Granata                     else if (stop_idx.first)
1987594f14fSEnrico Granata                     {
1997594f14fSEnrico Granata                         if (count.first)
2007594f14fSEnrico Granata                         {
2017594f14fSEnrico Granata                             if (stop_idx.second >= count.second)
2027594f14fSEnrico Granata                                 start_idx.second = stop_idx.second - count.second + 1;
2037594f14fSEnrico Granata                             else
2047594f14fSEnrico Granata                                 start_idx.second = 0;
2057594f14fSEnrico Granata                         }
2067594f14fSEnrico Granata                     }
2077594f14fSEnrico Granata                     else /* if (count.first) */
2087594f14fSEnrico Granata                     {
2097594f14fSEnrico Granata                         start_idx.second = 0;
2107594f14fSEnrico Granata                         stop_idx.second = count.second - 1;
2117594f14fSEnrico Granata                     }
2127594f14fSEnrico Granata                 }
2137594f14fSEnrico Granata                 history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second);
2147594f14fSEnrico Granata             }
2157594f14fSEnrico Granata         }
216a5a97ebeSJim Ingham         return result.Succeeded();
217a5a97ebeSJim Ingham 
218a5a97ebeSJim Ingham     }
2195a988416SJim Ingham 
2205a988416SJim Ingham     CommandOptions m_options;
221a5a97ebeSJim Ingham };
222a5a97ebeSJim Ingham 
223a5a97ebeSJim Ingham OptionDefinition
224a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] =
225a5a97ebeSJim Ingham {
226d37221dcSZachary Turner { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeUnsignedInteger,        "How many history commands to print."},
227d37221dcSZachary Turner { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeUnsignedInteger,  "Index at which to start printing history commands (or end to mean tail mode)."},
228d37221dcSZachary Turner { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeUnsignedInteger,    "Index at which to stop printing history commands."},
229d37221dcSZachary Turner { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, NULL, NULL, 0, eArgTypeBoolean,    "Clears the current command history."},
230d37221dcSZachary Turner { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
231a5a97ebeSJim Ingham };
232a5a97ebeSJim Ingham 
233a5a97ebeSJim Ingham 
234a5a97ebeSJim Ingham //-------------------------------------------------------------------------
235a5a97ebeSJim Ingham // CommandObjectCommandsSource
236a5a97ebeSJim Ingham //-------------------------------------------------------------------------
237a5a97ebeSJim Ingham 
2385a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed
239ebc09c36SJim Ingham {
2405a988416SJim Ingham public:
2415a988416SJim Ingham     CommandObjectCommandsSource(CommandInterpreter &interpreter) :
2425a988416SJim Ingham         CommandObjectParsed (interpreter,
2435a988416SJim Ingham                              "command source",
2445a988416SJim Ingham                              "Read in debugger commands from the file <filename> and execute them.",
2455a988416SJim Ingham                              NULL),
2465a988416SJim Ingham         m_options (interpreter)
2475a988416SJim Ingham     {
2485a988416SJim Ingham         CommandArgumentEntry arg;
2495a988416SJim Ingham         CommandArgumentData file_arg;
2505a988416SJim Ingham 
2515a988416SJim Ingham         // Define the first (and only) variant of this arg.
2525a988416SJim Ingham         file_arg.arg_type = eArgTypeFilename;
2535a988416SJim Ingham         file_arg.arg_repetition = eArgRepeatPlain;
2545a988416SJim Ingham 
2555a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
2565a988416SJim Ingham         arg.push_back (file_arg);
2575a988416SJim Ingham 
2585a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
2595a988416SJim Ingham         m_arguments.push_back (arg);
2605a988416SJim Ingham     }
2615a988416SJim Ingham 
2625a988416SJim Ingham     ~CommandObjectCommandsSource () {}
2635a988416SJim Ingham 
2645a988416SJim Ingham     virtual const char*
2655a988416SJim Ingham     GetRepeatCommand (Args &current_command_args, uint32_t index)
2665a988416SJim Ingham     {
2675a988416SJim Ingham         return "";
2685a988416SJim Ingham     }
2695a988416SJim Ingham 
270c7bece56SGreg Clayton     virtual int
2715a988416SJim Ingham     HandleArgumentCompletion (Args &input,
2725a988416SJim Ingham                               int &cursor_index,
2735a988416SJim Ingham                               int &cursor_char_position,
2745a988416SJim Ingham                               OptionElementVector &opt_element_vector,
2755a988416SJim Ingham                               int match_start_point,
2765a988416SJim Ingham                               int max_return_elements,
2775a988416SJim Ingham                               bool &word_complete,
2785a988416SJim Ingham                               StringList &matches)
2795a988416SJim Ingham     {
2805a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
2815a988416SJim Ingham         completion_str.erase (cursor_char_position);
2825a988416SJim Ingham 
2835a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
2845a988416SJim Ingham                                                              CommandCompletions::eDiskFileCompletion,
2855a988416SJim Ingham                                                              completion_str.c_str(),
2865a988416SJim Ingham                                                              match_start_point,
2875a988416SJim Ingham                                                              max_return_elements,
2885a988416SJim Ingham                                                              NULL,
2895a988416SJim Ingham                                                              word_complete,
2905a988416SJim Ingham                                                              matches);
2915a988416SJim Ingham         return matches.GetSize();
2925a988416SJim Ingham     }
2935a988416SJim Ingham 
2945a988416SJim Ingham     virtual Options *
2955a988416SJim Ingham     GetOptions ()
2965a988416SJim Ingham     {
2975a988416SJim Ingham         return &m_options;
2985a988416SJim Ingham     }
2995a988416SJim Ingham 
3005a988416SJim Ingham protected:
301e16c50a1SJim Ingham 
302e16c50a1SJim Ingham     class CommandOptions : public Options
303e16c50a1SJim Ingham     {
304e16c50a1SJim Ingham     public:
305e16c50a1SJim Ingham 
306eb0103f2SGreg Clayton         CommandOptions (CommandInterpreter &interpreter) :
307012d4fcaSEnrico Granata             Options (interpreter),
308340b0309SGreg Clayton             m_stop_on_error (true),
309340b0309SGreg Clayton             m_silent_run (false),
310340b0309SGreg Clayton             m_stop_on_continue (true)
311eb0103f2SGreg Clayton         {
312eb0103f2SGreg Clayton         }
313e16c50a1SJim Ingham 
314e16c50a1SJim Ingham         virtual
315e16c50a1SJim Ingham         ~CommandOptions (){}
316e16c50a1SJim Ingham 
317e16c50a1SJim Ingham         virtual Error
318f6b8b581SGreg Clayton         SetOptionValue (uint32_t option_idx, const char *option_arg)
319e16c50a1SJim Ingham         {
320e16c50a1SJim Ingham             Error error;
3213bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
322e16c50a1SJim Ingham 
323e16c50a1SJim Ingham             switch (short_option)
324e16c50a1SJim Ingham             {
325e16c50a1SJim Ingham                 case 'e':
326c95f7e2aSPavel Labath                     error = m_stop_on_error.SetValueFromString(option_arg);
327e16c50a1SJim Ingham                     break;
328340b0309SGreg Clayton 
329e16c50a1SJim Ingham                 case 'c':
330c95f7e2aSPavel Labath                     error = m_stop_on_continue.SetValueFromString(option_arg);
331e16c50a1SJim Ingham                     break;
332340b0309SGreg Clayton 
33360986174SMichael Sartain                 case 's':
334c95f7e2aSPavel Labath                     error = m_silent_run.SetValueFromString(option_arg);
33560986174SMichael Sartain                     break;
336340b0309SGreg Clayton 
337e16c50a1SJim Ingham                 default:
33886edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
339e16c50a1SJim Ingham                     break;
340e16c50a1SJim Ingham             }
341e16c50a1SJim Ingham 
342e16c50a1SJim Ingham             return error;
343e16c50a1SJim Ingham         }
344e16c50a1SJim Ingham 
345e16c50a1SJim Ingham         void
346f6b8b581SGreg Clayton         OptionParsingStarting ()
347e16c50a1SJim Ingham         {
348012d4fcaSEnrico Granata             m_stop_on_error.Clear();
349340b0309SGreg Clayton             m_silent_run.Clear();
350340b0309SGreg Clayton             m_stop_on_continue.Clear();
351e16c50a1SJim Ingham         }
352e16c50a1SJim Ingham 
353e0d378b3SGreg Clayton         const OptionDefinition*
354e16c50a1SJim Ingham         GetDefinitions ()
355e16c50a1SJim Ingham         {
356e16c50a1SJim Ingham             return g_option_table;
357e16c50a1SJim Ingham         }
358e16c50a1SJim Ingham 
359e16c50a1SJim Ingham         // Options table: Required for subclasses of Options.
360e16c50a1SJim Ingham 
361e0d378b3SGreg Clayton         static OptionDefinition g_option_table[];
362e16c50a1SJim Ingham 
363e16c50a1SJim Ingham         // Instance variables to hold the values for command options.
364e16c50a1SJim Ingham 
365012d4fcaSEnrico Granata         OptionValueBoolean m_stop_on_error;
366340b0309SGreg Clayton         OptionValueBoolean m_silent_run;
367340b0309SGreg Clayton         OptionValueBoolean m_stop_on_continue;
368e16c50a1SJim Ingham     };
369e16c50a1SJim Ingham 
370ebc09c36SJim Ingham     bool
3715a988416SJim Ingham     DoExecute(Args& command, CommandReturnObject &result)
372ebc09c36SJim Ingham     {
373c7bece56SGreg Clayton         const size_t argc = command.GetArgumentCount();
374ebc09c36SJim Ingham         if (argc == 1)
375ebc09c36SJim Ingham         {
3765a988416SJim Ingham             const char *filename = command.GetArgumentAtIndex(0);
377ebc09c36SJim Ingham 
3781ee3853fSJohnny Chen             FileSpec cmd_file (filename, true);
379e16c50a1SJim Ingham             ExecutionContext *exe_ctx = NULL;  // Just use the default context.
380ebc09c36SJim Ingham 
381340b0309SGreg Clayton             // If any options were set, then use them
382340b0309SGreg Clayton             if (m_options.m_stop_on_error.OptionWasSet()    ||
383340b0309SGreg Clayton                 m_options.m_silent_run.OptionWasSet()       ||
384340b0309SGreg Clayton                 m_options.m_stop_on_continue.OptionWasSet())
385340b0309SGreg Clayton             {
386340b0309SGreg Clayton                 // Use user set settings
38726c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
38826c7bf93SJim Ingham                 options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue());
38926c7bf93SJim Ingham                 options.SetStopOnError (m_options.m_stop_on_error.GetCurrentValue());
3907d8555c4SJim Ingham                 options.SetEchoCommands (!m_options.m_silent_run.GetCurrentValue());
3917d8555c4SJim Ingham                 options.SetPrintResults (!m_options.m_silent_run.GetCurrentValue());
39226c7bf93SJim Ingham 
393e16c50a1SJim Ingham                 m_interpreter.HandleCommandsFromFile (cmd_file,
394e16c50a1SJim Ingham                                                       exe_ctx,
39526c7bf93SJim Ingham                                                       options,
396e16c50a1SJim Ingham                                                       result);
397340b0309SGreg Clayton 
398340b0309SGreg Clayton             }
399340b0309SGreg Clayton             else
400340b0309SGreg Clayton             {
401340b0309SGreg Clayton                 // No options were set, inherit any settings from nested "command source" commands,
402340b0309SGreg Clayton                 // or set to sane default settings...
40326c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
404340b0309SGreg Clayton                 m_interpreter.HandleCommandsFromFile (cmd_file,
405340b0309SGreg Clayton                                                       exe_ctx,
40626c7bf93SJim Ingham                                                       options,
407340b0309SGreg Clayton                                                       result);
408340b0309SGreg Clayton 
409340b0309SGreg Clayton             }
410ebc09c36SJim Ingham         }
411ebc09c36SJim Ingham         else
412ebc09c36SJim Ingham         {
413ebc09c36SJim Ingham             result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName());
414ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
415ebc09c36SJim Ingham         }
416ebc09c36SJim Ingham         return result.Succeeded();
417ebc09c36SJim Ingham 
418ebc09c36SJim Ingham     }
4195a988416SJim Ingham     CommandOptions m_options;
420ebc09c36SJim Ingham };
421ebc09c36SJim Ingham 
422e0d378b3SGreg Clayton OptionDefinition
423e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] =
424e16c50a1SJim Ingham {
425d37221dcSZachary Turner { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean,    "If true, stop executing commands on error."},
426d37221dcSZachary Turner { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on continue."},
427d37221dcSZachary Turner { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "If true don't echo commands while executing."},
428d37221dcSZachary Turner { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
429e16c50a1SJim Ingham };
430e16c50a1SJim Ingham 
431ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
432ebc09c36SJim Ingham //-------------------------------------------------------------------------
433ebc09c36SJim Ingham // CommandObjectCommandsAlias
434ebc09c36SJim Ingham //-------------------------------------------------------------------------
435ebc09c36SJim Ingham 
436be93a35aSEnrico Granata static const char *g_python_command_instructions =   "Enter your Python command(s). Type 'DONE' to end.\n"
437be93a35aSEnrico Granata                                                      "You must define a Python function with this signature:\n"
43844d93782SGreg Clayton                                                      "def my_command_impl(debugger, args, result, internal_dict):\n";
439be93a35aSEnrico Granata 
440be93a35aSEnrico Granata 
4415a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw
442ebc09c36SJim Ingham {
443be93a35aSEnrico Granata 
444be93a35aSEnrico Granata 
445ebc09c36SJim Ingham public:
446a7015092SGreg Clayton     CommandObjectCommandsAlias (CommandInterpreter &interpreter) :
4475a988416SJim Ingham         CommandObjectRaw (interpreter,
4480e5e5a79SGreg Clayton                        "command alias",
449e3d26315SCaroline Tice                        "Allow users to define their own debugger command abbreviations.",
450405fe67fSCaroline Tice                        NULL)
451ebc09c36SJim Ingham     {
452ebc09c36SJim Ingham         SetHelpLong(
453ebc09c36SJim Ingham     "'alias' allows the user to create a short-cut or abbreviation for long \n\
454ebc09c36SJim Ingham     commands, multi-word commands, and commands that take particular options. \n\
455ebc09c36SJim Ingham     Below are some simple examples of how one might use the 'alias' command: \n\
45669c12ccbSJason Molenda     \n    'command alias sc script'            // Creates the abbreviation 'sc' for the 'script' \n\
457ebc09c36SJim Ingham                                          // command. \n\
45869c12ccbSJason Molenda     'command alias bp breakpoint'        // Creates the abbreviation 'bp' for the 'breakpoint' \n\
459ebc09c36SJim Ingham                                          // command.  Since breakpoint commands are two-word \n\
460ebc09c36SJim Ingham                                          // commands, the user will still need to enter the \n\
461ebc09c36SJim Ingham                                          // second word after 'bp', e.g. 'bp enable' or \n\
462ebc09c36SJim Ingham                                          // 'bp delete'. \n\
46369c12ccbSJason Molenda     'command alias bpl breakpoint list'  // Creates the abbreviation 'bpl' for the \n\
464ebc09c36SJim Ingham                                          // two-word command 'breakpoint list'. \n\
465ebc09c36SJim Ingham     \nAn alias can include some options for the command, with the values either \n\
466ebc09c36SJim Ingham     filled in at the time the alias is created, or specified as positional \n\
467ebc09c36SJim Ingham     arguments, to be filled in when the alias is invoked.  The following example \n\
468ebc09c36SJim Ingham     shows how to create aliases with options: \n\
469ebc09c36SJim Ingham     \n\
47069c12ccbSJason Molenda     'command alias bfl breakpoint set -f %1 -l %2' \n\
471ebc09c36SJim Ingham     \nThis creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \n\
472ebc09c36SJim Ingham     options already part of the alias.  So if the user wants to set a breakpoint \n\
473ebc09c36SJim Ingham     by file and line without explicitly having to use the -f and -l options, the \n\
474ebc09c36SJim Ingham     user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \n\
475ebc09c36SJim Ingham     for the actual arguments that will be passed when the alias command is used. \n\
476ebc09c36SJim Ingham     The number in the placeholder refers to the position/order the actual value \n\
47781ded935SJim Ingham     occupies when the alias is used.  All the occurrences of '%1' in the alias \n\
478ebc09c36SJim Ingham     will be replaced with the first argument, all the occurrences of '%2' in the \n\
479ebc09c36SJim Ingham     alias will be replaced with the second argument, and so on.  This also allows \n\
480ebc09c36SJim Ingham     actual arguments to be used multiple times within an alias (see 'process \n\
48181ded935SJim Ingham     launch' example below).  \n\
48281ded935SJim Ingham     Note: the positional arguments must substitute as whole words in the resultant\n\
48381ded935SJim Ingham     command, so you can't at present do something like:\n\
48481ded935SJim Ingham     \n\
48569c12ccbSJason Molenda     command alias bcppfl breakpoint set -f %1.cpp -l %2\n\
48681ded935SJim Ingham     \n\
48781ded935SJim Ingham     to get the file extension \".cpp\" automatically appended.  For more complex\n\
48881ded935SJim Ingham     aliasing, use the \"command regex\" command instead.\n\
48981ded935SJim Ingham     \nSo in the 'bfl' case, the actual file value will be \n\
490ebc09c36SJim Ingham     filled in with the first argument following 'bfl' and the actual line number \n\
491ebc09c36SJim Ingham     value will be filled in with the second argument.  The user would use this \n\
492ebc09c36SJim Ingham     alias as follows: \n\
49369c12ccbSJason Molenda     \n    (lldb)  command alias bfl breakpoint set -f %1 -l %2 \n\
494ebc09c36SJim Ingham     <... some time later ...> \n\
49509799af6SCaroline Tice     (lldb)  bfl my-file.c 137 \n\
496ebc09c36SJim Ingham     \nThis would be the same as if the user had entered \n\
497ebc09c36SJim Ingham     'breakpoint set -f my-file.c -l 137'. \n\
498ebc09c36SJim Ingham     \nAnother example: \n\
49969c12ccbSJason Molenda     \n    (lldb)  command alias pltty  process launch -s -o %1 -e %1 \n\
50009799af6SCaroline Tice     (lldb)  pltty /dev/tty0 \n\
501ebc09c36SJim Ingham            // becomes 'process launch -s -o /dev/tty0 -e /dev/tty0' \n\
502ebc09c36SJim Ingham     \nIf the user always wanted to pass the same value to a particular option, the \n\
503ebc09c36SJim Ingham     alias could be defined with that value directly in the alias as a constant, \n\
504ebc09c36SJim Ingham     rather than using a positional placeholder: \n\
50569c12ccbSJason Molenda     \n    command alias bl3  breakpoint set -f %1 -l 3  // Always sets a breakpoint on line \n\
506ebc09c36SJim Ingham                                                    // 3 of whatever file is indicated. \n");
507ebc09c36SJim Ingham 
508405fe67fSCaroline Tice         CommandArgumentEntry arg1;
509405fe67fSCaroline Tice         CommandArgumentEntry arg2;
510405fe67fSCaroline Tice         CommandArgumentEntry arg3;
511405fe67fSCaroline Tice         CommandArgumentData alias_arg;
512405fe67fSCaroline Tice         CommandArgumentData cmd_arg;
513405fe67fSCaroline Tice         CommandArgumentData options_arg;
514405fe67fSCaroline Tice 
515405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
516405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
517405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
518405fe67fSCaroline Tice 
519405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
520405fe67fSCaroline Tice         arg1.push_back (alias_arg);
521405fe67fSCaroline Tice 
522405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
523405fe67fSCaroline Tice         cmd_arg.arg_type = eArgTypeCommandName;
524405fe67fSCaroline Tice         cmd_arg.arg_repetition = eArgRepeatPlain;
525405fe67fSCaroline Tice 
526405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
527405fe67fSCaroline Tice         arg2.push_back (cmd_arg);
528405fe67fSCaroline Tice 
529405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
530405fe67fSCaroline Tice         options_arg.arg_type = eArgTypeAliasOptions;
531405fe67fSCaroline Tice         options_arg.arg_repetition = eArgRepeatOptional;
532405fe67fSCaroline Tice 
533405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
534405fe67fSCaroline Tice         arg3.push_back (options_arg);
535405fe67fSCaroline Tice 
536405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
537405fe67fSCaroline Tice         m_arguments.push_back (arg1);
538405fe67fSCaroline Tice         m_arguments.push_back (arg2);
539405fe67fSCaroline Tice         m_arguments.push_back (arg3);
540ebc09c36SJim Ingham     }
541ebc09c36SJim Ingham 
542ebc09c36SJim Ingham     ~CommandObjectCommandsAlias ()
543ebc09c36SJim Ingham     {
544ebc09c36SJim Ingham     }
545ebc09c36SJim Ingham 
5465a988416SJim Ingham protected:
5475a988416SJim Ingham     virtual bool
5485a988416SJim Ingham     DoExecute (const char *raw_command_line, CommandReturnObject &result)
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()));
639844d2303SCaroline Tice                 if (temp_option_arg_sp.get())
640844d2303SCaroline Tice                 {
641844d2303SCaroline Tice                     if (option_arg_vector->size() == 0)
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);
651844d2303SCaroline Tice                 if (option_arg_vector->size() > 0)
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;
694ebc09c36SJim Ingham              if (command_obj_sp.get())
695ebc09c36SJim Ingham              {
696ebc09c36SJim Ingham                  CommandObject *cmd_obj = command_obj_sp.get();
697c982c768SGreg Clayton                  CommandObject *sub_cmd_obj = NULL;
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());
708ebc09c36SJim Ingham                          if (subcommand_obj_sp.get())
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()));
751ebc09c36SJim Ingham                      if (tmp_option_arg_sp.get())
752ebc09c36SJim Ingham                      {
753ebc09c36SJim Ingham                          if (option_arg_vector->size() == 0)
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);
764ebc09c36SJim Ingham                  if (option_arg_vector->size() > 0)
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     }
7785a988416SJim Ingham 
779ebc09c36SJim Ingham };
780ebc09c36SJim Ingham 
781ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
782ebc09c36SJim Ingham //-------------------------------------------------------------------------
783ebc09c36SJim Ingham // CommandObjectCommandsUnalias
784ebc09c36SJim Ingham //-------------------------------------------------------------------------
785ebc09c36SJim Ingham 
7865a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed
787ebc09c36SJim Ingham {
788ebc09c36SJim Ingham public:
789a7015092SGreg Clayton     CommandObjectCommandsUnalias (CommandInterpreter &interpreter) :
7905a988416SJim Ingham         CommandObjectParsed (interpreter,
7910e5e5a79SGreg Clayton                        "command unalias",
79286ddae50SCaroline Tice                        "Allow the user to remove/delete a user-defined command abbreviation.",
793405fe67fSCaroline Tice                        NULL)
794ebc09c36SJim Ingham     {
795405fe67fSCaroline Tice         CommandArgumentEntry arg;
796405fe67fSCaroline Tice         CommandArgumentData alias_arg;
797405fe67fSCaroline Tice 
798405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
799405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
800405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
801405fe67fSCaroline Tice 
802405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
803405fe67fSCaroline Tice         arg.push_back (alias_arg);
804405fe67fSCaroline Tice 
805405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
806405fe67fSCaroline Tice         m_arguments.push_back (arg);
807ebc09c36SJim Ingham     }
808ebc09c36SJim Ingham 
809ebc09c36SJim Ingham     ~CommandObjectCommandsUnalias()
810ebc09c36SJim Ingham     {
811ebc09c36SJim Ingham     }
812ebc09c36SJim Ingham 
8135a988416SJim Ingham protected:
814ebc09c36SJim Ingham     bool
8155a988416SJim Ingham     DoExecute (Args& args, CommandReturnObject &result)
816ebc09c36SJim Ingham     {
817ebc09c36SJim Ingham         CommandObject::CommandMap::iterator pos;
818ebc09c36SJim Ingham         CommandObject *cmd_obj;
819ebc09c36SJim Ingham 
820ebc09c36SJim Ingham         if (args.GetArgumentCount() != 0)
821ebc09c36SJim Ingham         {
822ebc09c36SJim Ingham             const char *command_name = args.GetArgumentAtIndex(0);
823a7015092SGreg Clayton             cmd_obj = m_interpreter.GetCommandObject(command_name);
824ebc09c36SJim Ingham             if (cmd_obj)
825ebc09c36SJim Ingham             {
826a7015092SGreg Clayton                 if (m_interpreter.CommandExists (command_name))
827ebc09c36SJim Ingham                 {
828b547278cSGreg Clayton                     if (cmd_obj->IsRemovable())
829b547278cSGreg Clayton                     {
830b547278cSGreg Clayton                         result.AppendErrorWithFormat ("'%s' is not an alias, it is a debugger command which can be removed using the 'command delete' command.\n",
831b547278cSGreg Clayton                                                       command_name);
832b547278cSGreg Clayton                     }
833b547278cSGreg Clayton                     else
834b547278cSGreg Clayton                     {
835ebc09c36SJim Ingham                         result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
836ebc09c36SJim Ingham                                                       command_name);
837b547278cSGreg Clayton                     }
838ebc09c36SJim Ingham                     result.SetStatus (eReturnStatusFailed);
839ebc09c36SJim Ingham                 }
840ebc09c36SJim Ingham                 else
841ebc09c36SJim Ingham                 {
842ebc09c36SJim Ingham 
843a7015092SGreg Clayton                     if (m_interpreter.RemoveAlias (command_name) == false)
844ebc09c36SJim Ingham                     {
845a7015092SGreg Clayton                         if (m_interpreter.AliasExists (command_name))
846ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n",
847ebc09c36SJim Ingham                                                           command_name);
848ebc09c36SJim Ingham                         else
849ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name);
850ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusFailed);
851ebc09c36SJim Ingham                     }
852ebc09c36SJim Ingham                     else
853ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
854ebc09c36SJim Ingham                 }
855ebc09c36SJim Ingham             }
856ebc09c36SJim Ingham             else
857ebc09c36SJim Ingham             {
858ebc09c36SJim Ingham                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a "
859ebc09c36SJim Ingham                                               "current list of commands.\n",
860ebc09c36SJim Ingham                                              command_name);
861ebc09c36SJim Ingham                 result.SetStatus (eReturnStatusFailed);
862ebc09c36SJim Ingham             }
863ebc09c36SJim Ingham         }
864ebc09c36SJim Ingham         else
865ebc09c36SJim Ingham         {
866ebc09c36SJim Ingham             result.AppendError ("must call 'unalias' with a valid alias");
867ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
868ebc09c36SJim Ingham         }
869ebc09c36SJim Ingham 
870ebc09c36SJim Ingham         return result.Succeeded();
871ebc09c36SJim Ingham     }
872ebc09c36SJim Ingham };
873ebc09c36SJim Ingham 
874b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
875b547278cSGreg Clayton //-------------------------------------------------------------------------
876b547278cSGreg Clayton // CommandObjectCommandsDelete
877b547278cSGreg Clayton //-------------------------------------------------------------------------
878b547278cSGreg Clayton 
879b547278cSGreg Clayton class CommandObjectCommandsDelete : public CommandObjectParsed
880b547278cSGreg Clayton {
881b547278cSGreg Clayton public:
882b547278cSGreg Clayton     CommandObjectCommandsDelete (CommandInterpreter &interpreter) :
883b547278cSGreg Clayton     CommandObjectParsed (interpreter,
884b547278cSGreg Clayton                          "command delete",
885b547278cSGreg Clayton                          "Allow the user to delete user-defined regular expression, python or multi-word commands.",
886b547278cSGreg Clayton                          NULL)
887b547278cSGreg Clayton     {
888b547278cSGreg Clayton         CommandArgumentEntry arg;
889b547278cSGreg Clayton         CommandArgumentData alias_arg;
890b547278cSGreg Clayton 
891b547278cSGreg Clayton         // Define the first (and only) variant of this arg.
892b547278cSGreg Clayton         alias_arg.arg_type = eArgTypeCommandName;
893b547278cSGreg Clayton         alias_arg.arg_repetition = eArgRepeatPlain;
894b547278cSGreg Clayton 
895b547278cSGreg Clayton         // There is only one variant this argument could be; put it into the argument entry.
896b547278cSGreg Clayton         arg.push_back (alias_arg);
897b547278cSGreg Clayton 
898b547278cSGreg Clayton         // Push the data for the first argument into the m_arguments vector.
899b547278cSGreg Clayton         m_arguments.push_back (arg);
900b547278cSGreg Clayton     }
901b547278cSGreg Clayton 
902b547278cSGreg Clayton     ~CommandObjectCommandsDelete()
903b547278cSGreg Clayton     {
904b547278cSGreg Clayton     }
905b547278cSGreg Clayton 
906b547278cSGreg Clayton protected:
907b547278cSGreg Clayton     bool
908b547278cSGreg Clayton     DoExecute (Args& args, CommandReturnObject &result)
909b547278cSGreg Clayton     {
910b547278cSGreg Clayton         CommandObject::CommandMap::iterator pos;
911b547278cSGreg Clayton 
912b547278cSGreg Clayton         if (args.GetArgumentCount() != 0)
913b547278cSGreg Clayton         {
914b547278cSGreg Clayton             const char *command_name = args.GetArgumentAtIndex(0);
915b547278cSGreg Clayton             if (m_interpreter.CommandExists (command_name))
916b547278cSGreg Clayton             {
917b547278cSGreg Clayton                 if (m_interpreter.RemoveCommand (command_name))
918b547278cSGreg Clayton                 {
919b547278cSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
920b547278cSGreg Clayton                 }
921b547278cSGreg Clayton                 else
922b547278cSGreg Clayton                 {
923b547278cSGreg Clayton                     result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
924b547278cSGreg Clayton                                                   command_name);
925b547278cSGreg Clayton                     result.SetStatus (eReturnStatusFailed);
926b547278cSGreg Clayton                 }
927b547278cSGreg Clayton             }
928b547278cSGreg Clayton             else
929b547278cSGreg Clayton             {
930b547278cSGreg Clayton                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a current list of commands.\n",
931b547278cSGreg Clayton                                               command_name);
932b547278cSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
933b547278cSGreg Clayton             }
934b547278cSGreg Clayton         }
935b547278cSGreg Clayton         else
936b547278cSGreg Clayton         {
937b547278cSGreg Clayton             result.AppendErrorWithFormat ("must call '%s' with one or more valid user defined regular expression, python or multi-word command names", GetCommandName ());
938b547278cSGreg Clayton             result.SetStatus (eReturnStatusFailed);
939b547278cSGreg Clayton         }
940b547278cSGreg Clayton 
941b547278cSGreg Clayton         return result.Succeeded();
942b547278cSGreg Clayton     }
943b547278cSGreg Clayton };
944b547278cSGreg Clayton 
945de164aaaSGreg Clayton //-------------------------------------------------------------------------
946de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
947de164aaaSGreg Clayton //-------------------------------------------------------------------------
9485a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
949de164aaaSGreg Clayton 
95044d93782SGreg Clayton class CommandObjectCommandsAddRegex :
95144d93782SGreg Clayton     public CommandObjectParsed,
952ea508635SGreg Clayton     public IOHandlerDelegateMultiline
953de164aaaSGreg Clayton {
954de164aaaSGreg Clayton public:
955de164aaaSGreg Clayton     CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) :
9565a988416SJim Ingham         CommandObjectParsed (interpreter,
9570e5e5a79SGreg Clayton                        "command regex",
958de164aaaSGreg Clayton                        "Allow the user to create a regular expression command.",
9590e5e5a79SGreg Clayton                        "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
960ea508635SGreg Clayton         IOHandlerDelegateMultiline ("", IOHandlerDelegate::Completion::LLDBCommand),
961de164aaaSGreg Clayton         m_options (interpreter)
962de164aaaSGreg Clayton     {
9630e5e5a79SGreg Clayton         SetHelpLong(
9640e5e5a79SGreg Clayton "This command allows the user to create powerful regular expression commands\n"
9650e5e5a79SGreg Clayton "with substitutions. The regular expressions and substitutions are specified\n"
966d93c4a33SBruce Mitchener "using the regular expression substitution format of:\n"
9670e5e5a79SGreg Clayton "\n"
9680e5e5a79SGreg Clayton "    s/<regex>/<subst>/\n"
9690e5e5a79SGreg Clayton "\n"
9700e5e5a79SGreg Clayton "<regex> is a regular expression that can use parenthesis to capture regular\n"
9710e5e5a79SGreg Clayton "expression input and substitute the captured matches in the output using %1\n"
9720e5e5a79SGreg Clayton "for the first match, %2 for the second, and so on.\n"
9730e5e5a79SGreg Clayton "\n"
9740e5e5a79SGreg Clayton "The regular expressions can all be specified on the command line if more than\n"
9750e5e5a79SGreg Clayton "one argument is provided. If just the command name is provided on the command\n"
9760e5e5a79SGreg Clayton "line, then the regular expressions and substitutions can be entered on separate\n"
9770e5e5a79SGreg Clayton " lines, followed by an empty line to terminate the command definition.\n"
9780e5e5a79SGreg Clayton "\n"
9790e5e5a79SGreg Clayton "EXAMPLES\n"
9800e5e5a79SGreg Clayton "\n"
981adc43c99SSean Callanan "The following example will define a regular expression command named 'f' that\n"
9820e5e5a79SGreg Clayton "will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if\n"
9830e5e5a79SGreg Clayton "a number follows 'f':\n"
984adc43c99SSean Callanan "\n"
9850e5e5a79SGreg Clayton "    (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/'\n"
986adc43c99SSean Callanan "\n"
9870e5e5a79SGreg Clayton                     );
988de164aaaSGreg Clayton     }
989de164aaaSGreg Clayton 
990de164aaaSGreg Clayton     ~CommandObjectCommandsAddRegex()
991de164aaaSGreg Clayton     {
992de164aaaSGreg Clayton     }
993de164aaaSGreg Clayton 
994de164aaaSGreg Clayton 
9955a988416SJim Ingham protected:
99644d93782SGreg Clayton 
997ea508635SGreg Clayton     void
998ea508635SGreg Clayton     IOHandlerActivated (IOHandler &io_handler) override
99944d93782SGreg Clayton     {
100044d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
100144d93782SGreg Clayton         if (output_sp)
100244d93782SGreg Clayton         {
100344d93782SGreg 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");
100444d93782SGreg Clayton             output_sp->Flush();
100544d93782SGreg Clayton         }
100644d93782SGreg Clayton     }
100744d93782SGreg Clayton 
1008ea508635SGreg Clayton     void
1009ea508635SGreg Clayton     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
101044d93782SGreg Clayton     {
101144d93782SGreg Clayton         io_handler.SetIsDone(true);
101244d93782SGreg Clayton         if (m_regex_cmd_ap.get())
101344d93782SGreg Clayton         {
101444d93782SGreg Clayton             StringList lines;
101544d93782SGreg Clayton             if (lines.SplitIntoLines (data))
101644d93782SGreg Clayton             {
101744d93782SGreg Clayton                 const size_t num_lines = lines.GetSize();
101844d93782SGreg Clayton                 bool check_only = false;
101944d93782SGreg Clayton                 for (size_t i=0; i<num_lines; ++i)
102044d93782SGreg Clayton                 {
102144d93782SGreg Clayton                     llvm::StringRef bytes_strref (lines[i]);
102244d93782SGreg Clayton                     Error error = AppendRegexSubstitution (bytes_strref, check_only);
102344d93782SGreg Clayton                     if (error.Fail())
102444d93782SGreg Clayton                     {
102544d93782SGreg Clayton                         if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode())
102644d93782SGreg Clayton                         {
102744d93782SGreg Clayton                             StreamSP out_stream = m_interpreter.GetDebugger().GetAsyncOutputStream();
102844d93782SGreg Clayton                             out_stream->Printf("error: %s\n", error.AsCString());
102944d93782SGreg Clayton                         }
103044d93782SGreg Clayton                     }
103144d93782SGreg Clayton                 }
103244d93782SGreg Clayton             }
103344d93782SGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
103444d93782SGreg Clayton             {
103544d93782SGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
103644d93782SGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
103744d93782SGreg Clayton             }
103844d93782SGreg Clayton         }
103944d93782SGreg Clayton     }
104044d93782SGreg Clayton 
1041de164aaaSGreg Clayton     bool
1042b0a1814fSEric Christopher     DoExecute (Args& command, CommandReturnObject &result) override
1043de164aaaSGreg Clayton     {
10445a988416SJim Ingham         const size_t argc = command.GetArgumentCount();
10450e5e5a79SGreg Clayton         if (argc == 0)
1046de164aaaSGreg Clayton         {
104769c12ccbSJason Molenda             result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
10480e5e5a79SGreg Clayton             result.SetStatus (eReturnStatusFailed);
10490e5e5a79SGreg Clayton         }
10500e5e5a79SGreg Clayton         else
10510e5e5a79SGreg Clayton         {
10520e5e5a79SGreg Clayton             Error error;
10535a988416SJim Ingham             const char *name = command.GetArgumentAtIndex(0);
1054de164aaaSGreg Clayton             m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter,
1055de164aaaSGreg Clayton                                                                  name,
1056de164aaaSGreg Clayton                                                                  m_options.GetHelp (),
1057de164aaaSGreg Clayton                                                                  m_options.GetSyntax (),
1058b547278cSGreg Clayton                                                                  10,
1059b547278cSGreg Clayton                                                                  0,
1060b547278cSGreg Clayton                                                                  true));
10610e5e5a79SGreg Clayton 
10620e5e5a79SGreg Clayton             if (argc == 1)
10630e5e5a79SGreg Clayton             {
106444d93782SGreg Clayton                 Debugger &debugger = m_interpreter.GetDebugger();
1065e30f11d9SKate Stone                 bool color_prompt = debugger.GetUseColor();
106644d93782SGreg Clayton                 const bool multiple_lines = true; // Get multiple lines
106744d93782SGreg Clayton                 IOHandlerSP io_handler_sp (new IOHandlerEditline (debugger,
1068e30f11d9SKate Stone                                                                   IOHandler::Type::Other,
106973d80faaSGreg Clayton                                                                   "lldb-regex", // Name of input reader for history
1070ea508635SGreg Clayton                                                                   "> ",         // Prompt
1071e30f11d9SKate Stone                                                                   NULL,         // Continuation prompt
107244d93782SGreg Clayton                                                                   multiple_lines,
1073e30f11d9SKate Stone                                                                   color_prompt,
1074f6913cd7SGreg Clayton                                                                   0,            // Don't show line numbers
107544d93782SGreg Clayton                                                                   *this));
107644d93782SGreg Clayton 
107744d93782SGreg Clayton                 if (io_handler_sp)
1078de164aaaSGreg Clayton                 {
107944d93782SGreg Clayton                     debugger.PushIOHandler(io_handler_sp);
1080de164aaaSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1081de164aaaSGreg Clayton                 }
1082de164aaaSGreg Clayton             }
1083de164aaaSGreg Clayton             else
1084de164aaaSGreg Clayton             {
10850e5e5a79SGreg Clayton                 for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx)
10860e5e5a79SGreg Clayton                 {
10875a988416SJim Ingham                     llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx));
108844d93782SGreg Clayton                     bool check_only = false;
108944d93782SGreg Clayton                     error = AppendRegexSubstitution (arg_strref, check_only);
10900e5e5a79SGreg Clayton                     if (error.Fail())
10910e5e5a79SGreg Clayton                         break;
10920e5e5a79SGreg Clayton                 }
10930e5e5a79SGreg Clayton 
10940e5e5a79SGreg Clayton                 if (error.Success())
10950e5e5a79SGreg Clayton                 {
10960e5e5a79SGreg Clayton                     AddRegexCommandToInterpreter();
10970e5e5a79SGreg Clayton                 }
10980e5e5a79SGreg Clayton             }
10990e5e5a79SGreg Clayton             if (error.Fail())
11000e5e5a79SGreg Clayton             {
11010e5e5a79SGreg Clayton                 result.AppendError (error.AsCString());
1102de164aaaSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
1103de164aaaSGreg Clayton             }
11040e5e5a79SGreg Clayton         }
11050e5e5a79SGreg Clayton 
1106de164aaaSGreg Clayton         return result.Succeeded();
1107de164aaaSGreg Clayton     }
1108de164aaaSGreg Clayton 
11090e5e5a79SGreg Clayton     Error
111044d93782SGreg Clayton     AppendRegexSubstitution (const llvm::StringRef &regex_sed, bool check_only)
1111de164aaaSGreg Clayton     {
11120e5e5a79SGreg Clayton         Error error;
11130e5e5a79SGreg Clayton 
11140e5e5a79SGreg Clayton         if (m_regex_cmd_ap.get() == NULL)
1115de164aaaSGreg Clayton         {
11160e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'",
11170e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11180e5e5a79SGreg Clayton                                            regex_sed.data());
11190e5e5a79SGreg Clayton             return error;
1120de164aaaSGreg Clayton         }
11210e5e5a79SGreg Clayton 
11220e5e5a79SGreg Clayton         size_t regex_sed_size = regex_sed.size();
11230e5e5a79SGreg Clayton 
11240e5e5a79SGreg Clayton         if (regex_sed_size <= 1)
11250e5e5a79SGreg Clayton         {
11260e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'",
11270e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11280e5e5a79SGreg Clayton                                            regex_sed.data());
11290e5e5a79SGreg Clayton             return error;
11300e5e5a79SGreg Clayton         }
11310e5e5a79SGreg Clayton 
11320e5e5a79SGreg Clayton         if (regex_sed[0] != 's')
11330e5e5a79SGreg Clayton         {
11340e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'",
11350e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11360e5e5a79SGreg Clayton                                            regex_sed.data());
11370e5e5a79SGreg Clayton             return error;
11380e5e5a79SGreg Clayton         }
11390e5e5a79SGreg Clayton         const size_t first_separator_char_pos = 1;
11400e5e5a79SGreg Clayton         // use the char that follows 's' as the regex separator character
11410e5e5a79SGreg Clayton         // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
11420e5e5a79SGreg Clayton         const char separator_char = regex_sed[first_separator_char_pos];
11430e5e5a79SGreg Clayton         const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1);
11440e5e5a79SGreg Clayton 
11450e5e5a79SGreg Clayton         if (second_separator_char_pos == std::string::npos)
11460e5e5a79SGreg Clayton         {
1147ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s' in '%.*s'",
11480e5e5a79SGreg Clayton                                            separator_char,
11490e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - first_separator_char_pos - 1),
1150ea508635SGreg Clayton                                            regex_sed.data() + (first_separator_char_pos + 1),
1151ea508635SGreg Clayton                                            (int)regex_sed.size(),
1152ea508635SGreg Clayton                                            regex_sed.data());
11530e5e5a79SGreg Clayton             return error;
11540e5e5a79SGreg Clayton         }
11550e5e5a79SGreg Clayton 
11560e5e5a79SGreg Clayton         const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1);
11570e5e5a79SGreg Clayton 
11580e5e5a79SGreg Clayton         if (third_separator_char_pos == std::string::npos)
11590e5e5a79SGreg Clayton         {
1160ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s' in '%.*s'",
11610e5e5a79SGreg Clayton                                            separator_char,
11620e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - second_separator_char_pos - 1),
1163ea508635SGreg Clayton                                            regex_sed.data() + (second_separator_char_pos + 1),
1164ea508635SGreg Clayton                                            (int)regex_sed.size(),
1165ea508635SGreg Clayton                                            regex_sed.data());
11660e5e5a79SGreg Clayton             return error;
11670e5e5a79SGreg Clayton         }
11680e5e5a79SGreg Clayton 
11690e5e5a79SGreg Clayton         if (third_separator_char_pos != regex_sed_size - 1)
11700e5e5a79SGreg Clayton         {
11710e5e5a79SGreg Clayton             // Make sure that everything that follows the last regex
11720e5e5a79SGreg Clayton             // separator char
11730e5e5a79SGreg Clayton             if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos)
11740e5e5a79SGreg Clayton             {
11750e5e5a79SGreg Clayton                 error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'",
11760e5e5a79SGreg Clayton                                                (int)third_separator_char_pos + 1,
11770e5e5a79SGreg Clayton                                                regex_sed.data(),
11780e5e5a79SGreg Clayton                                                (int)(regex_sed.size() - third_separator_char_pos - 1),
11790e5e5a79SGreg Clayton                                                regex_sed.data() + (third_separator_char_pos + 1));
11800e5e5a79SGreg Clayton                 return error;
11810e5e5a79SGreg Clayton             }
11820e5e5a79SGreg Clayton 
11830e5e5a79SGreg Clayton         }
11840e5e5a79SGreg Clayton         else if (first_separator_char_pos + 1 == second_separator_char_pos)
11850e5e5a79SGreg Clayton         {
11860e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11870e5e5a79SGreg Clayton                                            separator_char,
11880e5e5a79SGreg Clayton                                            separator_char,
11890e5e5a79SGreg Clayton                                            separator_char,
11900e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11910e5e5a79SGreg Clayton                                            regex_sed.data());
11920e5e5a79SGreg Clayton             return error;
11930e5e5a79SGreg Clayton         }
11940e5e5a79SGreg Clayton         else if (second_separator_char_pos + 1 == third_separator_char_pos)
11950e5e5a79SGreg Clayton         {
11960e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11970e5e5a79SGreg Clayton                                            separator_char,
11980e5e5a79SGreg Clayton                                            separator_char,
11990e5e5a79SGreg Clayton                                            separator_char,
12000e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
12010e5e5a79SGreg Clayton                                            regex_sed.data());
12020e5e5a79SGreg Clayton             return error;
12030e5e5a79SGreg Clayton         }
120444d93782SGreg Clayton 
120544d93782SGreg Clayton         if (check_only == false)
120644d93782SGreg Clayton         {
12070e5e5a79SGreg Clayton             std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1));
12080e5e5a79SGreg Clayton             std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1));
12090e5e5a79SGreg Clayton             m_regex_cmd_ap->AddRegexCommand (regex.c_str(),
12100e5e5a79SGreg Clayton                                              subst.c_str());
121144d93782SGreg Clayton         }
12120e5e5a79SGreg Clayton         return error;
1213de164aaaSGreg Clayton     }
1214de164aaaSGreg Clayton 
1215de164aaaSGreg Clayton     void
12160e5e5a79SGreg Clayton     AddRegexCommandToInterpreter()
1217de164aaaSGreg Clayton     {
1218de164aaaSGreg Clayton         if (m_regex_cmd_ap.get())
1219de164aaaSGreg Clayton         {
1220de164aaaSGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
1221de164aaaSGreg Clayton             {
1222de164aaaSGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
1223de164aaaSGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1224de164aaaSGreg Clayton             }
1225de164aaaSGreg Clayton         }
1226de164aaaSGreg Clayton     }
1227de164aaaSGreg Clayton 
1228de164aaaSGreg Clayton private:
12297b0992d9SGreg Clayton     std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1230de164aaaSGreg Clayton 
1231de164aaaSGreg Clayton      class CommandOptions : public Options
1232de164aaaSGreg Clayton      {
1233de164aaaSGreg Clayton      public:
1234de164aaaSGreg Clayton 
1235de164aaaSGreg Clayton          CommandOptions (CommandInterpreter &interpreter) :
1236de164aaaSGreg Clayton             Options (interpreter)
1237de164aaaSGreg Clayton          {
1238de164aaaSGreg Clayton          }
1239de164aaaSGreg Clayton 
1240de164aaaSGreg Clayton          virtual
1241de164aaaSGreg Clayton          ~CommandOptions (){}
1242de164aaaSGreg Clayton 
1243de164aaaSGreg Clayton          virtual Error
1244de164aaaSGreg Clayton          SetOptionValue (uint32_t option_idx, const char *option_arg)
1245de164aaaSGreg Clayton          {
1246de164aaaSGreg Clayton              Error error;
12473bcdfc0eSGreg Clayton              const int short_option = m_getopt_table[option_idx].val;
1248de164aaaSGreg Clayton 
1249de164aaaSGreg Clayton              switch (short_option)
1250de164aaaSGreg Clayton              {
1251de164aaaSGreg Clayton                  case 'h':
1252de164aaaSGreg Clayton                      m_help.assign (option_arg);
1253de164aaaSGreg Clayton                      break;
1254de164aaaSGreg Clayton                  case 's':
1255de164aaaSGreg Clayton                      m_syntax.assign (option_arg);
1256de164aaaSGreg Clayton                      break;
1257de164aaaSGreg Clayton 
1258de164aaaSGreg Clayton                  default:
125986edbf41SGreg Clayton                      error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1260de164aaaSGreg Clayton                      break;
1261de164aaaSGreg Clayton              }
1262de164aaaSGreg Clayton 
1263de164aaaSGreg Clayton              return error;
1264de164aaaSGreg Clayton          }
1265de164aaaSGreg Clayton 
1266de164aaaSGreg Clayton          void
1267de164aaaSGreg Clayton          OptionParsingStarting ()
1268de164aaaSGreg Clayton          {
1269de164aaaSGreg Clayton              m_help.clear();
1270de164aaaSGreg Clayton              m_syntax.clear();
1271de164aaaSGreg Clayton          }
1272de164aaaSGreg Clayton 
1273de164aaaSGreg Clayton          const OptionDefinition*
1274de164aaaSGreg Clayton          GetDefinitions ()
1275de164aaaSGreg Clayton          {
1276de164aaaSGreg Clayton              return g_option_table;
1277de164aaaSGreg Clayton          }
1278de164aaaSGreg Clayton 
1279de164aaaSGreg Clayton          // Options table: Required for subclasses of Options.
1280de164aaaSGreg Clayton 
1281de164aaaSGreg Clayton          static OptionDefinition g_option_table[];
1282de164aaaSGreg Clayton 
1283de164aaaSGreg Clayton          const char *
1284de164aaaSGreg Clayton          GetHelp ()
1285de164aaaSGreg Clayton          {
1286de164aaaSGreg Clayton              if (m_help.empty())
1287de164aaaSGreg Clayton                  return NULL;
1288de164aaaSGreg Clayton              return m_help.c_str();
1289de164aaaSGreg Clayton          }
1290de164aaaSGreg Clayton          const char *
1291de164aaaSGreg Clayton          GetSyntax ()
1292de164aaaSGreg Clayton          {
1293de164aaaSGreg Clayton              if (m_syntax.empty())
1294de164aaaSGreg Clayton                  return NULL;
1295de164aaaSGreg Clayton              return m_syntax.c_str();
1296de164aaaSGreg Clayton          }
1297de164aaaSGreg Clayton          // Instance variables to hold the values for command options.
1298de164aaaSGreg Clayton      protected:
1299de164aaaSGreg Clayton          std::string m_help;
1300de164aaaSGreg Clayton          std::string m_syntax;
1301de164aaaSGreg Clayton      };
1302de164aaaSGreg Clayton 
1303b0a1814fSEric Christopher      Options *
1304b0a1814fSEric Christopher      GetOptions () override
1305de164aaaSGreg Clayton      {
1306de164aaaSGreg Clayton          return &m_options;
1307de164aaaSGreg Clayton      }
1308de164aaaSGreg Clayton 
13095a988416SJim Ingham      CommandOptions m_options;
1310de164aaaSGreg Clayton };
1311de164aaaSGreg Clayton 
1312de164aaaSGreg Clayton OptionDefinition
1313de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] =
1314de164aaaSGreg Clayton {
1315d37221dcSZachary Turner { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeNone, "The help text to display for this command."},
1316d37221dcSZachary Turner { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."},
1317d37221dcSZachary Turner { 0             , false,  NULL   , 0  , 0                , NULL, NULL, 0, eArgTypeNone, NULL }
1318de164aaaSGreg Clayton };
1319de164aaaSGreg Clayton 
1320de164aaaSGreg Clayton 
13215a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw
1322223383edSEnrico Granata {
1323223383edSEnrico Granata private:
1324223383edSEnrico Granata     std::string m_function_name;
13250a305db7SEnrico Granata     ScriptedCommandSynchronicity m_synchro;
1326fac939e9SEnrico Granata     bool m_fetched_help_long;
1327223383edSEnrico Granata 
1328223383edSEnrico Granata public:
1329223383edSEnrico Granata 
1330223383edSEnrico Granata     CommandObjectPythonFunction (CommandInterpreter &interpreter,
1331223383edSEnrico Granata                                  std::string name,
13320a305db7SEnrico Granata                                  std::string funct,
1333735152e3SEnrico Granata                                  std::string help,
13340a305db7SEnrico Granata                                  ScriptedCommandSynchronicity synch) :
13355a988416SJim Ingham         CommandObjectRaw (interpreter,
1336223383edSEnrico Granata                           name.c_str(),
1337735152e3SEnrico Granata                           NULL,
1338223383edSEnrico Granata                           NULL),
13390a305db7SEnrico Granata         m_function_name(funct),
1340fac939e9SEnrico Granata         m_synchro(synch),
1341fac939e9SEnrico Granata         m_fetched_help_long(false)
1342223383edSEnrico Granata     {
1343735152e3SEnrico Granata         if (!help.empty())
1344735152e3SEnrico Granata             SetHelp(help.c_str());
1345735152e3SEnrico Granata         else
1346735152e3SEnrico Granata         {
1347735152e3SEnrico Granata             StreamString stream;
1348735152e3SEnrico Granata             stream.Printf("For more information run 'help %s'",name.c_str());
1349735152e3SEnrico Granata             SetHelp(stream.GetData());
1350735152e3SEnrico Granata         }
1351223383edSEnrico Granata     }
1352223383edSEnrico Granata 
1353223383edSEnrico Granata     virtual
1354223383edSEnrico Granata     ~CommandObjectPythonFunction ()
1355223383edSEnrico Granata     {
1356223383edSEnrico Granata     }
1357223383edSEnrico Granata 
1358223383edSEnrico Granata     virtual bool
13593a18e319SGreg Clayton     IsRemovable () const
13605a988416SJim Ingham     {
13615a988416SJim Ingham         return true;
13625a988416SJim Ingham     }
13635a988416SJim Ingham 
13645a988416SJim Ingham     const std::string&
13655a988416SJim Ingham     GetFunctionName ()
13665a988416SJim Ingham     {
13675a988416SJim Ingham         return m_function_name;
13685a988416SJim Ingham     }
13695a988416SJim Ingham 
13705a988416SJim Ingham     ScriptedCommandSynchronicity
13715a988416SJim Ingham     GetSynchronicity ()
13725a988416SJim Ingham     {
13735a988416SJim Ingham         return m_synchro;
13745a988416SJim Ingham     }
13755a988416SJim Ingham 
1376fac939e9SEnrico Granata     virtual const char *
1377fac939e9SEnrico Granata     GetHelpLong ()
1378fac939e9SEnrico Granata     {
1379fac939e9SEnrico Granata         if (!m_fetched_help_long)
1380fac939e9SEnrico Granata         {
1381fac939e9SEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1382fac939e9SEnrico Granata             if (scripter)
1383fac939e9SEnrico Granata             {
1384fac939e9SEnrico Granata                 std::string docstring;
1385fac939e9SEnrico Granata                 m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring);
1386fac939e9SEnrico Granata                 if (!docstring.empty())
1387fac939e9SEnrico Granata                     SetHelpLong(docstring);
1388fac939e9SEnrico Granata             }
1389fac939e9SEnrico Granata         }
1390fac939e9SEnrico Granata         return CommandObjectRaw::GetHelpLong();
1391fac939e9SEnrico Granata     }
1392fac939e9SEnrico Granata 
13935a988416SJim Ingham protected:
13945a988416SJim Ingham     virtual bool
13955a988416SJim Ingham     DoExecute (const char *raw_command_line, CommandReturnObject &result)
1396223383edSEnrico Granata     {
1397223383edSEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1398223383edSEnrico Granata 
1399223383edSEnrico Granata         Error error;
1400223383edSEnrico Granata 
140170f11f88SJim Ingham         result.SetStatus(eReturnStatusInvalid);
140270f11f88SJim Ingham 
1403223383edSEnrico Granata         if (!scripter || scripter->RunScriptBasedCommand(m_function_name.c_str(),
1404223383edSEnrico Granata                                                          raw_command_line,
14050a305db7SEnrico Granata                                                          m_synchro,
1406223383edSEnrico Granata                                                          result,
140706be059aSEnrico Granata                                                          error,
140806be059aSEnrico Granata                                                          m_exe_ctx) == false)
1409223383edSEnrico Granata         {
1410223383edSEnrico Granata             result.AppendError(error.AsCString());
1411223383edSEnrico Granata             result.SetStatus(eReturnStatusFailed);
1412223383edSEnrico Granata         }
1413223383edSEnrico Granata         else
141470f11f88SJim Ingham         {
141570f11f88SJim Ingham             // Don't change the status if the command already set it...
141670f11f88SJim Ingham             if (result.GetStatus() == eReturnStatusInvalid)
141770f11f88SJim Ingham             {
14189a71a7d8SDaniel Malea                 if (result.GetOutputData() == NULL || result.GetOutputData()[0] == '\0')
1419223383edSEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
142070f11f88SJim Ingham                 else
142170f11f88SJim Ingham                     result.SetStatus(eReturnStatusSuccessFinishResult);
142270f11f88SJim Ingham             }
142370f11f88SJim Ingham         }
1424223383edSEnrico Granata 
1425223383edSEnrico Granata         return result.Succeeded();
1426223383edSEnrico Granata     }
1427223383edSEnrico Granata 
1428223383edSEnrico Granata };
1429223383edSEnrico Granata 
14309fe00e52SEnrico Granata class CommandObjectScriptingObject : public CommandObjectRaw
14319fe00e52SEnrico Granata {
14329fe00e52SEnrico Granata private:
14330641ca1aSZachary Turner     StructuredData::GenericSP m_cmd_obj_sp;
14349fe00e52SEnrico Granata     ScriptedCommandSynchronicity m_synchro;
14356f79bb2dSEnrico Granata     bool m_fetched_help_short:1;
14366f79bb2dSEnrico Granata     bool m_fetched_help_long:1;
14379fe00e52SEnrico Granata 
14389fe00e52SEnrico Granata public:
14399fe00e52SEnrico Granata 
14409fe00e52SEnrico Granata     CommandObjectScriptingObject (CommandInterpreter &interpreter,
14419fe00e52SEnrico Granata                                   std::string name,
14420641ca1aSZachary Turner                                   StructuredData::GenericSP cmd_obj_sp,
14439fe00e52SEnrico Granata                                   ScriptedCommandSynchronicity synch) :
14449fe00e52SEnrico Granata     CommandObjectRaw (interpreter,
14459fe00e52SEnrico Granata                       name.c_str(),
14469fe00e52SEnrico Granata                       NULL,
14479fe00e52SEnrico Granata                       NULL),
14489fe00e52SEnrico Granata     m_cmd_obj_sp(cmd_obj_sp),
14496f79bb2dSEnrico Granata     m_synchro(synch),
14506f79bb2dSEnrico Granata     m_fetched_help_short(false),
14516f79bb2dSEnrico Granata     m_fetched_help_long(false)
14529fe00e52SEnrico Granata     {
14539fe00e52SEnrico Granata         StreamString stream;
14549fe00e52SEnrico Granata         stream.Printf("For more information run 'help %s'",name.c_str());
14559fe00e52SEnrico Granata         SetHelp(stream.GetData());
1456e87764f2SEnrico Granata         if (ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter())
1457e87764f2SEnrico Granata             GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
14589fe00e52SEnrico Granata     }
14599fe00e52SEnrico Granata 
14609fe00e52SEnrico Granata     virtual
14619fe00e52SEnrico Granata     ~CommandObjectScriptingObject ()
14629fe00e52SEnrico Granata     {
14639fe00e52SEnrico Granata     }
14649fe00e52SEnrico Granata 
14659fe00e52SEnrico Granata     virtual bool
14669fe00e52SEnrico Granata     IsRemovable () const
14679fe00e52SEnrico Granata     {
14689fe00e52SEnrico Granata         return true;
14699fe00e52SEnrico Granata     }
14709fe00e52SEnrico Granata 
14710641ca1aSZachary Turner     StructuredData::GenericSP
14729fe00e52SEnrico Granata     GetImplementingObject ()
14739fe00e52SEnrico Granata     {
14749fe00e52SEnrico Granata         return m_cmd_obj_sp;
14759fe00e52SEnrico Granata     }
14769fe00e52SEnrico Granata 
14779fe00e52SEnrico Granata     ScriptedCommandSynchronicity
14789fe00e52SEnrico Granata     GetSynchronicity ()
14799fe00e52SEnrico Granata     {
14809fe00e52SEnrico Granata         return m_synchro;
14819fe00e52SEnrico Granata     }
14829fe00e52SEnrico Granata 
14839fe00e52SEnrico Granata     virtual const char *
14846f79bb2dSEnrico Granata     GetHelp ()
14856f79bb2dSEnrico Granata     {
14866f79bb2dSEnrico Granata         if (!m_fetched_help_short)
14876f79bb2dSEnrico Granata         {
14886f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14896f79bb2dSEnrico Granata             if (scripter)
14906f79bb2dSEnrico Granata             {
14916f79bb2dSEnrico Granata                 std::string docstring;
14926f79bb2dSEnrico Granata                 m_fetched_help_short = scripter->GetShortHelpForCommandObject(m_cmd_obj_sp,docstring);
14936f79bb2dSEnrico Granata                 if (!docstring.empty())
14946f79bb2dSEnrico Granata                     SetHelp(docstring);
14956f79bb2dSEnrico Granata             }
14966f79bb2dSEnrico Granata         }
14976f79bb2dSEnrico Granata         return CommandObjectRaw::GetHelp();
14986f79bb2dSEnrico Granata     }
14996f79bb2dSEnrico Granata 
15006f79bb2dSEnrico Granata     virtual const char *
15019fe00e52SEnrico Granata     GetHelpLong ()
15029fe00e52SEnrico Granata     {
15036f79bb2dSEnrico Granata         if (!m_fetched_help_long)
15046f79bb2dSEnrico Granata         {
15056f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
15066f79bb2dSEnrico Granata             if (scripter)
15076f79bb2dSEnrico Granata             {
15086f79bb2dSEnrico Granata                 std::string docstring;
15096f79bb2dSEnrico Granata                 m_fetched_help_long = scripter->GetLongHelpForCommandObject(m_cmd_obj_sp,docstring);
15106f79bb2dSEnrico Granata                 if (!docstring.empty())
15116f79bb2dSEnrico Granata                     SetHelpLong(docstring);
15126f79bb2dSEnrico Granata             }
15136f79bb2dSEnrico Granata         }
15149fe00e52SEnrico Granata         return CommandObjectRaw::GetHelpLong();
15159fe00e52SEnrico Granata     }
15169fe00e52SEnrico Granata 
15179fe00e52SEnrico Granata protected:
15189fe00e52SEnrico Granata     virtual bool
15199fe00e52SEnrico Granata     DoExecute (const char *raw_command_line, CommandReturnObject &result)
15209fe00e52SEnrico Granata     {
15219fe00e52SEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
15229fe00e52SEnrico Granata 
15239fe00e52SEnrico Granata         Error error;
15249fe00e52SEnrico Granata 
15259fe00e52SEnrico Granata         result.SetStatus(eReturnStatusInvalid);
15269fe00e52SEnrico Granata 
15279fe00e52SEnrico Granata         if (!scripter || scripter->RunScriptBasedCommand(m_cmd_obj_sp,
15289fe00e52SEnrico Granata                                                          raw_command_line,
15299fe00e52SEnrico Granata                                                          m_synchro,
15309fe00e52SEnrico Granata                                                          result,
15319fe00e52SEnrico Granata                                                          error,
15329fe00e52SEnrico Granata                                                          m_exe_ctx) == false)
15339fe00e52SEnrico Granata         {
15349fe00e52SEnrico Granata             result.AppendError(error.AsCString());
15359fe00e52SEnrico Granata             result.SetStatus(eReturnStatusFailed);
15369fe00e52SEnrico Granata         }
15379fe00e52SEnrico Granata         else
15389fe00e52SEnrico Granata         {
15399fe00e52SEnrico Granata             // Don't change the status if the command already set it...
15409fe00e52SEnrico Granata             if (result.GetStatus() == eReturnStatusInvalid)
15419fe00e52SEnrico Granata             {
15429fe00e52SEnrico Granata                 if (result.GetOutputData() == NULL || result.GetOutputData()[0] == '\0')
15439fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
15449fe00e52SEnrico Granata                 else
15459fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishResult);
15469fe00e52SEnrico Granata             }
15479fe00e52SEnrico Granata         }
15489fe00e52SEnrico Granata 
15499fe00e52SEnrico Granata         return result.Succeeded();
15509fe00e52SEnrico Granata     }
15519fe00e52SEnrico Granata 
15529fe00e52SEnrico Granata };
15539fe00e52SEnrico Granata 
1554a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1555a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1556a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1557a9dbf432SEnrico Granata 
15585a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed
1559a9dbf432SEnrico Granata {
15605a988416SJim Ingham public:
15615a988416SJim Ingham     CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) :
15625a988416SJim Ingham         CommandObjectParsed (interpreter,
15635a988416SJim Ingham                              "command script import",
15645a988416SJim Ingham                              "Import a scripting module in LLDB.",
15655a988416SJim Ingham                              NULL),
15665a988416SJim Ingham         m_options(interpreter)
15675a988416SJim Ingham     {
15685a988416SJim Ingham         CommandArgumentEntry arg1;
15695a988416SJim Ingham         CommandArgumentData cmd_arg;
15705a988416SJim Ingham 
15715a988416SJim Ingham         // Define the first (and only) variant of this arg.
15725a988416SJim Ingham         cmd_arg.arg_type = eArgTypeFilename;
15733b00e35bSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlus;
15745a988416SJim Ingham 
15755a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
15765a988416SJim Ingham         arg1.push_back (cmd_arg);
15775a988416SJim Ingham 
15785a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
15795a988416SJim Ingham         m_arguments.push_back (arg1);
15805a988416SJim Ingham     }
15815a988416SJim Ingham 
15825a988416SJim Ingham     ~CommandObjectCommandsScriptImport ()
15835a988416SJim Ingham     {
15845a988416SJim Ingham     }
15855a988416SJim Ingham 
1586c7bece56SGreg Clayton     virtual int
15875a988416SJim Ingham     HandleArgumentCompletion (Args &input,
15885a988416SJim Ingham                               int &cursor_index,
15895a988416SJim Ingham                               int &cursor_char_position,
15905a988416SJim Ingham                               OptionElementVector &opt_element_vector,
15915a988416SJim Ingham                               int match_start_point,
15925a988416SJim Ingham                               int max_return_elements,
15935a988416SJim Ingham                               bool &word_complete,
15945a988416SJim Ingham                               StringList &matches)
15955a988416SJim Ingham     {
15965a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
15975a988416SJim Ingham         completion_str.erase (cursor_char_position);
15985a988416SJim Ingham 
15995a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
16005a988416SJim Ingham                                                              CommandCompletions::eDiskFileCompletion,
16015a988416SJim Ingham                                                              completion_str.c_str(),
16025a988416SJim Ingham                                                              match_start_point,
16035a988416SJim Ingham                                                              max_return_elements,
16045a988416SJim Ingham                                                              NULL,
16055a988416SJim Ingham                                                              word_complete,
16065a988416SJim Ingham                                                              matches);
16075a988416SJim Ingham         return matches.GetSize();
16085a988416SJim Ingham     }
16095a988416SJim Ingham 
16105a988416SJim Ingham     virtual Options *
16115a988416SJim Ingham     GetOptions ()
16125a988416SJim Ingham     {
16135a988416SJim Ingham         return &m_options;
16145a988416SJim Ingham     }
16155a988416SJim Ingham 
16165a988416SJim Ingham protected:
16170a305db7SEnrico Granata 
16180a305db7SEnrico Granata     class CommandOptions : public Options
16190a305db7SEnrico Granata     {
16200a305db7SEnrico Granata     public:
16210a305db7SEnrico Granata 
16220a305db7SEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
16230a305db7SEnrico Granata             Options (interpreter)
16240a305db7SEnrico Granata         {
16250a305db7SEnrico Granata         }
16260a305db7SEnrico Granata 
16270a305db7SEnrico Granata         virtual
16280a305db7SEnrico Granata         ~CommandOptions (){}
16290a305db7SEnrico Granata 
16300a305db7SEnrico Granata         virtual Error
16310a305db7SEnrico Granata         SetOptionValue (uint32_t option_idx, const char *option_arg)
16320a305db7SEnrico Granata         {
16330a305db7SEnrico Granata             Error error;
16343bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
16350a305db7SEnrico Granata 
16360a305db7SEnrico Granata             switch (short_option)
16370a305db7SEnrico Granata             {
16380a305db7SEnrico Granata                 case 'r':
16390a305db7SEnrico Granata                     m_allow_reload = true;
16400a305db7SEnrico Granata                     break;
16410a305db7SEnrico Granata                 default:
16420a305db7SEnrico Granata                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
16430a305db7SEnrico Granata                     break;
16440a305db7SEnrico Granata             }
16450a305db7SEnrico Granata 
16460a305db7SEnrico Granata             return error;
16470a305db7SEnrico Granata         }
16480a305db7SEnrico Granata 
16490a305db7SEnrico Granata         void
16500a305db7SEnrico Granata         OptionParsingStarting ()
16510a305db7SEnrico Granata         {
1652e0c70f1bSEnrico Granata             m_allow_reload = true;
16530a305db7SEnrico Granata         }
16540a305db7SEnrico Granata 
16550a305db7SEnrico Granata         const OptionDefinition*
16560a305db7SEnrico Granata         GetDefinitions ()
16570a305db7SEnrico Granata         {
16580a305db7SEnrico Granata             return g_option_table;
16590a305db7SEnrico Granata         }
16600a305db7SEnrico Granata 
16610a305db7SEnrico Granata         // Options table: Required for subclasses of Options.
16620a305db7SEnrico Granata 
16630a305db7SEnrico Granata         static OptionDefinition g_option_table[];
16640a305db7SEnrico Granata 
16650a305db7SEnrico Granata         // Instance variables to hold the values for command options.
16660a305db7SEnrico Granata 
16670a305db7SEnrico Granata         bool m_allow_reload;
16680a305db7SEnrico Granata     };
16690a305db7SEnrico Granata 
1670a9dbf432SEnrico Granata     bool
16715a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1672a9dbf432SEnrico Granata     {
1673a9dbf432SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
1674a9dbf432SEnrico Granata         {
1675a9dbf432SEnrico Granata             result.AppendError ("only scripting language supported for module importing is currently Python");
1676a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1677a9dbf432SEnrico Granata             return false;
1678a9dbf432SEnrico Granata         }
1679a9dbf432SEnrico Granata 
16805a988416SJim Ingham         size_t argc = command.GetArgumentCount();
16813b00e35bSEnrico Granata         if (0 == argc)
1682a9dbf432SEnrico Granata         {
16833b00e35bSEnrico Granata             result.AppendError("command script import needs one or more arguments");
1684a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1685a9dbf432SEnrico Granata             return false;
1686a9dbf432SEnrico Granata         }
1687a9dbf432SEnrico Granata 
1688*0e978481SEd Maste         for (size_t i = 0;
16893b00e35bSEnrico Granata              i < argc;
16903b00e35bSEnrico Granata              i++)
16913b00e35bSEnrico Granata         {
16923b00e35bSEnrico Granata             std::string path = command.GetArgumentAtIndex(i);
1693a9dbf432SEnrico Granata             Error error;
1694a9dbf432SEnrico Granata 
1695c9d645d3SGreg Clayton             const bool init_session = true;
1696078551c7SEnrico Granata             // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that
1697078551c7SEnrico Granata             // commands won't ever be recursively invoked, but it's actually possible to craft
1698078551c7SEnrico Granata             // a Python script that does other "command script imports" in __lldb_init_module
1699078551c7SEnrico Granata             // the real fix is to have recursive commands possible with a CommandInvocation object
1700078551c7SEnrico Granata             // separate from the CommandObject itself, so that recursive command invocations
1701078551c7SEnrico Granata             // won't stomp on each other (wrt to execution contents, options, and more)
1702078551c7SEnrico Granata             m_exe_ctx.Clear();
1703a9dbf432SEnrico Granata             if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(),
17040a305db7SEnrico Granata                                                                           m_options.m_allow_reload,
1705c9d645d3SGreg Clayton                                                                           init_session,
1706a9dbf432SEnrico Granata                                                                           error))
1707a9dbf432SEnrico Granata             {
1708a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
1709a9dbf432SEnrico Granata             }
1710a9dbf432SEnrico Granata             else
1711a9dbf432SEnrico Granata             {
1712a9dbf432SEnrico Granata                 result.AppendErrorWithFormat("module importing failed: %s", error.AsCString());
1713a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1714a9dbf432SEnrico Granata             }
17153b00e35bSEnrico Granata         }
1716a9dbf432SEnrico Granata 
1717a9dbf432SEnrico Granata         return result.Succeeded();
1718a9dbf432SEnrico Granata     }
17190a305db7SEnrico Granata 
17205a988416SJim Ingham     CommandOptions m_options;
1721a9dbf432SEnrico Granata };
1722223383edSEnrico Granata 
17230a305db7SEnrico Granata OptionDefinition
17240a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] =
17250a305db7SEnrico Granata {
1726d37221dcSZachary Turner     { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, NULL, NULL, 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."},
1727d37221dcSZachary Turner     { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
17280a305db7SEnrico Granata };
17290a305db7SEnrico Granata 
17300a305db7SEnrico Granata 
1731223383edSEnrico Granata //-------------------------------------------------------------------------
1732223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1733223383edSEnrico Granata //-------------------------------------------------------------------------
1734223383edSEnrico Granata 
173544d93782SGreg Clayton class CommandObjectCommandsScriptAdd :
173644d93782SGreg Clayton     public CommandObjectParsed,
173744d93782SGreg Clayton     public IOHandlerDelegateMultiline
1738223383edSEnrico Granata {
17395a988416SJim Ingham public:
17405a988416SJim Ingham     CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) :
17415a988416SJim Ingham         CommandObjectParsed (interpreter,
17425a988416SJim Ingham                              "command script add",
17435a988416SJim Ingham                              "Add a scripted function as an LLDB command.",
17445a988416SJim Ingham                              NULL),
1745c3d874a5SGreg Clayton         IOHandlerDelegateMultiline ("DONE"),
17465a988416SJim Ingham         m_options (interpreter)
17475a988416SJim Ingham     {
17485a988416SJim Ingham         CommandArgumentEntry arg1;
17495a988416SJim Ingham         CommandArgumentData cmd_arg;
17505a988416SJim Ingham 
17515a988416SJim Ingham         // Define the first (and only) variant of this arg.
17525a988416SJim Ingham         cmd_arg.arg_type = eArgTypeCommandName;
17535a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
17545a988416SJim Ingham 
17555a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
17565a988416SJim Ingham         arg1.push_back (cmd_arg);
17575a988416SJim Ingham 
17585a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
17595a988416SJim Ingham         m_arguments.push_back (arg1);
17605a988416SJim Ingham     }
17615a988416SJim Ingham 
17625a988416SJim Ingham     ~CommandObjectCommandsScriptAdd ()
17635a988416SJim Ingham     {
17645a988416SJim Ingham     }
17655a988416SJim Ingham 
17665a988416SJim Ingham     virtual Options *
17675a988416SJim Ingham     GetOptions ()
17685a988416SJim Ingham     {
17695a988416SJim Ingham         return &m_options;
17705a988416SJim Ingham     }
17715a988416SJim Ingham 
17725a988416SJim Ingham protected:
1773223383edSEnrico Granata 
1774223383edSEnrico Granata     class CommandOptions : public Options
1775223383edSEnrico Granata     {
1776223383edSEnrico Granata     public:
1777223383edSEnrico Granata 
1778223383edSEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
17799fe00e52SEnrico Granata             Options (interpreter),
17809fe00e52SEnrico Granata             m_class_name(),
17819fe00e52SEnrico Granata             m_funct_name(),
17829fe00e52SEnrico Granata             m_short_help(),
17839fe00e52SEnrico Granata             m_synchronicity(eScriptedCommandSynchronicitySynchronous)
1784223383edSEnrico Granata         {
1785223383edSEnrico Granata         }
1786223383edSEnrico Granata 
1787223383edSEnrico Granata         virtual
1788223383edSEnrico Granata         ~CommandOptions (){}
1789223383edSEnrico Granata 
1790223383edSEnrico Granata         virtual Error
1791223383edSEnrico Granata         SetOptionValue (uint32_t option_idx, const char *option_arg)
1792223383edSEnrico Granata         {
1793223383edSEnrico Granata             Error error;
17943bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
1795223383edSEnrico Granata 
1796223383edSEnrico Granata             switch (short_option)
1797223383edSEnrico Granata             {
1798223383edSEnrico Granata                 case 'f':
1799735152e3SEnrico Granata                     if (option_arg)
1800735152e3SEnrico Granata                         m_funct_name.assign(option_arg);
1801735152e3SEnrico Granata                     break;
18029fe00e52SEnrico Granata                 case 'c':
18039fe00e52SEnrico Granata                     if (option_arg)
18049fe00e52SEnrico Granata                         m_class_name.assign(option_arg);
18059fe00e52SEnrico Granata                     break;
1806735152e3SEnrico Granata                 case 'h':
1807735152e3SEnrico Granata                     if (option_arg)
1808735152e3SEnrico Granata                         m_short_help.assign(option_arg);
1809223383edSEnrico Granata                     break;
18100a305db7SEnrico Granata                 case 's':
181144d93782SGreg Clayton                     m_synchronicity = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error);
18120a305db7SEnrico Granata                     if (!error.Success())
18130a305db7SEnrico Granata                         error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg);
18140a305db7SEnrico Granata                     break;
1815223383edSEnrico Granata                 default:
181686edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1817223383edSEnrico Granata                     break;
1818223383edSEnrico Granata             }
1819223383edSEnrico Granata 
1820223383edSEnrico Granata             return error;
1821223383edSEnrico Granata         }
1822223383edSEnrico Granata 
1823223383edSEnrico Granata         void
1824223383edSEnrico Granata         OptionParsingStarting ()
1825223383edSEnrico Granata         {
18269fe00e52SEnrico Granata             m_class_name.clear();
1827735152e3SEnrico Granata             m_funct_name.clear();
1828735152e3SEnrico Granata             m_short_help.clear();
182944d93782SGreg Clayton             m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1830223383edSEnrico Granata         }
1831223383edSEnrico Granata 
1832223383edSEnrico Granata         const OptionDefinition*
1833223383edSEnrico Granata         GetDefinitions ()
1834223383edSEnrico Granata         {
1835223383edSEnrico Granata             return g_option_table;
1836223383edSEnrico Granata         }
1837223383edSEnrico Granata 
1838223383edSEnrico Granata         // Options table: Required for subclasses of Options.
1839223383edSEnrico Granata 
1840223383edSEnrico Granata         static OptionDefinition g_option_table[];
1841223383edSEnrico Granata 
1842223383edSEnrico Granata         // Instance variables to hold the values for command options.
1843223383edSEnrico Granata 
18449fe00e52SEnrico Granata         std::string m_class_name;
1845223383edSEnrico Granata         std::string m_funct_name;
1846735152e3SEnrico Granata         std::string m_short_help;
184744d93782SGreg Clayton         ScriptedCommandSynchronicity m_synchronicity;
1848223383edSEnrico Granata     };
1849223383edSEnrico Granata 
185044d93782SGreg Clayton     virtual void
185144d93782SGreg Clayton     IOHandlerActivated (IOHandler &io_handler)
1852223383edSEnrico Granata     {
185344d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
185444d93782SGreg Clayton         if (output_sp)
1855223383edSEnrico Granata         {
185644d93782SGreg Clayton             output_sp->PutCString(g_python_command_instructions);
185744d93782SGreg Clayton             output_sp->Flush();
1858223383edSEnrico Granata         }
1859223383edSEnrico Granata     }
1860223383edSEnrico Granata 
1861223383edSEnrico Granata 
186244d93782SGreg Clayton     virtual void
186344d93782SGreg Clayton     IOHandlerInputComplete (IOHandler &io_handler, std::string &data)
1864223383edSEnrico Granata     {
186544d93782SGreg Clayton         StreamFileSP error_sp = io_handler.GetErrorStreamFile();
186644d93782SGreg Clayton 
186744d93782SGreg Clayton         ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter();
186844d93782SGreg Clayton         if (interpreter)
186944d93782SGreg Clayton         {
187044d93782SGreg Clayton 
187144d93782SGreg Clayton             StringList lines;
187244d93782SGreg Clayton             lines.SplitIntoLines(data);
187344d93782SGreg Clayton             if (lines.GetSize() > 0)
187444d93782SGreg Clayton             {
1875a73b7df7SEnrico Granata                 std::string funct_name_str;
187644d93782SGreg Clayton                 if (interpreter->GenerateScriptAliasFunction (lines, funct_name_str))
1877223383edSEnrico Granata                 {
1878a73b7df7SEnrico Granata                     if (funct_name_str.empty())
1879223383edSEnrico Granata                     {
188044d93782SGreg Clayton                         error_sp->Printf ("error: unable to obtain a function name, didn't add python command.\n");
188144d93782SGreg Clayton                         error_sp->Flush();
1882223383edSEnrico Granata                     }
188344d93782SGreg Clayton                     else
188444d93782SGreg Clayton                     {
1885223383edSEnrico Granata                         // everything should be fine now, let's add this alias
1886223383edSEnrico Granata 
1887223383edSEnrico Granata                         CommandObjectSP command_obj_sp(new CommandObjectPythonFunction (m_interpreter,
1888223383edSEnrico Granata                                                                                         m_cmd_name,
1889a73b7df7SEnrico Granata                                                                                         funct_name_str.c_str(),
1890735152e3SEnrico Granata                                                                                         m_short_help,
189144d93782SGreg Clayton                                                                                         m_synchronicity));
1892223383edSEnrico Granata 
18930a305db7SEnrico Granata                         if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true))
1894223383edSEnrico Granata                         {
189544d93782SGreg Clayton                             error_sp->Printf ("error: unable to add selected command, didn't add python command.\n");
189644d93782SGreg Clayton                             error_sp->Flush();
1897223383edSEnrico Granata                         }
1898223383edSEnrico Granata                     }
189944d93782SGreg Clayton                 }
190044d93782SGreg Clayton                 else
190144d93782SGreg Clayton                 {
190244d93782SGreg Clayton                     error_sp->Printf ("error: unable to create function, didn't add python command.\n");
190344d93782SGreg Clayton                     error_sp->Flush();
190444d93782SGreg Clayton                 }
190544d93782SGreg Clayton             }
190644d93782SGreg Clayton             else
190744d93782SGreg Clayton             {
190844d93782SGreg Clayton                 error_sp->Printf ("error: empty function, didn't add python command.\n");
190944d93782SGreg Clayton                 error_sp->Flush();
191044d93782SGreg Clayton             }
191144d93782SGreg Clayton         }
191244d93782SGreg Clayton         else
191344d93782SGreg Clayton         {
191444d93782SGreg Clayton             error_sp->Printf ("error: script interpreter missing, didn't add python command.\n");
191544d93782SGreg Clayton             error_sp->Flush();
191644d93782SGreg Clayton         }
191744d93782SGreg Clayton 
191844d93782SGreg Clayton         io_handler.SetIsDone(true);
191944d93782SGreg Clayton 
192044d93782SGreg Clayton 
192144d93782SGreg Clayton     }
1922223383edSEnrico Granata 
19235a988416SJim Ingham protected:
1924223383edSEnrico Granata     bool
19255a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1926223383edSEnrico Granata     {
192799f0b8f9SEnrico Granata 
192899f0b8f9SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
192999f0b8f9SEnrico Granata         {
193099f0b8f9SEnrico Granata             result.AppendError ("only scripting language supported for scripted commands is currently Python");
193199f0b8f9SEnrico Granata             result.SetStatus (eReturnStatusFailed);
193299f0b8f9SEnrico Granata             return false;
193399f0b8f9SEnrico Granata         }
193499f0b8f9SEnrico Granata 
19355a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1936223383edSEnrico Granata 
1937223383edSEnrico Granata         if (argc != 1)
1938223383edSEnrico Granata         {
1939223383edSEnrico Granata             result.AppendError ("'command script add' requires one argument");
1940223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1941223383edSEnrico Granata             return false;
1942223383edSEnrico Granata         }
1943223383edSEnrico Granata 
1944735152e3SEnrico Granata         // Store the options in case we get multi-line input
194544d93782SGreg Clayton         m_cmd_name = command.GetArgumentAtIndex(0);
1946735152e3SEnrico Granata         m_short_help.assign(m_options.m_short_help);
194744d93782SGreg Clayton         m_synchronicity = m_options.m_synchronicity;
1948223383edSEnrico Granata 
19499fe00e52SEnrico Granata         if (m_options.m_class_name.empty())
19509fe00e52SEnrico Granata         {
1951223383edSEnrico Granata             if (m_options.m_funct_name.empty())
1952223383edSEnrico Granata             {
195344d93782SGreg Clayton                 m_interpreter.GetPythonCommandsFromIOHandler ("     ",  // Prompt
195444d93782SGreg Clayton                                                               *this,    // IOHandlerDelegate
195544d93782SGreg Clayton                                                               true,     // Run IOHandler in async mode
195644d93782SGreg Clayton                                                               NULL);    // Baton for the "io_handler" that will be passed back into our IOHandlerDelegate functions
1957223383edSEnrico Granata             }
1958223383edSEnrico Granata             else
1959223383edSEnrico Granata             {
19600a305db7SEnrico Granata                 CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter,
196144d93782SGreg Clayton                                                                         m_cmd_name,
19620a305db7SEnrico Granata                                                                         m_options.m_funct_name,
1963735152e3SEnrico Granata                                                                         m_options.m_short_help,
196444d93782SGreg Clayton                                                                         m_synchronicity));
196544d93782SGreg Clayton                 if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
1966223383edSEnrico Granata                 {
1967223383edSEnrico Granata                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1968223383edSEnrico Granata                 }
1969223383edSEnrico Granata                 else
1970223383edSEnrico Granata                 {
1971223383edSEnrico Granata                     result.AppendError("cannot add command");
1972223383edSEnrico Granata                     result.SetStatus (eReturnStatusFailed);
1973223383edSEnrico Granata                 }
1974223383edSEnrico Granata             }
19759fe00e52SEnrico Granata         }
19769fe00e52SEnrico Granata         else
19779fe00e52SEnrico Granata         {
19789fe00e52SEnrico Granata             ScriptInterpreter *interpreter = GetCommandInterpreter().GetScriptInterpreter();
19799fe00e52SEnrico Granata             if (!interpreter)
19809fe00e52SEnrico Granata             {
19819fe00e52SEnrico Granata                 result.AppendError("cannot find ScriptInterpreter");
19829fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19839fe00e52SEnrico Granata                 return false;
19849fe00e52SEnrico Granata             }
19859fe00e52SEnrico Granata 
19869fe00e52SEnrico Granata             auto cmd_obj_sp = interpreter->CreateScriptCommandObject(m_options.m_class_name.c_str());
19879fe00e52SEnrico Granata             if (!cmd_obj_sp)
19889fe00e52SEnrico Granata             {
19899fe00e52SEnrico Granata                 result.AppendError("cannot create helper object");
19909fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19919fe00e52SEnrico Granata                 return false;
19929fe00e52SEnrico Granata             }
19939fe00e52SEnrico Granata 
19949fe00e52SEnrico Granata             CommandObjectSP new_cmd(new CommandObjectScriptingObject(m_interpreter,
19959fe00e52SEnrico Granata                                                                      m_cmd_name,
19969fe00e52SEnrico Granata                                                                      cmd_obj_sp,
19979fe00e52SEnrico Granata                                                                      m_synchronicity));
19989fe00e52SEnrico Granata             if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
19999fe00e52SEnrico Granata             {
20009fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
20019fe00e52SEnrico Granata             }
20029fe00e52SEnrico Granata             else
20039fe00e52SEnrico Granata             {
20049fe00e52SEnrico Granata                 result.AppendError("cannot add command");
20059fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
20069fe00e52SEnrico Granata             }
20079fe00e52SEnrico Granata         }
2008223383edSEnrico Granata 
2009223383edSEnrico Granata         return result.Succeeded();
2010223383edSEnrico Granata 
2011223383edSEnrico Granata     }
20125a988416SJim Ingham 
20135a988416SJim Ingham     CommandOptions m_options;
201444d93782SGreg Clayton     std::string m_cmd_name;
2015735152e3SEnrico Granata     std::string m_short_help;
201644d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
2017223383edSEnrico Granata };
2018223383edSEnrico Granata 
20190a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] =
20200a305db7SEnrico Granata {
20210a305db7SEnrico Granata     { eScriptedCommandSynchronicitySynchronous,      "synchronous",       "Run synchronous"},
20220a305db7SEnrico Granata     { eScriptedCommandSynchronicityAsynchronous,     "asynchronous",      "Run asynchronous"},
20230a305db7SEnrico Granata     { eScriptedCommandSynchronicityCurrentValue,     "current",           "Do not alter current setting"},
20240a305db7SEnrico Granata     { 0, NULL, NULL }
20250a305db7SEnrico Granata };
20260a305db7SEnrico Granata 
2027223383edSEnrico Granata OptionDefinition
2028223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] =
2029223383edSEnrico Granata {
2030d37221dcSZachary Turner     { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypePythonFunction,        "Name of the Python function to bind to this command name."},
20319fe00e52SEnrico Granata     { LLDB_OPT_SET_2, false, "class", 'c', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypePythonClass,        "Name of the Python class to bind to this command name."},
20329370d278SEnrico Granata     { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeHelpText, "The help text to display for this command."},
20339fe00e52SEnrico Granata     { LLDB_OPT_SET_ALL, false, "synchronicity", 's', OptionParser::eRequiredArgument, NULL, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity,        "Set the synchronicity of this command's executions with regard to LLDB event system."},
2034d37221dcSZachary Turner     { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
2035223383edSEnrico Granata };
2036223383edSEnrico Granata 
2037223383edSEnrico Granata //-------------------------------------------------------------------------
2038223383edSEnrico Granata // CommandObjectCommandsScriptList
2039223383edSEnrico Granata //-------------------------------------------------------------------------
2040223383edSEnrico Granata 
20415a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed
2042223383edSEnrico Granata {
2043223383edSEnrico Granata private:
2044223383edSEnrico Granata 
2045223383edSEnrico Granata public:
2046223383edSEnrico Granata     CommandObjectCommandsScriptList(CommandInterpreter &interpreter) :
20475a988416SJim Ingham     CommandObjectParsed (interpreter,
2048223383edSEnrico Granata                    "command script list",
2049223383edSEnrico Granata                    "List defined scripted commands.",
2050223383edSEnrico Granata                    NULL)
2051223383edSEnrico Granata     {
2052223383edSEnrico Granata     }
2053223383edSEnrico Granata 
2054223383edSEnrico Granata     ~CommandObjectCommandsScriptList ()
2055223383edSEnrico Granata     {
2056223383edSEnrico Granata     }
2057223383edSEnrico Granata 
2058223383edSEnrico Granata     bool
20595a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
2060223383edSEnrico Granata     {
2061223383edSEnrico Granata 
2062223383edSEnrico Granata         m_interpreter.GetHelp(result,
2063223383edSEnrico Granata                               CommandInterpreter::eCommandTypesUserDef);
2064223383edSEnrico Granata 
2065223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2066223383edSEnrico Granata 
2067223383edSEnrico Granata         return true;
2068223383edSEnrico Granata 
2069223383edSEnrico Granata 
2070223383edSEnrico Granata     }
2071223383edSEnrico Granata };
2072223383edSEnrico Granata 
2073223383edSEnrico Granata //-------------------------------------------------------------------------
2074223383edSEnrico Granata // CommandObjectCommandsScriptClear
2075223383edSEnrico Granata //-------------------------------------------------------------------------
2076223383edSEnrico Granata 
20775a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed
2078223383edSEnrico Granata {
2079223383edSEnrico Granata private:
2080223383edSEnrico Granata 
2081223383edSEnrico Granata public:
2082223383edSEnrico Granata     CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) :
20835a988416SJim Ingham         CommandObjectParsed (interpreter,
2084223383edSEnrico Granata                              "command script clear",
2085223383edSEnrico Granata                              "Delete all scripted commands.",
2086223383edSEnrico Granata                              NULL)
2087223383edSEnrico Granata     {
2088223383edSEnrico Granata     }
2089223383edSEnrico Granata 
2090223383edSEnrico Granata     ~CommandObjectCommandsScriptClear ()
2091223383edSEnrico Granata     {
2092223383edSEnrico Granata     }
2093223383edSEnrico Granata 
20945a988416SJim Ingham protected:
2095223383edSEnrico Granata     bool
20965a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
2097223383edSEnrico Granata     {
2098223383edSEnrico Granata 
2099223383edSEnrico Granata         m_interpreter.RemoveAllUser();
2100223383edSEnrico Granata 
2101223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2102223383edSEnrico Granata 
2103223383edSEnrico Granata         return true;
2104223383edSEnrico Granata     }
2105223383edSEnrico Granata };
2106223383edSEnrico Granata 
2107223383edSEnrico Granata //-------------------------------------------------------------------------
2108223383edSEnrico Granata // CommandObjectCommandsScriptDelete
2109223383edSEnrico Granata //-------------------------------------------------------------------------
2110223383edSEnrico Granata 
21115a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed
2112223383edSEnrico Granata {
2113223383edSEnrico Granata public:
2114223383edSEnrico Granata     CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) :
21155a988416SJim Ingham         CommandObjectParsed (interpreter,
2116223383edSEnrico Granata                              "command script delete",
2117223383edSEnrico Granata                              "Delete a scripted command.",
2118223383edSEnrico Granata                              NULL)
2119223383edSEnrico Granata     {
2120223383edSEnrico Granata         CommandArgumentEntry arg1;
2121223383edSEnrico Granata         CommandArgumentData cmd_arg;
2122223383edSEnrico Granata 
2123223383edSEnrico Granata         // Define the first (and only) variant of this arg.
2124223383edSEnrico Granata         cmd_arg.arg_type = eArgTypeCommandName;
2125223383edSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlain;
2126223383edSEnrico Granata 
2127223383edSEnrico Granata         // There is only one variant this argument could be; put it into the argument entry.
2128223383edSEnrico Granata         arg1.push_back (cmd_arg);
2129223383edSEnrico Granata 
2130223383edSEnrico Granata         // Push the data for the first argument into the m_arguments vector.
2131223383edSEnrico Granata         m_arguments.push_back (arg1);
2132223383edSEnrico Granata     }
2133223383edSEnrico Granata 
2134223383edSEnrico Granata     ~CommandObjectCommandsScriptDelete ()
2135223383edSEnrico Granata     {
2136223383edSEnrico Granata     }
2137223383edSEnrico Granata 
21385a988416SJim Ingham protected:
2139223383edSEnrico Granata     bool
21405a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
2141223383edSEnrico Granata     {
2142223383edSEnrico Granata 
21435a988416SJim Ingham         size_t argc = command.GetArgumentCount();
2144223383edSEnrico Granata 
2145223383edSEnrico Granata         if (argc != 1)
2146223383edSEnrico Granata         {
2147223383edSEnrico Granata             result.AppendError ("'command script delete' requires one argument");
2148223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2149223383edSEnrico Granata             return false;
2150223383edSEnrico Granata         }
2151223383edSEnrico Granata 
21525a988416SJim Ingham         const char* cmd_name = command.GetArgumentAtIndex(0);
2153223383edSEnrico Granata 
2154223383edSEnrico Granata         if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name))
2155223383edSEnrico Granata         {
2156223383edSEnrico Granata             m_interpreter.RemoveUser(cmd_name);
2157223383edSEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishResult);
2158223383edSEnrico Granata         }
2159223383edSEnrico Granata         else
2160223383edSEnrico Granata         {
2161223383edSEnrico Granata             result.AppendErrorWithFormat ("command %s not found", cmd_name);
2162223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2163223383edSEnrico Granata         }
2164223383edSEnrico Granata 
2165223383edSEnrico Granata         return result.Succeeded();
2166223383edSEnrico Granata 
2167223383edSEnrico Granata     }
2168223383edSEnrico Granata };
2169223383edSEnrico Granata 
2170223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
2171223383edSEnrico Granata 
2172223383edSEnrico Granata //-------------------------------------------------------------------------
2173223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
2174223383edSEnrico Granata //-------------------------------------------------------------------------
2175223383edSEnrico Granata 
2176223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword
2177223383edSEnrico Granata {
2178223383edSEnrico Granata public:
2179223383edSEnrico Granata     CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) :
2180223383edSEnrico Granata     CommandObjectMultiword (interpreter,
2181223383edSEnrico Granata                             "command script",
2182223383edSEnrico Granata                             "A set of commands for managing or customizing script commands.",
2183223383edSEnrico Granata                             "command script <subcommand> [<subcommand-options>]")
2184223383edSEnrico Granata     {
2185223383edSEnrico Granata         LoadSubCommand ("add",    CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter)));
2186223383edSEnrico Granata         LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter)));
2187223383edSEnrico Granata         LoadSubCommand ("clear",  CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter)));
2188223383edSEnrico Granata         LoadSubCommand ("list",   CommandObjectSP (new CommandObjectCommandsScriptList (interpreter)));
2189a9dbf432SEnrico Granata         LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter)));
2190223383edSEnrico Granata     }
2191223383edSEnrico Granata 
2192223383edSEnrico Granata     ~CommandObjectMultiwordCommandsScript ()
2193223383edSEnrico Granata     {
2194223383edSEnrico Granata     }
2195223383edSEnrico Granata 
2196223383edSEnrico Granata };
2197223383edSEnrico Granata 
2198223383edSEnrico Granata 
2199ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
2200ebc09c36SJim Ingham 
2201ebc09c36SJim Ingham //-------------------------------------------------------------------------
2202ebc09c36SJim Ingham // CommandObjectMultiwordCommands
2203ebc09c36SJim Ingham //-------------------------------------------------------------------------
2204ebc09c36SJim Ingham 
2205ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) :
2206a7015092SGreg Clayton     CommandObjectMultiword (interpreter,
22070e5e5a79SGreg Clayton                             "command",
22083f4c09c1SCaroline Tice                             "A set of commands for managing or customizing the debugger commands.",
22090e5e5a79SGreg Clayton                             "command <subcommand> [<subcommand-options>]")
2210ebc09c36SJim Ingham {
2211a7015092SGreg Clayton     LoadSubCommand ("source",  CommandObjectSP (new CommandObjectCommandsSource (interpreter)));
2212a7015092SGreg Clayton     LoadSubCommand ("alias",   CommandObjectSP (new CommandObjectCommandsAlias (interpreter)));
2213a7015092SGreg Clayton     LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter)));
2214b547278cSGreg Clayton     LoadSubCommand ("delete",  CommandObjectSP (new CommandObjectCommandsDelete (interpreter)));
2215de164aaaSGreg Clayton     LoadSubCommand ("regex",   CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter)));
2216a5a97ebeSJim Ingham     LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter)));
2217223383edSEnrico Granata     LoadSubCommand ("script",  CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter)));
2218ebc09c36SJim Ingham }
2219ebc09c36SJim Ingham 
2220ebc09c36SJim Ingham CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands ()
2221ebc09c36SJim Ingham {
2222ebc09c36SJim Ingham }
2223ebc09c36SJim Ingham 
2224