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 
1093a64300SDaniel Malea #include "lldb/lldb-python.h"
1193a64300SDaniel Malea 
12ebc09c36SJim Ingham #include "CommandObjectCommands.h"
13ebc09c36SJim Ingham 
14ebc09c36SJim Ingham // C Includes
15ebc09c36SJim Ingham // C++ Includes
16ebc09c36SJim Ingham // Other libraries and framework includes
170e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h"
180e5e5a79SGreg Clayton 
19ebc09c36SJim Ingham // Project includes
20ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
21de164aaaSGreg Clayton #include "lldb/Core/InputReader.h"
22be93a35aSEnrico Granata #include "lldb/Core/InputReaderEZ.h"
23be93a35aSEnrico Granata #include "lldb/Core/StringList.h"
24de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h"
257594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
26ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
27de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h"
28ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
29012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
307594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
31ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
3299f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
3399f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreterPython.h"
34ebc09c36SJim Ingham 
35ebc09c36SJim Ingham using namespace lldb;
36ebc09c36SJim Ingham using namespace lldb_private;
37ebc09c36SJim Ingham 
38ebc09c36SJim Ingham //-------------------------------------------------------------------------
39ebc09c36SJim Ingham // CommandObjectCommandsSource
40ebc09c36SJim Ingham //-------------------------------------------------------------------------
41ebc09c36SJim Ingham 
425a988416SJim Ingham class CommandObjectCommandsHistory : public CommandObjectParsed
43a5a97ebeSJim Ingham {
445a988416SJim Ingham public:
455a988416SJim Ingham     CommandObjectCommandsHistory(CommandInterpreter &interpreter) :
465a988416SJim Ingham         CommandObjectParsed (interpreter,
475a988416SJim Ingham                              "command history",
485a988416SJim Ingham                              "Dump the history of commands in this session.",
495a988416SJim Ingham                              NULL),
505a988416SJim Ingham         m_options (interpreter)
515a988416SJim Ingham     {
525a988416SJim Ingham     }
535a988416SJim Ingham 
545a988416SJim Ingham     ~CommandObjectCommandsHistory () {}
555a988416SJim Ingham 
565a988416SJim Ingham     virtual Options *
575a988416SJim Ingham     GetOptions ()
585a988416SJim Ingham     {
595a988416SJim Ingham         return &m_options;
605a988416SJim Ingham     }
615a988416SJim Ingham 
625a988416SJim Ingham protected:
63a5a97ebeSJim Ingham 
64a5a97ebeSJim Ingham     class CommandOptions : public Options
65a5a97ebeSJim Ingham     {
66a5a97ebeSJim Ingham     public:
67a5a97ebeSJim Ingham 
68a5a97ebeSJim Ingham         CommandOptions (CommandInterpreter &interpreter) :
697594f14fSEnrico Granata             Options (interpreter),
707594f14fSEnrico Granata             m_start_idx(0),
717594f14fSEnrico Granata             m_stop_idx(0),
727594f14fSEnrico Granata             m_count(0),
7363123b64SEnrico Granata             m_clear(false)
74a5a97ebeSJim Ingham         {
75a5a97ebeSJim Ingham         }
76a5a97ebeSJim Ingham 
77a5a97ebeSJim Ingham         virtual
78a5a97ebeSJim Ingham         ~CommandOptions (){}
79a5a97ebeSJim Ingham 
80a5a97ebeSJim Ingham         virtual Error
81a5a97ebeSJim Ingham         SetOptionValue (uint32_t option_idx, const char *option_arg)
82a5a97ebeSJim Ingham         {
83a5a97ebeSJim Ingham             Error error;
843bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
85a5a97ebeSJim Ingham 
86a5a97ebeSJim Ingham             switch (short_option)
87a5a97ebeSJim Ingham             {
88a5a97ebeSJim Ingham                 case 'c':
897594f14fSEnrico Granata                     error = m_count.SetValueFromCString(option_arg,eVarSetOperationAssign);
90a5a97ebeSJim Ingham                     break;
91a5a97ebeSJim Ingham                 case 's':
927594f14fSEnrico Granata                     if (option_arg && strcmp("end", option_arg) == 0)
937594f14fSEnrico Granata                     {
947594f14fSEnrico Granata                         m_start_idx.SetCurrentValue(UINT64_MAX);
957594f14fSEnrico Granata                         m_start_idx.SetOptionWasSet();
967594f14fSEnrico Granata                     }
977594f14fSEnrico Granata                     else
987594f14fSEnrico Granata                         error = m_start_idx.SetValueFromCString(option_arg,eVarSetOperationAssign);
997594f14fSEnrico Granata                     break;
1007594f14fSEnrico Granata                 case 'e':
1017594f14fSEnrico Granata                     error = m_stop_idx.SetValueFromCString(option_arg,eVarSetOperationAssign);
1027594f14fSEnrico Granata                     break;
10363123b64SEnrico Granata                 case 'C':
10463123b64SEnrico Granata                     m_clear.SetCurrentValue(true);
10563123b64SEnrico Granata                     m_clear.SetOptionWasSet();
106a5a97ebeSJim Ingham                     break;
107a5a97ebeSJim Ingham                 default:
10886edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
109a5a97ebeSJim Ingham                     break;
110a5a97ebeSJim Ingham             }
111a5a97ebeSJim Ingham 
112a5a97ebeSJim Ingham             return error;
113a5a97ebeSJim Ingham         }
114a5a97ebeSJim Ingham 
115a5a97ebeSJim Ingham         void
116a5a97ebeSJim Ingham         OptionParsingStarting ()
117a5a97ebeSJim Ingham         {
1187594f14fSEnrico Granata             m_start_idx.Clear();
1197594f14fSEnrico Granata             m_stop_idx.Clear();
1207594f14fSEnrico Granata             m_count.Clear();
12163123b64SEnrico Granata             m_clear.Clear();
122a5a97ebeSJim Ingham         }
123a5a97ebeSJim Ingham 
124a5a97ebeSJim Ingham         const OptionDefinition*
125a5a97ebeSJim Ingham         GetDefinitions ()
126a5a97ebeSJim Ingham         {
127a5a97ebeSJim Ingham             return g_option_table;
128a5a97ebeSJim Ingham         }
129a5a97ebeSJim Ingham 
130a5a97ebeSJim Ingham         // Options table: Required for subclasses of Options.
131a5a97ebeSJim Ingham 
132a5a97ebeSJim Ingham         static OptionDefinition g_option_table[];
133a5a97ebeSJim Ingham 
134a5a97ebeSJim Ingham         // Instance variables to hold the values for command options.
135a5a97ebeSJim Ingham 
1367594f14fSEnrico Granata         OptionValueUInt64 m_start_idx;
1377594f14fSEnrico Granata         OptionValueUInt64 m_stop_idx;
1387594f14fSEnrico Granata         OptionValueUInt64 m_count;
13963123b64SEnrico Granata         OptionValueBoolean m_clear;
140a5a97ebeSJim Ingham     };
141a5a97ebeSJim Ingham 
142a5a97ebeSJim Ingham     bool
1435a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
144a5a97ebeSJim Ingham     {
14563123b64SEnrico Granata         if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet())
1467594f14fSEnrico Granata         {
1477594f14fSEnrico Granata             m_interpreter.GetCommandHistory().Clear();
1487594f14fSEnrico Granata             result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
1497594f14fSEnrico Granata         }
1507594f14fSEnrico Granata         else
1517594f14fSEnrico Granata         {
1527594f14fSEnrico Granata             if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet())
1537594f14fSEnrico Granata             {
1547594f14fSEnrico Granata                 result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation");
1557594f14fSEnrico Granata                 result.SetStatus(lldb::eReturnStatusFailed);
1567594f14fSEnrico Granata             }
1577594f14fSEnrico Granata             else
1587594f14fSEnrico Granata             {
15984400ec7SVirgile Bello                 std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue());
16084400ec7SVirgile Bello                 std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue());
16184400ec7SVirgile Bello                 std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue());
162a5a97ebeSJim Ingham 
1637594f14fSEnrico Granata                 const CommandHistory& history(m_interpreter.GetCommandHistory());
1647594f14fSEnrico Granata 
1657594f14fSEnrico Granata                 if (start_idx.first && start_idx.second == UINT64_MAX)
1667594f14fSEnrico Granata                 {
1677594f14fSEnrico Granata                     if (count.first)
1687594f14fSEnrico Granata                     {
1697594f14fSEnrico Granata                         start_idx.second = history.GetSize() - count.second;
1707594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1717594f14fSEnrico Granata                     }
1727594f14fSEnrico Granata                     else if (stop_idx.first)
1737594f14fSEnrico Granata                     {
1747594f14fSEnrico Granata                         start_idx.second = stop_idx.second;
1757594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1767594f14fSEnrico Granata                     }
1777594f14fSEnrico Granata                     else
1787594f14fSEnrico Granata                     {
1797594f14fSEnrico Granata                         start_idx.second = 0;
1807594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1817594f14fSEnrico Granata                     }
1827594f14fSEnrico Granata                 }
1837594f14fSEnrico Granata                 else
1847594f14fSEnrico Granata                 {
1857594f14fSEnrico Granata                     if (!start_idx.first && !stop_idx.first && !count.first)
1867594f14fSEnrico Granata                     {
1877594f14fSEnrico Granata                         start_idx.second = 0;
1887594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1897594f14fSEnrico Granata                     }
1907594f14fSEnrico Granata                     else if (start_idx.first)
1917594f14fSEnrico Granata                     {
1927594f14fSEnrico Granata                         if (count.first)
1937594f14fSEnrico Granata                         {
1947594f14fSEnrico Granata                             stop_idx.second = start_idx.second + count.second - 1;
1957594f14fSEnrico Granata                         }
1967594f14fSEnrico Granata                         else if (!stop_idx.first)
1977594f14fSEnrico Granata                         {
1987594f14fSEnrico Granata                             stop_idx.second = history.GetSize() - 1;
1997594f14fSEnrico Granata                         }
2007594f14fSEnrico Granata                     }
2017594f14fSEnrico Granata                     else if (stop_idx.first)
2027594f14fSEnrico Granata                     {
2037594f14fSEnrico Granata                         if (count.first)
2047594f14fSEnrico Granata                         {
2057594f14fSEnrico Granata                             if (stop_idx.second >= count.second)
2067594f14fSEnrico Granata                                 start_idx.second = stop_idx.second - count.second + 1;
2077594f14fSEnrico Granata                             else
2087594f14fSEnrico Granata                                 start_idx.second = 0;
2097594f14fSEnrico Granata                         }
2107594f14fSEnrico Granata                     }
2117594f14fSEnrico Granata                     else /* if (count.first) */
2127594f14fSEnrico Granata                     {
2137594f14fSEnrico Granata                         start_idx.second = 0;
2147594f14fSEnrico Granata                         stop_idx.second = count.second - 1;
2157594f14fSEnrico Granata                     }
2167594f14fSEnrico Granata                 }
2177594f14fSEnrico Granata                 history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second);
2187594f14fSEnrico Granata             }
2197594f14fSEnrico Granata         }
220a5a97ebeSJim Ingham         return result.Succeeded();
221a5a97ebeSJim Ingham 
222a5a97ebeSJim Ingham     }
2235a988416SJim Ingham 
2245a988416SJim Ingham     CommandOptions m_options;
225a5a97ebeSJim Ingham };
226a5a97ebeSJim Ingham 
227a5a97ebeSJim Ingham OptionDefinition
228a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] =
229a5a97ebeSJim Ingham {
230*e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger,        "How many history commands to print."},
231*e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger,  "Index at which to start printing history commands (or end to mean tail mode)."},
232*e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger,    "Index at which to stop printing history commands."},
233*e2607b50SVirgile Bello { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, NULL, 0, eArgTypeBoolean,    "Clears the current command history."},
234a5a97ebeSJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL }
235a5a97ebeSJim Ingham };
236a5a97ebeSJim Ingham 
237a5a97ebeSJim Ingham 
238a5a97ebeSJim Ingham //-------------------------------------------------------------------------
239a5a97ebeSJim Ingham // CommandObjectCommandsSource
240a5a97ebeSJim Ingham //-------------------------------------------------------------------------
241a5a97ebeSJim Ingham 
2425a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed
243ebc09c36SJim Ingham {
2445a988416SJim Ingham public:
2455a988416SJim Ingham     CommandObjectCommandsSource(CommandInterpreter &interpreter) :
2465a988416SJim Ingham         CommandObjectParsed (interpreter,
2475a988416SJim Ingham                              "command source",
2485a988416SJim Ingham                              "Read in debugger commands from the file <filename> and execute them.",
2495a988416SJim Ingham                              NULL),
2505a988416SJim Ingham         m_options (interpreter)
2515a988416SJim Ingham     {
2525a988416SJim Ingham         CommandArgumentEntry arg;
2535a988416SJim Ingham         CommandArgumentData file_arg;
2545a988416SJim Ingham 
2555a988416SJim Ingham         // Define the first (and only) variant of this arg.
2565a988416SJim Ingham         file_arg.arg_type = eArgTypeFilename;
2575a988416SJim Ingham         file_arg.arg_repetition = eArgRepeatPlain;
2585a988416SJim Ingham 
2595a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
2605a988416SJim Ingham         arg.push_back (file_arg);
2615a988416SJim Ingham 
2625a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
2635a988416SJim Ingham         m_arguments.push_back (arg);
2645a988416SJim Ingham     }
2655a988416SJim Ingham 
2665a988416SJim Ingham     ~CommandObjectCommandsSource () {}
2675a988416SJim Ingham 
2685a988416SJim Ingham     virtual const char*
2695a988416SJim Ingham     GetRepeatCommand (Args &current_command_args, uint32_t index)
2705a988416SJim Ingham     {
2715a988416SJim Ingham         return "";
2725a988416SJim Ingham     }
2735a988416SJim Ingham 
274c7bece56SGreg Clayton     virtual int
2755a988416SJim Ingham     HandleArgumentCompletion (Args &input,
2765a988416SJim Ingham                               int &cursor_index,
2775a988416SJim Ingham                               int &cursor_char_position,
2785a988416SJim Ingham                               OptionElementVector &opt_element_vector,
2795a988416SJim Ingham                               int match_start_point,
2805a988416SJim Ingham                               int max_return_elements,
2815a988416SJim Ingham                               bool &word_complete,
2825a988416SJim Ingham                               StringList &matches)
2835a988416SJim Ingham     {
2845a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
2855a988416SJim Ingham         completion_str.erase (cursor_char_position);
2865a988416SJim Ingham 
2875a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
2885a988416SJim Ingham                                                              CommandCompletions::eDiskFileCompletion,
2895a988416SJim Ingham                                                              completion_str.c_str(),
2905a988416SJim Ingham                                                              match_start_point,
2915a988416SJim Ingham                                                              max_return_elements,
2925a988416SJim Ingham                                                              NULL,
2935a988416SJim Ingham                                                              word_complete,
2945a988416SJim Ingham                                                              matches);
2955a988416SJim Ingham         return matches.GetSize();
2965a988416SJim Ingham     }
2975a988416SJim Ingham 
2985a988416SJim Ingham     virtual Options *
2995a988416SJim Ingham     GetOptions ()
3005a988416SJim Ingham     {
3015a988416SJim Ingham         return &m_options;
3025a988416SJim Ingham     }
3035a988416SJim Ingham 
3045a988416SJim Ingham protected:
305e16c50a1SJim Ingham 
306e16c50a1SJim Ingham     class CommandOptions : public Options
307e16c50a1SJim Ingham     {
308e16c50a1SJim Ingham     public:
309e16c50a1SJim Ingham 
310eb0103f2SGreg Clayton         CommandOptions (CommandInterpreter &interpreter) :
311012d4fcaSEnrico Granata             Options (interpreter),
312012d4fcaSEnrico Granata             m_stop_on_error (true)
313eb0103f2SGreg Clayton         {
314eb0103f2SGreg Clayton         }
315e16c50a1SJim Ingham 
316e16c50a1SJim Ingham         virtual
317e16c50a1SJim Ingham         ~CommandOptions (){}
318e16c50a1SJim Ingham 
319e16c50a1SJim Ingham         virtual Error
320f6b8b581SGreg Clayton         SetOptionValue (uint32_t option_idx, const char *option_arg)
321e16c50a1SJim Ingham         {
322e16c50a1SJim Ingham             Error error;
3233bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
324e16c50a1SJim Ingham             bool success;
325e16c50a1SJim Ingham 
326e16c50a1SJim Ingham             switch (short_option)
327e16c50a1SJim Ingham             {
328e16c50a1SJim Ingham                 case 'e':
32915571f15SEnrico Granata                     error = m_stop_on_error.SetValueFromCString(option_arg);
330e16c50a1SJim Ingham                     break;
331e16c50a1SJim Ingham                 case 'c':
332e16c50a1SJim Ingham                     m_stop_on_continue = Args::StringToBoolean(option_arg, true, &success);
333e16c50a1SJim Ingham                     if (!success)
33486edbf41SGreg Clayton                         error.SetErrorStringWithFormat("invalid value for stop-on-continue: %s", option_arg);
335e16c50a1SJim Ingham                     break;
33660986174SMichael Sartain                 case 's':
33760986174SMichael Sartain                     m_silent_run = Args::StringToBoolean(option_arg, true, &success);
33860986174SMichael Sartain                     if (!success)
33960986174SMichael Sartain                         error.SetErrorStringWithFormat("invalid value for silent-run: %s", option_arg);
34060986174SMichael Sartain                     break;
341e16c50a1SJim Ingham                 default:
34286edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
343e16c50a1SJim Ingham                     break;
344e16c50a1SJim Ingham             }
345e16c50a1SJim Ingham 
346e16c50a1SJim Ingham             return error;
347e16c50a1SJim Ingham         }
348e16c50a1SJim Ingham 
349e16c50a1SJim Ingham         void
350f6b8b581SGreg Clayton         OptionParsingStarting ()
351e16c50a1SJim Ingham         {
352012d4fcaSEnrico Granata             m_stop_on_error.Clear();
35360986174SMichael Sartain             m_silent_run = false;
354e16c50a1SJim Ingham             m_stop_on_continue = true;
355e16c50a1SJim Ingham         }
356e16c50a1SJim Ingham 
357e0d378b3SGreg Clayton         const OptionDefinition*
358e16c50a1SJim Ingham         GetDefinitions ()
359e16c50a1SJim Ingham         {
360e16c50a1SJim Ingham             return g_option_table;
361e16c50a1SJim Ingham         }
362e16c50a1SJim Ingham 
363e16c50a1SJim Ingham         // Options table: Required for subclasses of Options.
364e16c50a1SJim Ingham 
365e0d378b3SGreg Clayton         static OptionDefinition g_option_table[];
366e16c50a1SJim Ingham 
367e16c50a1SJim Ingham         // Instance variables to hold the values for command options.
368e16c50a1SJim Ingham 
369012d4fcaSEnrico Granata         OptionValueBoolean m_stop_on_error;
37060986174SMichael Sartain 	    bool m_silent_run;
371e16c50a1SJim Ingham         bool m_stop_on_continue;
372e16c50a1SJim Ingham     };
373e16c50a1SJim Ingham 
374ebc09c36SJim Ingham     bool
3755a988416SJim Ingham     DoExecute(Args& command, CommandReturnObject &result)
376ebc09c36SJim Ingham     {
377c7bece56SGreg Clayton         const size_t argc = command.GetArgumentCount();
378ebc09c36SJim Ingham         if (argc == 1)
379ebc09c36SJim Ingham         {
3805a988416SJim Ingham             const char *filename = command.GetArgumentAtIndex(0);
381ebc09c36SJim Ingham 
382ebc09c36SJim Ingham             result.AppendMessageWithFormat ("Executing commands in '%s'.\n", filename);
383ebc09c36SJim Ingham 
3841ee3853fSJohnny Chen             FileSpec cmd_file (filename, true);
385e16c50a1SJim Ingham             ExecutionContext *exe_ctx = NULL;  // Just use the default context.
38660986174SMichael Sartain             bool echo_commands    = !m_options.m_silent_run;
387e16c50a1SJim Ingham             bool print_results    = true;
388012d4fcaSEnrico Granata             bool stop_on_error = m_options.m_stop_on_error.OptionWasSet() ? (bool)m_options.m_stop_on_error : m_interpreter.GetStopCmdSourceOnError();
389ebc09c36SJim Ingham 
390e16c50a1SJim Ingham             m_interpreter.HandleCommandsFromFile (cmd_file,
391e16c50a1SJim Ingham                                                   exe_ctx,
392e16c50a1SJim Ingham                                                   m_options.m_stop_on_continue,
393012d4fcaSEnrico Granata                                                   stop_on_error,
394e16c50a1SJim Ingham                                                   echo_commands,
395e16c50a1SJim Ingham                                                   print_results,
3965f5ab602SEnrico Granata                                                   eLazyBoolCalculate,
397e16c50a1SJim Ingham                                                   result);
398ebc09c36SJim Ingham         }
399ebc09c36SJim Ingham         else
400ebc09c36SJim Ingham         {
401ebc09c36SJim Ingham             result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName());
402ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
403ebc09c36SJim Ingham         }
404ebc09c36SJim Ingham         return result.Succeeded();
405ebc09c36SJim Ingham 
406ebc09c36SJim Ingham     }
4075a988416SJim Ingham     CommandOptions m_options;
408ebc09c36SJim Ingham };
409ebc09c36SJim Ingham 
410e0d378b3SGreg Clayton OptionDefinition
411e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] =
412e16c50a1SJim Ingham {
413*e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean,    "If true, stop executing commands on error."},
414*e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on continue."},
415*e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean, "If true don't echo commands while executing."},
416e16c50a1SJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL }
417e16c50a1SJim Ingham };
418e16c50a1SJim Ingham 
419ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
420ebc09c36SJim Ingham //-------------------------------------------------------------------------
421ebc09c36SJim Ingham // CommandObjectCommandsAlias
422ebc09c36SJim Ingham //-------------------------------------------------------------------------
423ebc09c36SJim Ingham 
424be93a35aSEnrico Granata static const char *g_python_command_instructions =   "Enter your Python command(s). Type 'DONE' to end.\n"
425be93a35aSEnrico Granata                                                      "You must define a Python function with this signature:\n"
42640d55710SEnrico Granata                                                      "def my_command_impl(debugger, args, result, internal_dict):";
427be93a35aSEnrico Granata 
428be93a35aSEnrico Granata 
4295a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw
430ebc09c36SJim Ingham {
431be93a35aSEnrico Granata 
432be93a35aSEnrico Granata 
433ebc09c36SJim Ingham public:
434a7015092SGreg Clayton     CommandObjectCommandsAlias (CommandInterpreter &interpreter) :
4355a988416SJim Ingham         CommandObjectRaw (interpreter,
4360e5e5a79SGreg Clayton                        "command alias",
437e3d26315SCaroline Tice                        "Allow users to define their own debugger command abbreviations.",
438405fe67fSCaroline Tice                        NULL)
439ebc09c36SJim Ingham     {
440ebc09c36SJim Ingham         SetHelpLong(
441ebc09c36SJim Ingham     "'alias' allows the user to create a short-cut or abbreviation for long \n\
442ebc09c36SJim Ingham     commands, multi-word commands, and commands that take particular options. \n\
443ebc09c36SJim Ingham     Below are some simple examples of how one might use the 'alias' command: \n\
44469c12ccbSJason Molenda     \n    'command alias sc script'            // Creates the abbreviation 'sc' for the 'script' \n\
445ebc09c36SJim Ingham                                          // command. \n\
44669c12ccbSJason Molenda     'command alias bp breakpoint'        // Creates the abbreviation 'bp' for the 'breakpoint' \n\
447ebc09c36SJim Ingham                                          // command.  Since breakpoint commands are two-word \n\
448ebc09c36SJim Ingham                                          // commands, the user will still need to enter the \n\
449ebc09c36SJim Ingham                                          // second word after 'bp', e.g. 'bp enable' or \n\
450ebc09c36SJim Ingham                                          // 'bp delete'. \n\
45169c12ccbSJason Molenda     'command alias bpl breakpoint list'  // Creates the abbreviation 'bpl' for the \n\
452ebc09c36SJim Ingham                                          // two-word command 'breakpoint list'. \n\
453ebc09c36SJim Ingham     \nAn alias can include some options for the command, with the values either \n\
454ebc09c36SJim Ingham     filled in at the time the alias is created, or specified as positional \n\
455ebc09c36SJim Ingham     arguments, to be filled in when the alias is invoked.  The following example \n\
456ebc09c36SJim Ingham     shows how to create aliases with options: \n\
457ebc09c36SJim Ingham     \n\
45869c12ccbSJason Molenda     'command alias bfl breakpoint set -f %1 -l %2' \n\
459ebc09c36SJim Ingham     \nThis creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \n\
460ebc09c36SJim Ingham     options already part of the alias.  So if the user wants to set a breakpoint \n\
461ebc09c36SJim Ingham     by file and line without explicitly having to use the -f and -l options, the \n\
462ebc09c36SJim Ingham     user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \n\
463ebc09c36SJim Ingham     for the actual arguments that will be passed when the alias command is used. \n\
464ebc09c36SJim Ingham     The number in the placeholder refers to the position/order the actual value \n\
46581ded935SJim Ingham     occupies when the alias is used.  All the occurrences of '%1' in the alias \n\
466ebc09c36SJim Ingham     will be replaced with the first argument, all the occurrences of '%2' in the \n\
467ebc09c36SJim Ingham     alias will be replaced with the second argument, and so on.  This also allows \n\
468ebc09c36SJim Ingham     actual arguments to be used multiple times within an alias (see 'process \n\
46981ded935SJim Ingham     launch' example below).  \n\
47081ded935SJim Ingham     Note: the positional arguments must substitute as whole words in the resultant\n\
47181ded935SJim Ingham     command, so you can't at present do something like:\n\
47281ded935SJim Ingham     \n\
47369c12ccbSJason Molenda     command alias bcppfl breakpoint set -f %1.cpp -l %2\n\
47481ded935SJim Ingham     \n\
47581ded935SJim Ingham     to get the file extension \".cpp\" automatically appended.  For more complex\n\
47681ded935SJim Ingham     aliasing, use the \"command regex\" command instead.\n\
47781ded935SJim Ingham     \nSo in the 'bfl' case, the actual file value will be \n\
478ebc09c36SJim Ingham     filled in with the first argument following 'bfl' and the actual line number \n\
479ebc09c36SJim Ingham     value will be filled in with the second argument.  The user would use this \n\
480ebc09c36SJim Ingham     alias as follows: \n\
48169c12ccbSJason Molenda     \n    (lldb)  command alias bfl breakpoint set -f %1 -l %2 \n\
482ebc09c36SJim Ingham     <... some time later ...> \n\
48309799af6SCaroline Tice     (lldb)  bfl my-file.c 137 \n\
484ebc09c36SJim Ingham     \nThis would be the same as if the user had entered \n\
485ebc09c36SJim Ingham     'breakpoint set -f my-file.c -l 137'. \n\
486ebc09c36SJim Ingham     \nAnother example: \n\
48769c12ccbSJason Molenda     \n    (lldb)  command alias pltty  process launch -s -o %1 -e %1 \n\
48809799af6SCaroline Tice     (lldb)  pltty /dev/tty0 \n\
489ebc09c36SJim Ingham            // becomes 'process launch -s -o /dev/tty0 -e /dev/tty0' \n\
490ebc09c36SJim Ingham     \nIf the user always wanted to pass the same value to a particular option, the \n\
491ebc09c36SJim Ingham     alias could be defined with that value directly in the alias as a constant, \n\
492ebc09c36SJim Ingham     rather than using a positional placeholder: \n\
49369c12ccbSJason Molenda     \n    command alias bl3  breakpoint set -f %1 -l 3  // Always sets a breakpoint on line \n\
494ebc09c36SJim Ingham                                                    // 3 of whatever file is indicated. \n");
495ebc09c36SJim Ingham 
496405fe67fSCaroline Tice         CommandArgumentEntry arg1;
497405fe67fSCaroline Tice         CommandArgumentEntry arg2;
498405fe67fSCaroline Tice         CommandArgumentEntry arg3;
499405fe67fSCaroline Tice         CommandArgumentData alias_arg;
500405fe67fSCaroline Tice         CommandArgumentData cmd_arg;
501405fe67fSCaroline Tice         CommandArgumentData options_arg;
502405fe67fSCaroline Tice 
503405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
504405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
505405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
506405fe67fSCaroline Tice 
507405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
508405fe67fSCaroline Tice         arg1.push_back (alias_arg);
509405fe67fSCaroline Tice 
510405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
511405fe67fSCaroline Tice         cmd_arg.arg_type = eArgTypeCommandName;
512405fe67fSCaroline Tice         cmd_arg.arg_repetition = eArgRepeatPlain;
513405fe67fSCaroline Tice 
514405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
515405fe67fSCaroline Tice         arg2.push_back (cmd_arg);
516405fe67fSCaroline Tice 
517405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
518405fe67fSCaroline Tice         options_arg.arg_type = eArgTypeAliasOptions;
519405fe67fSCaroline Tice         options_arg.arg_repetition = eArgRepeatOptional;
520405fe67fSCaroline Tice 
521405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
522405fe67fSCaroline Tice         arg3.push_back (options_arg);
523405fe67fSCaroline Tice 
524405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
525405fe67fSCaroline Tice         m_arguments.push_back (arg1);
526405fe67fSCaroline Tice         m_arguments.push_back (arg2);
527405fe67fSCaroline Tice         m_arguments.push_back (arg3);
528ebc09c36SJim Ingham     }
529ebc09c36SJim Ingham 
530ebc09c36SJim Ingham     ~CommandObjectCommandsAlias ()
531ebc09c36SJim Ingham     {
532ebc09c36SJim Ingham     }
533ebc09c36SJim Ingham 
5345a988416SJim Ingham protected:
5355a988416SJim Ingham     virtual bool
5365a988416SJim Ingham     DoExecute (const char *raw_command_line, CommandReturnObject &result)
537844d2303SCaroline Tice     {
538844d2303SCaroline Tice         Args args (raw_command_line);
539844d2303SCaroline Tice         std::string raw_command_string (raw_command_line);
540844d2303SCaroline Tice 
541844d2303SCaroline Tice         size_t argc = args.GetArgumentCount();
542844d2303SCaroline Tice 
543844d2303SCaroline Tice         if (argc < 2)
544844d2303SCaroline Tice         {
545844d2303SCaroline Tice             result.AppendError ("'alias' requires at least two arguments");
546844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
547844d2303SCaroline Tice             return false;
548844d2303SCaroline Tice         }
549844d2303SCaroline Tice 
550844d2303SCaroline Tice         // Get the alias command.
551844d2303SCaroline Tice 
552844d2303SCaroline Tice         const std::string alias_command = args.GetArgumentAtIndex (0);
553844d2303SCaroline Tice 
554844d2303SCaroline Tice         // Strip the new alias name off 'raw_command_string'  (leave it on args, which gets passed to 'Execute', which
555844d2303SCaroline Tice         // does the stripping itself.
556844d2303SCaroline Tice         size_t pos = raw_command_string.find (alias_command);
557844d2303SCaroline Tice         if (pos == 0)
558844d2303SCaroline Tice         {
559844d2303SCaroline Tice             raw_command_string = raw_command_string.substr (alias_command.size());
560844d2303SCaroline Tice             pos = raw_command_string.find_first_not_of (' ');
561844d2303SCaroline Tice             if ((pos != std::string::npos) && (pos > 0))
562844d2303SCaroline Tice                 raw_command_string = raw_command_string.substr (pos);
563844d2303SCaroline Tice         }
564844d2303SCaroline Tice         else
565844d2303SCaroline Tice         {
566844d2303SCaroline Tice             result.AppendError ("Error parsing command string.  No alias created.");
567844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
568844d2303SCaroline Tice             return false;
569844d2303SCaroline Tice         }
570844d2303SCaroline Tice 
571844d2303SCaroline Tice 
572844d2303SCaroline Tice         // Verify that the command is alias-able.
573844d2303SCaroline Tice         if (m_interpreter.CommandExists (alias_command.c_str()))
574844d2303SCaroline Tice         {
575844d2303SCaroline Tice             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
576844d2303SCaroline Tice                                           alias_command.c_str());
577844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
578844d2303SCaroline Tice             return false;
579844d2303SCaroline Tice         }
580844d2303SCaroline Tice 
581844d2303SCaroline Tice         // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string.
582844d2303SCaroline Tice         // raw_command_string is returned with the name of the command object stripped off the front.
583844d2303SCaroline Tice         CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string);
584844d2303SCaroline Tice 
585844d2303SCaroline Tice         if (!cmd_obj)
586844d2303SCaroline Tice         {
58786edbf41SGreg Clayton             result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command."
588844d2303SCaroline Tice                                           "  No alias created.", raw_command_string.c_str());
589844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
590844d2303SCaroline Tice             return false;
591844d2303SCaroline Tice         }
592844d2303SCaroline Tice         else if (!cmd_obj->WantsRawCommandString ())
593844d2303SCaroline Tice         {
594844d2303SCaroline Tice             // Note that args was initialized with the original command, and has not been updated to this point.
595844d2303SCaroline Tice             // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias.
5965a988416SJim Ingham             return HandleAliasingNormalCommand (args, result);
597844d2303SCaroline Tice         }
598844d2303SCaroline Tice         else
599844d2303SCaroline Tice         {
6005a988416SJim Ingham             return HandleAliasingRawCommand (alias_command, raw_command_string, *cmd_obj, result);
6015a988416SJim Ingham         }
6025a988416SJim Ingham         return result.Succeeded();
6035a988416SJim Ingham     }
6045a988416SJim Ingham 
6055a988416SJim Ingham     bool
6065a988416SJim Ingham     HandleAliasingRawCommand (const std::string &alias_command, std::string &raw_command_string, CommandObject &cmd_obj, CommandReturnObject &result)
6075a988416SJim Ingham     {
608844d2303SCaroline Tice             // Verify & handle any options/arguments passed to the alias command
609844d2303SCaroline Tice 
610844d2303SCaroline Tice             OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
611844d2303SCaroline Tice             OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
612844d2303SCaroline Tice 
6135a988416SJim Ingham             CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj.GetCommandName(), false);
614844d2303SCaroline Tice 
615ca90c47eSCaroline Tice             if (!m_interpreter.ProcessAliasOptionsArgs (cmd_obj_sp, raw_command_string.c_str(), option_arg_vector_sp))
616844d2303SCaroline Tice             {
617844d2303SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
618ca90c47eSCaroline Tice                 result.SetStatus (eReturnStatusFailed);
619844d2303SCaroline Tice                 return false;
620844d2303SCaroline Tice             }
621844d2303SCaroline Tice 
622844d2303SCaroline Tice             // Create the alias
623844d2303SCaroline Tice             if (m_interpreter.AliasExists (alias_command.c_str())
624844d2303SCaroline Tice                 || m_interpreter.UserCommandExists (alias_command.c_str()))
625844d2303SCaroline Tice             {
626844d2303SCaroline Tice                 OptionArgVectorSP temp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
627844d2303SCaroline Tice                 if (temp_option_arg_sp.get())
628844d2303SCaroline Tice                 {
629844d2303SCaroline Tice                     if (option_arg_vector->size() == 0)
630844d2303SCaroline Tice                         m_interpreter.RemoveAliasOptions (alias_command.c_str());
631844d2303SCaroline Tice                 }
632844d2303SCaroline Tice                 result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
633844d2303SCaroline Tice                                                 alias_command.c_str());
634844d2303SCaroline Tice             }
635844d2303SCaroline Tice 
636472362e6SCaroline Tice             if (cmd_obj_sp)
637472362e6SCaroline Tice             {
638844d2303SCaroline Tice                 m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp);
639844d2303SCaroline Tice                 if (option_arg_vector->size() > 0)
640844d2303SCaroline Tice                     m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
641844d2303SCaroline Tice                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
642844d2303SCaroline Tice             }
643472362e6SCaroline Tice             else
644472362e6SCaroline Tice             {
645472362e6SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
646472362e6SCaroline Tice                 result.SetStatus (eReturnStatusFailed);
647472362e6SCaroline Tice             }
648844d2303SCaroline Tice             return result.Succeeded ();
649844d2303SCaroline Tice     }
650ebc09c36SJim Ingham 
651ebc09c36SJim Ingham     bool
6525a988416SJim Ingham     HandleAliasingNormalCommand (Args& args, CommandReturnObject &result)
653ebc09c36SJim Ingham     {
654867b185dSCaroline Tice         size_t argc = args.GetArgumentCount();
655ebc09c36SJim Ingham 
656ebc09c36SJim Ingham         if (argc < 2)
657ebc09c36SJim Ingham         {
658ebc09c36SJim Ingham             result.AppendError ("'alias' requires at least two arguments");
659ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
660ebc09c36SJim Ingham             return false;
661ebc09c36SJim Ingham         }
662ebc09c36SJim Ingham 
663ebc09c36SJim Ingham         const std::string alias_command = args.GetArgumentAtIndex(0);
664ebc09c36SJim Ingham         const std::string actual_command = args.GetArgumentAtIndex(1);
665ebc09c36SJim Ingham 
666ebc09c36SJim Ingham         args.Shift();  // Shift the alias command word off the argument vector.
667ebc09c36SJim Ingham         args.Shift();  // Shift the old command word off the argument vector.
668ebc09c36SJim Ingham 
669ebc09c36SJim Ingham         // Verify that the command is alias'able, and get the appropriate command object.
670ebc09c36SJim Ingham 
671a7015092SGreg Clayton         if (m_interpreter.CommandExists (alias_command.c_str()))
672ebc09c36SJim Ingham         {
673ebc09c36SJim Ingham             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
674ebc09c36SJim Ingham                                          alias_command.c_str());
675ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
676ebc09c36SJim Ingham         }
677ebc09c36SJim Ingham         else
678ebc09c36SJim Ingham         {
679a7015092SGreg Clayton              CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true));
680ebc09c36SJim Ingham              CommandObjectSP subcommand_obj_sp;
681ebc09c36SJim Ingham              bool use_subcommand = false;
682ebc09c36SJim Ingham              if (command_obj_sp.get())
683ebc09c36SJim Ingham              {
684ebc09c36SJim Ingham                  CommandObject *cmd_obj = command_obj_sp.get();
685c982c768SGreg Clayton                  CommandObject *sub_cmd_obj = NULL;
686ebc09c36SJim Ingham                  OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
687ebc09c36SJim Ingham                  OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
688ebc09c36SJim Ingham 
689844d2303SCaroline Tice                  while (cmd_obj->IsMultiwordObject() && args.GetArgumentCount() > 0)
690ebc09c36SJim Ingham                  {
691ebc09c36SJim Ingham                      if (argc >= 3)
692ebc09c36SJim Ingham                      {
693ebc09c36SJim Ingham                          const std::string sub_command = args.GetArgumentAtIndex(0);
694ebc09c36SJim Ingham                          assert (sub_command.length() != 0);
695998255bfSGreg Clayton                          subcommand_obj_sp = cmd_obj->GetSubcommandSP (sub_command.c_str());
696ebc09c36SJim Ingham                          if (subcommand_obj_sp.get())
697ebc09c36SJim Ingham                          {
698ebc09c36SJim Ingham                              sub_cmd_obj = subcommand_obj_sp.get();
699ebc09c36SJim Ingham                              use_subcommand = true;
700ebc09c36SJim Ingham                              args.Shift();  // Shift the sub_command word off the argument vector.
701844d2303SCaroline Tice                              cmd_obj = sub_cmd_obj;
702ebc09c36SJim Ingham                          }
703ebc09c36SJim Ingham                          else
704ebc09c36SJim Ingham                          {
705f415eeb4SCaroline Tice                              result.AppendErrorWithFormat("'%s' is not a valid sub-command of '%s'.  "
706f415eeb4SCaroline Tice                                                           "Unable to create alias.\n",
707f415eeb4SCaroline Tice                                                           sub_command.c_str(), actual_command.c_str());
708ebc09c36SJim Ingham                              result.SetStatus (eReturnStatusFailed);
709ebc09c36SJim Ingham                              return false;
710ebc09c36SJim Ingham                          }
711ebc09c36SJim Ingham                      }
712ebc09c36SJim Ingham                  }
713ebc09c36SJim Ingham 
714ebc09c36SJim Ingham                  // Verify & handle any options/arguments passed to the alias command
715ebc09c36SJim Ingham 
716ebc09c36SJim Ingham                  if (args.GetArgumentCount () > 0)
717ebc09c36SJim Ingham                  {
718ca90c47eSCaroline Tice                     CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false);
719ebc09c36SJim Ingham                     if (use_subcommand)
720ca90c47eSCaroline Tice                         tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false);
721ca90c47eSCaroline Tice 
722ca90c47eSCaroline Tice                     std::string args_string;
723ca90c47eSCaroline Tice                     args.GetCommandString (args_string);
724ca90c47eSCaroline Tice 
725ca90c47eSCaroline Tice                     if (!m_interpreter.ProcessAliasOptionsArgs (tmp_sp, args_string.c_str(), option_arg_vector_sp))
726ebc09c36SJim Ingham                     {
727ca90c47eSCaroline Tice                         result.AppendError ("Unable to create requested alias.\n");
728ca90c47eSCaroline Tice                         result.SetStatus (eReturnStatusFailed);
729e7941795SCaroline Tice                         return false;
730867b185dSCaroline Tice                     }
731867b185dSCaroline Tice                  }
732867b185dSCaroline Tice 
733ebc09c36SJim Ingham                  // Create the alias.
734ebc09c36SJim Ingham 
735a7015092SGreg Clayton                  if (m_interpreter.AliasExists (alias_command.c_str())
736a7015092SGreg Clayton                      || m_interpreter.UserCommandExists (alias_command.c_str()))
737ebc09c36SJim Ingham                  {
738a7015092SGreg Clayton                      OptionArgVectorSP tmp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
739ebc09c36SJim Ingham                      if (tmp_option_arg_sp.get())
740ebc09c36SJim Ingham                      {
741ebc09c36SJim Ingham                          if (option_arg_vector->size() == 0)
742a7015092SGreg Clayton                              m_interpreter.RemoveAliasOptions (alias_command.c_str());
743ebc09c36SJim Ingham                      }
744ebc09c36SJim Ingham                      result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
745ebc09c36SJim Ingham                                                      alias_command.c_str());
746ebc09c36SJim Ingham                  }
747ebc09c36SJim Ingham 
748ebc09c36SJim Ingham                  if (use_subcommand)
749a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), subcommand_obj_sp);
750ebc09c36SJim Ingham                  else
751a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), command_obj_sp);
752ebc09c36SJim Ingham                  if (option_arg_vector->size() > 0)
753a7015092SGreg Clayton                      m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
754ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusSuccessFinishNoResult);
755ebc09c36SJim Ingham              }
756ebc09c36SJim Ingham              else
757ebc09c36SJim Ingham              {
758ebc09c36SJim Ingham                  result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str());
759ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusFailed);
760e7941795SCaroline Tice                  return false;
761ebc09c36SJim Ingham              }
762ebc09c36SJim Ingham         }
763ebc09c36SJim Ingham 
764ebc09c36SJim Ingham         return result.Succeeded();
765ebc09c36SJim Ingham     }
7665a988416SJim Ingham 
767ebc09c36SJim Ingham };
768ebc09c36SJim Ingham 
769ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
770ebc09c36SJim Ingham //-------------------------------------------------------------------------
771ebc09c36SJim Ingham // CommandObjectCommandsUnalias
772ebc09c36SJim Ingham //-------------------------------------------------------------------------
773ebc09c36SJim Ingham 
7745a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed
775ebc09c36SJim Ingham {
776ebc09c36SJim Ingham public:
777a7015092SGreg Clayton     CommandObjectCommandsUnalias (CommandInterpreter &interpreter) :
7785a988416SJim Ingham         CommandObjectParsed (interpreter,
7790e5e5a79SGreg Clayton                        "command unalias",
78086ddae50SCaroline Tice                        "Allow the user to remove/delete a user-defined command abbreviation.",
781405fe67fSCaroline Tice                        NULL)
782ebc09c36SJim Ingham     {
783405fe67fSCaroline Tice         CommandArgumentEntry arg;
784405fe67fSCaroline Tice         CommandArgumentData alias_arg;
785405fe67fSCaroline Tice 
786405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
787405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
788405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
789405fe67fSCaroline Tice 
790405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
791405fe67fSCaroline Tice         arg.push_back (alias_arg);
792405fe67fSCaroline Tice 
793405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
794405fe67fSCaroline Tice         m_arguments.push_back (arg);
795ebc09c36SJim Ingham     }
796ebc09c36SJim Ingham 
797ebc09c36SJim Ingham     ~CommandObjectCommandsUnalias()
798ebc09c36SJim Ingham     {
799ebc09c36SJim Ingham     }
800ebc09c36SJim Ingham 
8015a988416SJim Ingham protected:
802ebc09c36SJim Ingham     bool
8035a988416SJim Ingham     DoExecute (Args& args, CommandReturnObject &result)
804ebc09c36SJim Ingham     {
805ebc09c36SJim Ingham         CommandObject::CommandMap::iterator pos;
806ebc09c36SJim Ingham         CommandObject *cmd_obj;
807ebc09c36SJim Ingham 
808ebc09c36SJim Ingham         if (args.GetArgumentCount() != 0)
809ebc09c36SJim Ingham         {
810ebc09c36SJim Ingham             const char *command_name = args.GetArgumentAtIndex(0);
811a7015092SGreg Clayton             cmd_obj = m_interpreter.GetCommandObject(command_name);
812ebc09c36SJim Ingham             if (cmd_obj)
813ebc09c36SJim Ingham             {
814a7015092SGreg Clayton                 if (m_interpreter.CommandExists (command_name))
815ebc09c36SJim Ingham                 {
816ebc09c36SJim Ingham                     result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
817ebc09c36SJim Ingham                                                   command_name);
818ebc09c36SJim Ingham                     result.SetStatus (eReturnStatusFailed);
819ebc09c36SJim Ingham                 }
820ebc09c36SJim Ingham                 else
821ebc09c36SJim Ingham                 {
822ebc09c36SJim Ingham 
823a7015092SGreg Clayton                     if (m_interpreter.RemoveAlias (command_name) == false)
824ebc09c36SJim Ingham                     {
825a7015092SGreg Clayton                         if (m_interpreter.AliasExists (command_name))
826ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n",
827ebc09c36SJim Ingham                                                           command_name);
828ebc09c36SJim Ingham                         else
829ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name);
830ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusFailed);
831ebc09c36SJim Ingham                     }
832ebc09c36SJim Ingham                     else
833ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
834ebc09c36SJim Ingham                 }
835ebc09c36SJim Ingham             }
836ebc09c36SJim Ingham             else
837ebc09c36SJim Ingham             {
838ebc09c36SJim Ingham                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a "
839ebc09c36SJim Ingham                                               "current list of commands.\n",
840ebc09c36SJim Ingham                                              command_name);
841ebc09c36SJim Ingham                 result.SetStatus (eReturnStatusFailed);
842ebc09c36SJim Ingham             }
843ebc09c36SJim Ingham         }
844ebc09c36SJim Ingham         else
845ebc09c36SJim Ingham         {
846ebc09c36SJim Ingham             result.AppendError ("must call 'unalias' with a valid alias");
847ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
848ebc09c36SJim Ingham         }
849ebc09c36SJim Ingham 
850ebc09c36SJim Ingham         return result.Succeeded();
851ebc09c36SJim Ingham     }
852ebc09c36SJim Ingham };
853ebc09c36SJim Ingham 
854de164aaaSGreg Clayton //-------------------------------------------------------------------------
855de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
856de164aaaSGreg Clayton //-------------------------------------------------------------------------
8575a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
858de164aaaSGreg Clayton 
8595a988416SJim Ingham class CommandObjectCommandsAddRegex : public CommandObjectParsed
860de164aaaSGreg Clayton {
861de164aaaSGreg Clayton public:
862de164aaaSGreg Clayton     CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) :
8635a988416SJim Ingham         CommandObjectParsed (interpreter,
8640e5e5a79SGreg Clayton                        "command regex",
865de164aaaSGreg Clayton                        "Allow the user to create a regular expression command.",
8660e5e5a79SGreg Clayton                        "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
867de164aaaSGreg Clayton         m_options (interpreter)
868de164aaaSGreg Clayton     {
8690e5e5a79SGreg Clayton         SetHelpLong(
8700e5e5a79SGreg Clayton "This command allows the user to create powerful regular expression commands\n"
8710e5e5a79SGreg Clayton "with substitutions. The regular expressions and substitutions are specified\n"
8720e5e5a79SGreg Clayton "using the regular exression substitution format of:\n"
8730e5e5a79SGreg Clayton "\n"
8740e5e5a79SGreg Clayton "    s/<regex>/<subst>/\n"
8750e5e5a79SGreg Clayton "\n"
8760e5e5a79SGreg Clayton "<regex> is a regular expression that can use parenthesis to capture regular\n"
8770e5e5a79SGreg Clayton "expression input and substitute the captured matches in the output using %1\n"
8780e5e5a79SGreg Clayton "for the first match, %2 for the second, and so on.\n"
8790e5e5a79SGreg Clayton "\n"
8800e5e5a79SGreg Clayton "The regular expressions can all be specified on the command line if more than\n"
8810e5e5a79SGreg Clayton "one argument is provided. If just the command name is provided on the command\n"
8820e5e5a79SGreg Clayton "line, then the regular expressions and substitutions can be entered on separate\n"
8830e5e5a79SGreg Clayton " lines, followed by an empty line to terminate the command definition.\n"
8840e5e5a79SGreg Clayton "\n"
8850e5e5a79SGreg Clayton "EXAMPLES\n"
8860e5e5a79SGreg Clayton "\n"
887adc43c99SSean Callanan "The following example will define a regular expression command named 'f' that\n"
8880e5e5a79SGreg Clayton "will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if\n"
8890e5e5a79SGreg Clayton "a number follows 'f':\n"
890adc43c99SSean Callanan "\n"
8910e5e5a79SGreg Clayton "    (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/'\n"
892adc43c99SSean Callanan "\n"
8930e5e5a79SGreg Clayton                     );
894de164aaaSGreg Clayton     }
895de164aaaSGreg Clayton 
896de164aaaSGreg Clayton     ~CommandObjectCommandsAddRegex()
897de164aaaSGreg Clayton     {
898de164aaaSGreg Clayton     }
899de164aaaSGreg Clayton 
900de164aaaSGreg Clayton 
9015a988416SJim Ingham protected:
902de164aaaSGreg Clayton     bool
9035a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
904de164aaaSGreg Clayton     {
9055a988416SJim Ingham         const size_t argc = command.GetArgumentCount();
9060e5e5a79SGreg Clayton         if (argc == 0)
907de164aaaSGreg Clayton         {
90869c12ccbSJason Molenda             result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
9090e5e5a79SGreg Clayton             result.SetStatus (eReturnStatusFailed);
9100e5e5a79SGreg Clayton         }
9110e5e5a79SGreg Clayton         else
9120e5e5a79SGreg Clayton         {
9130e5e5a79SGreg Clayton             Error error;
9145a988416SJim Ingham             const char *name = command.GetArgumentAtIndex(0);
915de164aaaSGreg Clayton             m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter,
916de164aaaSGreg Clayton                                                                  name,
917de164aaaSGreg Clayton                                                                  m_options.GetHelp (),
918de164aaaSGreg Clayton                                                                  m_options.GetSyntax (),
919de164aaaSGreg Clayton                                                                  10));
9200e5e5a79SGreg Clayton 
9210e5e5a79SGreg Clayton             if (argc == 1)
9220e5e5a79SGreg Clayton             {
9230e5e5a79SGreg Clayton                 InputReaderSP reader_sp (new InputReader(m_interpreter.GetDebugger()));
924de164aaaSGreg Clayton                 if (reader_sp)
925de164aaaSGreg Clayton                 {
9260e5e5a79SGreg Clayton                     error =reader_sp->Initialize (CommandObjectCommandsAddRegex::InputReaderCallback,
927de164aaaSGreg Clayton                                                   this,                         // baton
928de164aaaSGreg Clayton                                                   eInputReaderGranularityLine,  // token size, to pass to callback function
9290e5e5a79SGreg Clayton                                                   NULL,                         // end token
930de164aaaSGreg Clayton                                                   "> ",                         // prompt
9310e5e5a79SGreg Clayton                                                   true);                        // echo input
9320e5e5a79SGreg Clayton                     if (error.Success())
933de164aaaSGreg Clayton                     {
934de164aaaSGreg Clayton                         m_interpreter.GetDebugger().PushInputReader (reader_sp);
935de164aaaSGreg Clayton                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
9360e5e5a79SGreg Clayton                         return true;
937de164aaaSGreg Clayton                     }
938de164aaaSGreg Clayton                 }
939de164aaaSGreg Clayton             }
940de164aaaSGreg Clayton             else
941de164aaaSGreg Clayton             {
9420e5e5a79SGreg Clayton                 for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx)
9430e5e5a79SGreg Clayton                 {
9445a988416SJim Ingham                     llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx));
9450e5e5a79SGreg Clayton                     error = AppendRegexSubstitution (arg_strref);
9460e5e5a79SGreg Clayton                     if (error.Fail())
9470e5e5a79SGreg Clayton                         break;
9480e5e5a79SGreg Clayton                 }
9490e5e5a79SGreg Clayton 
9500e5e5a79SGreg Clayton                 if (error.Success())
9510e5e5a79SGreg Clayton                 {
9520e5e5a79SGreg Clayton                     AddRegexCommandToInterpreter();
9530e5e5a79SGreg Clayton                 }
9540e5e5a79SGreg Clayton             }
9550e5e5a79SGreg Clayton             if (error.Fail())
9560e5e5a79SGreg Clayton             {
9570e5e5a79SGreg Clayton                 result.AppendError (error.AsCString());
958de164aaaSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
959de164aaaSGreg Clayton             }
9600e5e5a79SGreg Clayton         }
9610e5e5a79SGreg Clayton 
962de164aaaSGreg Clayton         return result.Succeeded();
963de164aaaSGreg Clayton     }
964de164aaaSGreg Clayton 
9650e5e5a79SGreg Clayton     Error
9660e5e5a79SGreg Clayton     AppendRegexSubstitution (const llvm::StringRef &regex_sed)
967de164aaaSGreg Clayton     {
9680e5e5a79SGreg Clayton         Error error;
9690e5e5a79SGreg Clayton 
9700e5e5a79SGreg Clayton         if (m_regex_cmd_ap.get() == NULL)
971de164aaaSGreg Clayton         {
9720e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'",
9730e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
9740e5e5a79SGreg Clayton                                            regex_sed.data());
9750e5e5a79SGreg Clayton             return error;
976de164aaaSGreg Clayton         }
9770e5e5a79SGreg Clayton 
9780e5e5a79SGreg Clayton         size_t regex_sed_size = regex_sed.size();
9790e5e5a79SGreg Clayton 
9800e5e5a79SGreg Clayton         if (regex_sed_size <= 1)
9810e5e5a79SGreg Clayton         {
9820e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'",
9830e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
9840e5e5a79SGreg Clayton                                            regex_sed.data());
9850e5e5a79SGreg Clayton             return error;
9860e5e5a79SGreg Clayton         }
9870e5e5a79SGreg Clayton 
9880e5e5a79SGreg Clayton         if (regex_sed[0] != 's')
9890e5e5a79SGreg Clayton         {
9900e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'",
9910e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
9920e5e5a79SGreg Clayton                                            regex_sed.data());
9930e5e5a79SGreg Clayton             return error;
9940e5e5a79SGreg Clayton         }
9950e5e5a79SGreg Clayton         const size_t first_separator_char_pos = 1;
9960e5e5a79SGreg Clayton         // use the char that follows 's' as the regex separator character
9970e5e5a79SGreg Clayton         // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9980e5e5a79SGreg Clayton         const char separator_char = regex_sed[first_separator_char_pos];
9990e5e5a79SGreg Clayton         const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1);
10000e5e5a79SGreg Clayton 
10010e5e5a79SGreg Clayton         if (second_separator_char_pos == std::string::npos)
10020e5e5a79SGreg Clayton         {
10030e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s'",
10040e5e5a79SGreg Clayton                                            separator_char,
10050e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - first_separator_char_pos - 1),
10060e5e5a79SGreg Clayton                                            regex_sed.data() + (first_separator_char_pos + 1));
10070e5e5a79SGreg Clayton             return error;
10080e5e5a79SGreg Clayton         }
10090e5e5a79SGreg Clayton 
10100e5e5a79SGreg Clayton         const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1);
10110e5e5a79SGreg Clayton 
10120e5e5a79SGreg Clayton         if (third_separator_char_pos == std::string::npos)
10130e5e5a79SGreg Clayton         {
10140e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s'",
10150e5e5a79SGreg Clayton                                            separator_char,
10160e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - second_separator_char_pos - 1),
10170e5e5a79SGreg Clayton                                            regex_sed.data() + (second_separator_char_pos + 1));
10180e5e5a79SGreg Clayton             return error;
10190e5e5a79SGreg Clayton         }
10200e5e5a79SGreg Clayton 
10210e5e5a79SGreg Clayton         if (third_separator_char_pos != regex_sed_size - 1)
10220e5e5a79SGreg Clayton         {
10230e5e5a79SGreg Clayton             // Make sure that everything that follows the last regex
10240e5e5a79SGreg Clayton             // separator char
10250e5e5a79SGreg Clayton             if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos)
10260e5e5a79SGreg Clayton             {
10270e5e5a79SGreg Clayton                 error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'",
10280e5e5a79SGreg Clayton                                                (int)third_separator_char_pos + 1,
10290e5e5a79SGreg Clayton                                                regex_sed.data(),
10300e5e5a79SGreg Clayton                                                (int)(regex_sed.size() - third_separator_char_pos - 1),
10310e5e5a79SGreg Clayton                                                regex_sed.data() + (third_separator_char_pos + 1));
10320e5e5a79SGreg Clayton                 return error;
10330e5e5a79SGreg Clayton             }
10340e5e5a79SGreg Clayton 
10350e5e5a79SGreg Clayton         }
10360e5e5a79SGreg Clayton         else if (first_separator_char_pos + 1 == second_separator_char_pos)
10370e5e5a79SGreg Clayton         {
10380e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
10390e5e5a79SGreg Clayton                                            separator_char,
10400e5e5a79SGreg Clayton                                            separator_char,
10410e5e5a79SGreg Clayton                                            separator_char,
10420e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
10430e5e5a79SGreg Clayton                                            regex_sed.data());
10440e5e5a79SGreg Clayton             return error;
10450e5e5a79SGreg Clayton         }
10460e5e5a79SGreg Clayton         else if (second_separator_char_pos + 1 == third_separator_char_pos)
10470e5e5a79SGreg Clayton         {
10480e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
10490e5e5a79SGreg Clayton                                            separator_char,
10500e5e5a79SGreg Clayton                                            separator_char,
10510e5e5a79SGreg Clayton                                            separator_char,
10520e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
10530e5e5a79SGreg Clayton                                            regex_sed.data());
10540e5e5a79SGreg Clayton             return error;
10550e5e5a79SGreg Clayton         }
10560e5e5a79SGreg Clayton         std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1));
10570e5e5a79SGreg Clayton         std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1));
10580e5e5a79SGreg Clayton         m_regex_cmd_ap->AddRegexCommand (regex.c_str(),
10590e5e5a79SGreg Clayton                                          subst.c_str());
10600e5e5a79SGreg Clayton         return error;
1061de164aaaSGreg Clayton     }
1062de164aaaSGreg Clayton 
1063de164aaaSGreg Clayton     void
10640e5e5a79SGreg Clayton     AddRegexCommandToInterpreter()
1065de164aaaSGreg Clayton     {
1066de164aaaSGreg Clayton         if (m_regex_cmd_ap.get())
1067de164aaaSGreg Clayton         {
1068de164aaaSGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
1069de164aaaSGreg Clayton             {
1070de164aaaSGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
1071de164aaaSGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1072de164aaaSGreg Clayton             }
1073de164aaaSGreg Clayton         }
1074de164aaaSGreg Clayton     }
1075de164aaaSGreg Clayton 
10760e5e5a79SGreg Clayton     void
10770e5e5a79SGreg Clayton     InputReaderDidCancel()
10780e5e5a79SGreg Clayton     {
10790e5e5a79SGreg Clayton         m_regex_cmd_ap.reset();
10800e5e5a79SGreg Clayton     }
10810e5e5a79SGreg Clayton 
1082de164aaaSGreg Clayton     static size_t
1083de164aaaSGreg Clayton     InputReaderCallback (void *baton,
1084de164aaaSGreg Clayton                          InputReader &reader,
1085de164aaaSGreg Clayton                          lldb::InputReaderAction notification,
1086de164aaaSGreg Clayton                          const char *bytes,
10875a988416SJim Ingham                          size_t bytes_len)
10885a988416SJim Ingham     {
10895a988416SJim Ingham         CommandObjectCommandsAddRegex *add_regex_cmd = (CommandObjectCommandsAddRegex *) baton;
10905a988416SJim Ingham         bool batch_mode = reader.GetDebugger().GetCommandInterpreter().GetBatchCommandMode();
10915a988416SJim Ingham 
10925a988416SJim Ingham         switch (notification)
10935a988416SJim Ingham         {
10945a988416SJim Ingham             case eInputReaderActivate:
10955a988416SJim Ingham                 if (!batch_mode)
10965a988416SJim Ingham                 {
10975a988416SJim Ingham                     StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream ();
10985a988416SJim Ingham                     out_stream->Printf("%s\n", "Enter regular expressions in the form 's/<regex>/<subst>/' and terminate with an empty line:");
10995a988416SJim Ingham                     out_stream->Flush();
11005a988416SJim Ingham                 }
11015a988416SJim Ingham                 break;
11025a988416SJim Ingham             case eInputReaderReactivate:
11035a988416SJim Ingham                 break;
11045a988416SJim Ingham 
11055a988416SJim Ingham             case eInputReaderDeactivate:
11065a988416SJim Ingham                 break;
11075a988416SJim Ingham 
11085a988416SJim Ingham             case eInputReaderAsynchronousOutputWritten:
11095a988416SJim Ingham                 break;
11105a988416SJim Ingham 
11115a988416SJim Ingham             case eInputReaderGotToken:
11125a988416SJim Ingham                 while (bytes_len > 0 && (bytes[bytes_len-1] == '\r' || bytes[bytes_len-1] == '\n'))
11135a988416SJim Ingham                     --bytes_len;
11145a988416SJim Ingham                 if (bytes_len == 0)
11155a988416SJim Ingham                     reader.SetIsDone(true);
11165a988416SJim Ingham                 else if (bytes)
11175a988416SJim Ingham                 {
11185a988416SJim Ingham                     llvm::StringRef bytes_strref (bytes, bytes_len);
11195a988416SJim Ingham                     Error error (add_regex_cmd->AppendRegexSubstitution (bytes_strref));
11205a988416SJim Ingham                     if (error.Fail())
11215a988416SJim Ingham                     {
11225a988416SJim Ingham                         if (!batch_mode)
11235a988416SJim Ingham                         {
11245a988416SJim Ingham                             StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream();
11255a988416SJim Ingham                             out_stream->Printf("error: %s\n", error.AsCString());
11265a988416SJim Ingham                             out_stream->Flush();
11275a988416SJim Ingham                         }
11285a988416SJim Ingham                         add_regex_cmd->InputReaderDidCancel ();
11295a988416SJim Ingham                         reader.SetIsDone (true);
11305a988416SJim Ingham                     }
11315a988416SJim Ingham                 }
11325a988416SJim Ingham                 break;
11335a988416SJim Ingham 
11345a988416SJim Ingham             case eInputReaderInterrupt:
11355a988416SJim Ingham                 {
11365a988416SJim Ingham                     reader.SetIsDone (true);
11375a988416SJim Ingham                     if (!batch_mode)
11385a988416SJim Ingham                     {
11395a988416SJim Ingham                         StreamSP out_stream = reader.GetDebugger().GetAsyncOutputStream();
11405a988416SJim Ingham                         out_stream->PutCString("Regular expression command creations was cancelled.\n");
11415a988416SJim Ingham                         out_stream->Flush();
11425a988416SJim Ingham                     }
11435a988416SJim Ingham                     add_regex_cmd->InputReaderDidCancel ();
11445a988416SJim Ingham                 }
11455a988416SJim Ingham                 break;
11465a988416SJim Ingham 
11475a988416SJim Ingham             case eInputReaderEndOfFile:
11485a988416SJim Ingham                 reader.SetIsDone (true);
11495a988416SJim Ingham                 break;
11505a988416SJim Ingham 
11515a988416SJim Ingham             case eInputReaderDone:
11525a988416SJim Ingham                 add_regex_cmd->AddRegexCommandToInterpreter();
11535a988416SJim Ingham                 break;
11545a988416SJim Ingham         }
11555a988416SJim Ingham 
11565a988416SJim Ingham         return bytes_len;
11575a988416SJim Ingham     }
11585a988416SJim Ingham 
1159de164aaaSGreg Clayton private:
11607b0992d9SGreg Clayton     std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1161de164aaaSGreg Clayton 
1162de164aaaSGreg Clayton      class CommandOptions : public Options
1163de164aaaSGreg Clayton      {
1164de164aaaSGreg Clayton      public:
1165de164aaaSGreg Clayton 
1166de164aaaSGreg Clayton          CommandOptions (CommandInterpreter &interpreter) :
1167de164aaaSGreg Clayton             Options (interpreter)
1168de164aaaSGreg Clayton          {
1169de164aaaSGreg Clayton          }
1170de164aaaSGreg Clayton 
1171de164aaaSGreg Clayton          virtual
1172de164aaaSGreg Clayton          ~CommandOptions (){}
1173de164aaaSGreg Clayton 
1174de164aaaSGreg Clayton          virtual Error
1175de164aaaSGreg Clayton          SetOptionValue (uint32_t option_idx, const char *option_arg)
1176de164aaaSGreg Clayton          {
1177de164aaaSGreg Clayton              Error error;
11783bcdfc0eSGreg Clayton              const int short_option = m_getopt_table[option_idx].val;
1179de164aaaSGreg Clayton 
1180de164aaaSGreg Clayton              switch (short_option)
1181de164aaaSGreg Clayton              {
1182de164aaaSGreg Clayton                  case 'h':
1183de164aaaSGreg Clayton                      m_help.assign (option_arg);
1184de164aaaSGreg Clayton                      break;
1185de164aaaSGreg Clayton                  case 's':
1186de164aaaSGreg Clayton                      m_syntax.assign (option_arg);
1187de164aaaSGreg Clayton                      break;
1188de164aaaSGreg Clayton 
1189de164aaaSGreg Clayton                  default:
119086edbf41SGreg Clayton                      error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1191de164aaaSGreg Clayton                      break;
1192de164aaaSGreg Clayton              }
1193de164aaaSGreg Clayton 
1194de164aaaSGreg Clayton              return error;
1195de164aaaSGreg Clayton          }
1196de164aaaSGreg Clayton 
1197de164aaaSGreg Clayton          void
1198de164aaaSGreg Clayton          OptionParsingStarting ()
1199de164aaaSGreg Clayton          {
1200de164aaaSGreg Clayton              m_help.clear();
1201de164aaaSGreg Clayton              m_syntax.clear();
1202de164aaaSGreg Clayton          }
1203de164aaaSGreg Clayton 
1204de164aaaSGreg Clayton          const OptionDefinition*
1205de164aaaSGreg Clayton          GetDefinitions ()
1206de164aaaSGreg Clayton          {
1207de164aaaSGreg Clayton              return g_option_table;
1208de164aaaSGreg Clayton          }
1209de164aaaSGreg Clayton 
1210de164aaaSGreg Clayton          // Options table: Required for subclasses of Options.
1211de164aaaSGreg Clayton 
1212de164aaaSGreg Clayton          static OptionDefinition g_option_table[];
1213de164aaaSGreg Clayton 
1214de164aaaSGreg Clayton          const char *
1215de164aaaSGreg Clayton          GetHelp ()
1216de164aaaSGreg Clayton          {
1217de164aaaSGreg Clayton              if (m_help.empty())
1218de164aaaSGreg Clayton                  return NULL;
1219de164aaaSGreg Clayton              return m_help.c_str();
1220de164aaaSGreg Clayton          }
1221de164aaaSGreg Clayton          const char *
1222de164aaaSGreg Clayton          GetSyntax ()
1223de164aaaSGreg Clayton          {
1224de164aaaSGreg Clayton              if (m_syntax.empty())
1225de164aaaSGreg Clayton                  return NULL;
1226de164aaaSGreg Clayton              return m_syntax.c_str();
1227de164aaaSGreg Clayton          }
1228de164aaaSGreg Clayton          // Instance variables to hold the values for command options.
1229de164aaaSGreg Clayton      protected:
1230de164aaaSGreg Clayton          std::string m_help;
1231de164aaaSGreg Clayton          std::string m_syntax;
1232de164aaaSGreg Clayton      };
1233de164aaaSGreg Clayton 
1234de164aaaSGreg Clayton      virtual Options *
1235de164aaaSGreg Clayton      GetOptions ()
1236de164aaaSGreg Clayton      {
1237de164aaaSGreg Clayton          return &m_options;
1238de164aaaSGreg Clayton      }
1239de164aaaSGreg Clayton 
12405a988416SJim Ingham      CommandOptions m_options;
1241de164aaaSGreg Clayton };
1242de164aaaSGreg Clayton 
1243de164aaaSGreg Clayton OptionDefinition
1244de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] =
1245de164aaaSGreg Clayton {
1246*e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, NULL, 0, eArgTypeNone, "The help text to display for this command."},
1247*e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."},
1248de164aaaSGreg Clayton { 0             , false,  NULL   , 0  , 0                , NULL, 0, eArgTypeNone, NULL }
1249de164aaaSGreg Clayton };
1250de164aaaSGreg Clayton 
1251de164aaaSGreg Clayton 
12525a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw
1253223383edSEnrico Granata {
1254223383edSEnrico Granata private:
1255223383edSEnrico Granata     std::string m_function_name;
12560a305db7SEnrico Granata     ScriptedCommandSynchronicity m_synchro;
1257fac939e9SEnrico Granata     bool m_fetched_help_long;
1258223383edSEnrico Granata 
1259223383edSEnrico Granata public:
1260223383edSEnrico Granata 
1261223383edSEnrico Granata     CommandObjectPythonFunction (CommandInterpreter &interpreter,
1262223383edSEnrico Granata                                  std::string name,
12630a305db7SEnrico Granata                                  std::string funct,
12640a305db7SEnrico Granata                                  ScriptedCommandSynchronicity synch) :
12655a988416SJim Ingham         CommandObjectRaw (interpreter,
1266223383edSEnrico Granata                           name.c_str(),
1267223383edSEnrico Granata                           (std::string("Run Python function ") + funct).c_str(),
1268223383edSEnrico Granata                           NULL),
12690a305db7SEnrico Granata         m_function_name(funct),
1270fac939e9SEnrico Granata         m_synchro(synch),
1271fac939e9SEnrico Granata         m_fetched_help_long(false)
1272223383edSEnrico Granata     {
1273223383edSEnrico Granata     }
1274223383edSEnrico Granata 
1275223383edSEnrico Granata     virtual
1276223383edSEnrico Granata     ~CommandObjectPythonFunction ()
1277223383edSEnrico Granata     {
1278223383edSEnrico Granata     }
1279223383edSEnrico Granata 
1280223383edSEnrico Granata     virtual bool
12813a18e319SGreg Clayton     IsRemovable () const
12825a988416SJim Ingham     {
12835a988416SJim Ingham         return true;
12845a988416SJim Ingham     }
12855a988416SJim Ingham 
12865a988416SJim Ingham     const std::string&
12875a988416SJim Ingham     GetFunctionName ()
12885a988416SJim Ingham     {
12895a988416SJim Ingham         return m_function_name;
12905a988416SJim Ingham     }
12915a988416SJim Ingham 
12925a988416SJim Ingham     ScriptedCommandSynchronicity
12935a988416SJim Ingham     GetSynchronicity ()
12945a988416SJim Ingham     {
12955a988416SJim Ingham         return m_synchro;
12965a988416SJim Ingham     }
12975a988416SJim Ingham 
1298fac939e9SEnrico Granata     virtual const char *
1299fac939e9SEnrico Granata     GetHelpLong ()
1300fac939e9SEnrico Granata     {
1301fac939e9SEnrico Granata         if (!m_fetched_help_long)
1302fac939e9SEnrico Granata         {
1303fac939e9SEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1304fac939e9SEnrico Granata             if (scripter)
1305fac939e9SEnrico Granata             {
1306fac939e9SEnrico Granata                 std::string docstring;
1307fac939e9SEnrico Granata                 m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring);
1308fac939e9SEnrico Granata                 if (!docstring.empty())
1309fac939e9SEnrico Granata                     SetHelpLong(docstring);
1310fac939e9SEnrico Granata             }
1311fac939e9SEnrico Granata         }
1312fac939e9SEnrico Granata         return CommandObjectRaw::GetHelpLong();
1313fac939e9SEnrico Granata     }
1314fac939e9SEnrico Granata 
13155a988416SJim Ingham protected:
13165a988416SJim Ingham     virtual bool
13175a988416SJim Ingham     DoExecute (const char *raw_command_line, CommandReturnObject &result)
1318223383edSEnrico Granata     {
1319223383edSEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1320223383edSEnrico Granata 
1321223383edSEnrico Granata         Error error;
1322223383edSEnrico Granata 
132370f11f88SJim Ingham         result.SetStatus(eReturnStatusInvalid);
132470f11f88SJim Ingham 
1325223383edSEnrico Granata         if (!scripter || scripter->RunScriptBasedCommand(m_function_name.c_str(),
1326223383edSEnrico Granata                                                          raw_command_line,
13270a305db7SEnrico Granata                                                          m_synchro,
1328223383edSEnrico Granata                                                          result,
1329223383edSEnrico Granata                                                          error) == false)
1330223383edSEnrico Granata         {
1331223383edSEnrico Granata             result.AppendError(error.AsCString());
1332223383edSEnrico Granata             result.SetStatus(eReturnStatusFailed);
1333223383edSEnrico Granata         }
1334223383edSEnrico Granata         else
133570f11f88SJim Ingham         {
133670f11f88SJim Ingham             // Don't change the status if the command already set it...
133770f11f88SJim Ingham             if (result.GetStatus() == eReturnStatusInvalid)
133870f11f88SJim Ingham             {
13399a71a7d8SDaniel Malea                 if (result.GetOutputData() == NULL || result.GetOutputData()[0] == '\0')
1340223383edSEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
134170f11f88SJim Ingham                 else
134270f11f88SJim Ingham                     result.SetStatus(eReturnStatusSuccessFinishResult);
134370f11f88SJim Ingham             }
134470f11f88SJim Ingham         }
1345223383edSEnrico Granata 
1346223383edSEnrico Granata         return result.Succeeded();
1347223383edSEnrico Granata     }
1348223383edSEnrico Granata 
1349223383edSEnrico Granata };
1350223383edSEnrico Granata 
1351a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1352a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1353a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1354a9dbf432SEnrico Granata 
13555a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed
1356a9dbf432SEnrico Granata {
13575a988416SJim Ingham public:
13585a988416SJim Ingham     CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) :
13595a988416SJim Ingham         CommandObjectParsed (interpreter,
13605a988416SJim Ingham                              "command script import",
13615a988416SJim Ingham                              "Import a scripting module in LLDB.",
13625a988416SJim Ingham                              NULL),
13635a988416SJim Ingham         m_options(interpreter)
13645a988416SJim Ingham     {
13655a988416SJim Ingham         CommandArgumentEntry arg1;
13665a988416SJim Ingham         CommandArgumentData cmd_arg;
13675a988416SJim Ingham 
13685a988416SJim Ingham         // Define the first (and only) variant of this arg.
13695a988416SJim Ingham         cmd_arg.arg_type = eArgTypeFilename;
13705a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
13715a988416SJim Ingham 
13725a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
13735a988416SJim Ingham         arg1.push_back (cmd_arg);
13745a988416SJim Ingham 
13755a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
13765a988416SJim Ingham         m_arguments.push_back (arg1);
13775a988416SJim Ingham     }
13785a988416SJim Ingham 
13795a988416SJim Ingham     ~CommandObjectCommandsScriptImport ()
13805a988416SJim Ingham     {
13815a988416SJim Ingham     }
13825a988416SJim Ingham 
1383c7bece56SGreg Clayton     virtual int
13845a988416SJim Ingham     HandleArgumentCompletion (Args &input,
13855a988416SJim Ingham                               int &cursor_index,
13865a988416SJim Ingham                               int &cursor_char_position,
13875a988416SJim Ingham                               OptionElementVector &opt_element_vector,
13885a988416SJim Ingham                               int match_start_point,
13895a988416SJim Ingham                               int max_return_elements,
13905a988416SJim Ingham                               bool &word_complete,
13915a988416SJim Ingham                               StringList &matches)
13925a988416SJim Ingham     {
13935a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
13945a988416SJim Ingham         completion_str.erase (cursor_char_position);
13955a988416SJim Ingham 
13965a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
13975a988416SJim Ingham                                                              CommandCompletions::eDiskFileCompletion,
13985a988416SJim Ingham                                                              completion_str.c_str(),
13995a988416SJim Ingham                                                              match_start_point,
14005a988416SJim Ingham                                                              max_return_elements,
14015a988416SJim Ingham                                                              NULL,
14025a988416SJim Ingham                                                              word_complete,
14035a988416SJim Ingham                                                              matches);
14045a988416SJim Ingham         return matches.GetSize();
14055a988416SJim Ingham     }
14065a988416SJim Ingham 
14075a988416SJim Ingham     virtual Options *
14085a988416SJim Ingham     GetOptions ()
14095a988416SJim Ingham     {
14105a988416SJim Ingham         return &m_options;
14115a988416SJim Ingham     }
14125a988416SJim Ingham 
14135a988416SJim Ingham protected:
14140a305db7SEnrico Granata 
14150a305db7SEnrico Granata     class CommandOptions : public Options
14160a305db7SEnrico Granata     {
14170a305db7SEnrico Granata     public:
14180a305db7SEnrico Granata 
14190a305db7SEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
14200a305db7SEnrico Granata             Options (interpreter)
14210a305db7SEnrico Granata         {
14220a305db7SEnrico Granata         }
14230a305db7SEnrico Granata 
14240a305db7SEnrico Granata         virtual
14250a305db7SEnrico Granata         ~CommandOptions (){}
14260a305db7SEnrico Granata 
14270a305db7SEnrico Granata         virtual Error
14280a305db7SEnrico Granata         SetOptionValue (uint32_t option_idx, const char *option_arg)
14290a305db7SEnrico Granata         {
14300a305db7SEnrico Granata             Error error;
14313bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
14320a305db7SEnrico Granata 
14330a305db7SEnrico Granata             switch (short_option)
14340a305db7SEnrico Granata             {
14350a305db7SEnrico Granata                 case 'r':
14360a305db7SEnrico Granata                     m_allow_reload = true;
14370a305db7SEnrico Granata                     break;
14380a305db7SEnrico Granata                 default:
14390a305db7SEnrico Granata                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
14400a305db7SEnrico Granata                     break;
14410a305db7SEnrico Granata             }
14420a305db7SEnrico Granata 
14430a305db7SEnrico Granata             return error;
14440a305db7SEnrico Granata         }
14450a305db7SEnrico Granata 
14460a305db7SEnrico Granata         void
14470a305db7SEnrico Granata         OptionParsingStarting ()
14480a305db7SEnrico Granata         {
1449e0c70f1bSEnrico Granata             m_allow_reload = true;
14500a305db7SEnrico Granata         }
14510a305db7SEnrico Granata 
14520a305db7SEnrico Granata         const OptionDefinition*
14530a305db7SEnrico Granata         GetDefinitions ()
14540a305db7SEnrico Granata         {
14550a305db7SEnrico Granata             return g_option_table;
14560a305db7SEnrico Granata         }
14570a305db7SEnrico Granata 
14580a305db7SEnrico Granata         // Options table: Required for subclasses of Options.
14590a305db7SEnrico Granata 
14600a305db7SEnrico Granata         static OptionDefinition g_option_table[];
14610a305db7SEnrico Granata 
14620a305db7SEnrico Granata         // Instance variables to hold the values for command options.
14630a305db7SEnrico Granata 
14640a305db7SEnrico Granata         bool m_allow_reload;
14650a305db7SEnrico Granata     };
14660a305db7SEnrico Granata 
1467a9dbf432SEnrico Granata     bool
14685a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1469a9dbf432SEnrico Granata     {
1470a9dbf432SEnrico Granata 
1471a9dbf432SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
1472a9dbf432SEnrico Granata         {
1473a9dbf432SEnrico Granata             result.AppendError ("only scripting language supported for module importing is currently Python");
1474a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1475a9dbf432SEnrico Granata             return false;
1476a9dbf432SEnrico Granata         }
1477a9dbf432SEnrico Granata 
14785a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1479a9dbf432SEnrico Granata 
1480a9dbf432SEnrico Granata         if (argc != 1)
1481a9dbf432SEnrico Granata         {
1482a9dbf432SEnrico Granata             result.AppendError ("'command script import' requires one argument");
1483a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1484a9dbf432SEnrico Granata             return false;
1485a9dbf432SEnrico Granata         }
1486a9dbf432SEnrico Granata 
14875a988416SJim Ingham         std::string path = command.GetArgumentAtIndex(0);
1488a9dbf432SEnrico Granata         Error error;
1489a9dbf432SEnrico Granata 
1490c9d645d3SGreg Clayton         const bool init_session = true;
1491078551c7SEnrico Granata         // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that
1492078551c7SEnrico Granata         // commands won't ever be recursively invoked, but it's actually possible to craft
1493078551c7SEnrico Granata         // a Python script that does other "command script imports" in __lldb_init_module
1494078551c7SEnrico Granata         // the real fix is to have recursive commands possible with a CommandInvocation object
1495078551c7SEnrico Granata         // separate from the CommandObject itself, so that recursive command invocations
1496078551c7SEnrico Granata         // won't stomp on each other (wrt to execution contents, options, and more)
1497078551c7SEnrico Granata         m_exe_ctx.Clear();
1498a9dbf432SEnrico Granata         if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(),
14990a305db7SEnrico Granata                                                                       m_options.m_allow_reload,
1500c9d645d3SGreg Clayton                                                                       init_session,
1501a9dbf432SEnrico Granata                                                                       error))
1502a9dbf432SEnrico Granata         {
1503a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishNoResult);
1504a9dbf432SEnrico Granata         }
1505a9dbf432SEnrico Granata         else
1506a9dbf432SEnrico Granata         {
1507a9dbf432SEnrico Granata             result.AppendErrorWithFormat("module importing failed: %s", error.AsCString());
1508a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1509a9dbf432SEnrico Granata         }
1510a9dbf432SEnrico Granata 
1511a9dbf432SEnrico Granata         return result.Succeeded();
1512a9dbf432SEnrico Granata     }
15130a305db7SEnrico Granata 
15145a988416SJim Ingham     CommandOptions m_options;
1515a9dbf432SEnrico Granata };
1516223383edSEnrico Granata 
15170a305db7SEnrico Granata OptionDefinition
15180a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] =
15190a305db7SEnrico Granata {
1520*e2607b50SVirgile Bello     { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, 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."},
15210a305db7SEnrico Granata     { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL }
15220a305db7SEnrico Granata };
15230a305db7SEnrico Granata 
15240a305db7SEnrico Granata 
1525223383edSEnrico Granata //-------------------------------------------------------------------------
1526223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1527223383edSEnrico Granata //-------------------------------------------------------------------------
1528223383edSEnrico Granata 
15295a988416SJim Ingham class CommandObjectCommandsScriptAdd : public CommandObjectParsed
1530223383edSEnrico Granata {
15315a988416SJim Ingham public:
15325a988416SJim Ingham     CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) :
15335a988416SJim Ingham         CommandObjectParsed (interpreter,
15345a988416SJim Ingham                              "command script add",
15355a988416SJim Ingham                              "Add a scripted function as an LLDB command.",
15365a988416SJim Ingham                              NULL),
15375a988416SJim Ingham         m_options (interpreter)
15385a988416SJim Ingham     {
15395a988416SJim Ingham         CommandArgumentEntry arg1;
15405a988416SJim Ingham         CommandArgumentData cmd_arg;
15415a988416SJim Ingham 
15425a988416SJim Ingham         // Define the first (and only) variant of this arg.
15435a988416SJim Ingham         cmd_arg.arg_type = eArgTypeCommandName;
15445a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
15455a988416SJim Ingham 
15465a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
15475a988416SJim Ingham         arg1.push_back (cmd_arg);
15485a988416SJim Ingham 
15495a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
15505a988416SJim Ingham         m_arguments.push_back (arg1);
15515a988416SJim Ingham     }
15525a988416SJim Ingham 
15535a988416SJim Ingham     ~CommandObjectCommandsScriptAdd ()
15545a988416SJim Ingham     {
15555a988416SJim Ingham     }
15565a988416SJim Ingham 
15575a988416SJim Ingham     virtual Options *
15585a988416SJim Ingham     GetOptions ()
15595a988416SJim Ingham     {
15605a988416SJim Ingham         return &m_options;
15615a988416SJim Ingham     }
15625a988416SJim Ingham 
15635a988416SJim Ingham protected:
1564223383edSEnrico Granata 
1565223383edSEnrico Granata     class CommandOptions : public Options
1566223383edSEnrico Granata     {
1567223383edSEnrico Granata     public:
1568223383edSEnrico Granata 
1569223383edSEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
1570223383edSEnrico Granata         Options (interpreter)
1571223383edSEnrico Granata         {
1572223383edSEnrico Granata         }
1573223383edSEnrico Granata 
1574223383edSEnrico Granata         virtual
1575223383edSEnrico Granata         ~CommandOptions (){}
1576223383edSEnrico Granata 
1577223383edSEnrico Granata         virtual Error
1578223383edSEnrico Granata         SetOptionValue (uint32_t option_idx, const char *option_arg)
1579223383edSEnrico Granata         {
1580223383edSEnrico Granata             Error error;
15813bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
1582223383edSEnrico Granata 
1583223383edSEnrico Granata             switch (short_option)
1584223383edSEnrico Granata             {
1585223383edSEnrico Granata                 case 'f':
1586223383edSEnrico Granata                     m_funct_name = std::string(option_arg);
1587223383edSEnrico Granata                     break;
15880a305db7SEnrico Granata                 case 's':
15890a305db7SEnrico Granata                     m_synchronous = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error);
15900a305db7SEnrico Granata                     if (!error.Success())
15910a305db7SEnrico Granata                         error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg);
15920a305db7SEnrico Granata                     break;
1593223383edSEnrico Granata                 default:
159486edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1595223383edSEnrico Granata                     break;
1596223383edSEnrico Granata             }
1597223383edSEnrico Granata 
1598223383edSEnrico Granata             return error;
1599223383edSEnrico Granata         }
1600223383edSEnrico Granata 
1601223383edSEnrico Granata         void
1602223383edSEnrico Granata         OptionParsingStarting ()
1603223383edSEnrico Granata         {
1604223383edSEnrico Granata             m_funct_name = "";
16050a305db7SEnrico Granata             m_synchronous = eScriptedCommandSynchronicitySynchronous;
1606223383edSEnrico Granata         }
1607223383edSEnrico Granata 
1608223383edSEnrico Granata         const OptionDefinition*
1609223383edSEnrico Granata         GetDefinitions ()
1610223383edSEnrico Granata         {
1611223383edSEnrico Granata             return g_option_table;
1612223383edSEnrico Granata         }
1613223383edSEnrico Granata 
1614223383edSEnrico Granata         // Options table: Required for subclasses of Options.
1615223383edSEnrico Granata 
1616223383edSEnrico Granata         static OptionDefinition g_option_table[];
1617223383edSEnrico Granata 
1618223383edSEnrico Granata         // Instance variables to hold the values for command options.
1619223383edSEnrico Granata 
1620223383edSEnrico Granata         std::string m_funct_name;
16210a305db7SEnrico Granata         ScriptedCommandSynchronicity m_synchronous;
1622223383edSEnrico Granata     };
1623223383edSEnrico Granata 
16245a988416SJim Ingham private:
1625223383edSEnrico Granata     class PythonAliasReader : public InputReaderEZ
1626223383edSEnrico Granata     {
1627223383edSEnrico Granata     private:
1628223383edSEnrico Granata         CommandInterpreter& m_interpreter;
1629223383edSEnrico Granata         std::string m_cmd_name;
16300a305db7SEnrico Granata         ScriptedCommandSynchronicity m_synchronous;
1631223383edSEnrico Granata         StringList m_user_input;
1632223383edSEnrico Granata         DISALLOW_COPY_AND_ASSIGN (PythonAliasReader);
1633223383edSEnrico Granata     public:
1634223383edSEnrico Granata         PythonAliasReader(Debugger& debugger,
1635223383edSEnrico Granata                           CommandInterpreter& interpreter,
16360a305db7SEnrico Granata                           std::string cmd_name,
16370a305db7SEnrico Granata                           ScriptedCommandSynchronicity synch) :
1638223383edSEnrico Granata         InputReaderEZ(debugger),
1639223383edSEnrico Granata         m_interpreter(interpreter),
1640223383edSEnrico Granata         m_cmd_name(cmd_name),
16410a305db7SEnrico Granata         m_synchronous(synch),
1642223383edSEnrico Granata         m_user_input()
1643223383edSEnrico Granata         {}
1644223383edSEnrico Granata 
1645223383edSEnrico Granata         virtual
1646223383edSEnrico Granata         ~PythonAliasReader()
1647223383edSEnrico Granata         {
1648223383edSEnrico Granata         }
1649223383edSEnrico Granata 
1650223383edSEnrico Granata         virtual void ActivateHandler(HandlerData& data)
1651223383edSEnrico Granata         {
1652223383edSEnrico Granata             StreamSP out_stream = data.GetOutStream();
1653223383edSEnrico Granata             bool batch_mode = data.GetBatchMode();
1654223383edSEnrico Granata             if (!batch_mode)
1655223383edSEnrico Granata             {
1656223383edSEnrico Granata                 out_stream->Printf ("%s\n", g_python_command_instructions);
1657223383edSEnrico Granata                 if (data.reader.GetPrompt())
1658223383edSEnrico Granata                     out_stream->Printf ("%s", data.reader.GetPrompt());
1659223383edSEnrico Granata                 out_stream->Flush();
1660223383edSEnrico Granata             }
1661223383edSEnrico Granata         }
1662223383edSEnrico Granata 
1663223383edSEnrico Granata         virtual void ReactivateHandler(HandlerData& data)
1664223383edSEnrico Granata         {
1665223383edSEnrico Granata             StreamSP out_stream = data.GetOutStream();
1666223383edSEnrico Granata             bool batch_mode = data.GetBatchMode();
1667223383edSEnrico Granata             if (data.reader.GetPrompt() && !batch_mode)
1668223383edSEnrico Granata             {
1669223383edSEnrico Granata                 out_stream->Printf ("%s", data.reader.GetPrompt());
1670223383edSEnrico Granata                 out_stream->Flush();
1671223383edSEnrico Granata             }
1672223383edSEnrico Granata         }
1673223383edSEnrico Granata         virtual void GotTokenHandler(HandlerData& data)
1674223383edSEnrico Granata         {
1675223383edSEnrico Granata             StreamSP out_stream = data.GetOutStream();
1676223383edSEnrico Granata             bool batch_mode = data.GetBatchMode();
1677223383edSEnrico Granata             if (data.bytes && data.bytes_len)
1678223383edSEnrico Granata             {
1679223383edSEnrico Granata                 m_user_input.AppendString(data.bytes, data.bytes_len);
1680223383edSEnrico Granata             }
1681223383edSEnrico Granata             if (!data.reader.IsDone() && data.reader.GetPrompt() && !batch_mode)
1682223383edSEnrico Granata             {
1683223383edSEnrico Granata                 out_stream->Printf ("%s", data.reader.GetPrompt());
1684223383edSEnrico Granata                 out_stream->Flush();
1685223383edSEnrico Granata             }
1686223383edSEnrico Granata         }
1687223383edSEnrico Granata         virtual void InterruptHandler(HandlerData& data)
1688223383edSEnrico Granata         {
1689223383edSEnrico Granata             StreamSP out_stream = data.GetOutStream();
1690223383edSEnrico Granata             bool batch_mode = data.GetBatchMode();
1691223383edSEnrico Granata             data.reader.SetIsDone (true);
1692223383edSEnrico Granata             if (!batch_mode)
1693223383edSEnrico Granata             {
16940a305db7SEnrico Granata                 out_stream->Printf ("Warning: No script attached.\n");
1695223383edSEnrico Granata                 out_stream->Flush();
1696223383edSEnrico Granata             }
1697223383edSEnrico Granata         }
1698223383edSEnrico Granata         virtual void EOFHandler(HandlerData& data)
1699223383edSEnrico Granata         {
1700223383edSEnrico Granata             data.reader.SetIsDone (true);
1701223383edSEnrico Granata         }
1702223383edSEnrico Granata         virtual void DoneHandler(HandlerData& data)
1703223383edSEnrico Granata         {
1704223383edSEnrico Granata             StreamSP out_stream = data.GetOutStream();
1705223383edSEnrico Granata 
1706223383edSEnrico Granata             ScriptInterpreter *interpreter = data.reader.GetDebugger().GetCommandInterpreter().GetScriptInterpreter();
1707223383edSEnrico Granata             if (!interpreter)
1708223383edSEnrico Granata             {
17090a305db7SEnrico Granata                 out_stream->Printf ("Script interpreter missing: no script attached.\n");
1710223383edSEnrico Granata                 out_stream->Flush();
1711223383edSEnrico Granata                 return;
1712223383edSEnrico Granata             }
1713a73b7df7SEnrico Granata             std::string funct_name_str;
1714223383edSEnrico Granata             if (!interpreter->GenerateScriptAliasFunction (m_user_input,
1715a73b7df7SEnrico Granata                                                            funct_name_str))
1716223383edSEnrico Granata             {
17170a305db7SEnrico Granata                 out_stream->Printf ("Unable to create function: no script attached.\n");
1718223383edSEnrico Granata                 out_stream->Flush();
1719223383edSEnrico Granata                 return;
1720223383edSEnrico Granata             }
1721a73b7df7SEnrico Granata             if (funct_name_str.empty())
1722223383edSEnrico Granata             {
17230a305db7SEnrico Granata                 out_stream->Printf ("Unable to obtain a function name: no script attached.\n");
1724223383edSEnrico Granata                 out_stream->Flush();
1725223383edSEnrico Granata                 return;
1726223383edSEnrico Granata             }
1727223383edSEnrico Granata             // everything should be fine now, let's add this alias
1728223383edSEnrico Granata 
1729223383edSEnrico Granata             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(m_interpreter,
1730223383edSEnrico Granata                                                                            m_cmd_name,
1731a73b7df7SEnrico Granata                                                                            funct_name_str.c_str(),
17320a305db7SEnrico Granata                                                                            m_synchronous));
1733223383edSEnrico Granata 
17340a305db7SEnrico Granata             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true))
1735223383edSEnrico Granata             {
17360a305db7SEnrico Granata                 out_stream->Printf ("Unable to add selected command: no script attached.\n");
1737223383edSEnrico Granata                 out_stream->Flush();
1738223383edSEnrico Granata                 return;
1739223383edSEnrico Granata             }
1740223383edSEnrico Granata         }
1741223383edSEnrico Granata     };
1742223383edSEnrico Granata 
17435a988416SJim Ingham protected:
1744223383edSEnrico Granata     bool
17455a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1746223383edSEnrico Granata     {
174799f0b8f9SEnrico Granata 
174899f0b8f9SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
174999f0b8f9SEnrico Granata         {
175099f0b8f9SEnrico Granata             result.AppendError ("only scripting language supported for scripted commands is currently Python");
175199f0b8f9SEnrico Granata             result.SetStatus (eReturnStatusFailed);
175299f0b8f9SEnrico Granata             return false;
175399f0b8f9SEnrico Granata         }
175499f0b8f9SEnrico Granata 
17555a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1756223383edSEnrico Granata 
1757223383edSEnrico Granata         if (argc != 1)
1758223383edSEnrico Granata         {
1759223383edSEnrico Granata             result.AppendError ("'command script add' requires one argument");
1760223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1761223383edSEnrico Granata             return false;
1762223383edSEnrico Granata         }
1763223383edSEnrico Granata 
17645a988416SJim Ingham         std::string cmd_name = command.GetArgumentAtIndex(0);
1765223383edSEnrico Granata 
1766223383edSEnrico Granata         if (m_options.m_funct_name.empty())
1767223383edSEnrico Granata         {
1768223383edSEnrico Granata             InputReaderSP reader_sp (new PythonAliasReader (m_interpreter.GetDebugger(),
1769223383edSEnrico Granata                                                             m_interpreter,
17700a305db7SEnrico Granata                                                             cmd_name,
17710a305db7SEnrico Granata                                                             m_options.m_synchronous));
1772223383edSEnrico Granata 
1773223383edSEnrico Granata             if (reader_sp)
1774223383edSEnrico Granata             {
1775223383edSEnrico Granata 
1776223383edSEnrico Granata                 InputReaderEZ::InitializationParameters ipr;
1777223383edSEnrico Granata 
1778223383edSEnrico Granata                 Error err (reader_sp->Initialize (ipr.SetBaton(NULL).SetPrompt("     ")));
1779223383edSEnrico Granata                 if (err.Success())
1780223383edSEnrico Granata                 {
1781223383edSEnrico Granata                     m_interpreter.GetDebugger().PushInputReader (reader_sp);
1782223383edSEnrico Granata                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1783223383edSEnrico Granata                 }
1784223383edSEnrico Granata                 else
1785223383edSEnrico Granata                 {
1786223383edSEnrico Granata                     result.AppendError (err.AsCString());
1787223383edSEnrico Granata                     result.SetStatus (eReturnStatusFailed);
1788223383edSEnrico Granata                 }
1789223383edSEnrico Granata             }
1790223383edSEnrico Granata             else
1791223383edSEnrico Granata             {
1792223383edSEnrico Granata                 result.AppendError("out of memory");
1793223383edSEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1794223383edSEnrico Granata             }
1795223383edSEnrico Granata         }
1796223383edSEnrico Granata         else
1797223383edSEnrico Granata         {
17980a305db7SEnrico Granata             CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter,
17990a305db7SEnrico Granata                                                                     cmd_name,
18000a305db7SEnrico Granata                                                                     m_options.m_funct_name,
18010a305db7SEnrico Granata                                                                     m_options.m_synchronous));
18020a305db7SEnrico Granata             if (m_interpreter.AddUserCommand(cmd_name, new_cmd, true))
1803223383edSEnrico Granata             {
1804223383edSEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
1805223383edSEnrico Granata             }
1806223383edSEnrico Granata             else
1807223383edSEnrico Granata             {
1808223383edSEnrico Granata                 result.AppendError("cannot add command");
1809223383edSEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1810223383edSEnrico Granata             }
1811223383edSEnrico Granata         }
1812223383edSEnrico Granata 
1813223383edSEnrico Granata         return result.Succeeded();
1814223383edSEnrico Granata 
1815223383edSEnrico Granata     }
18165a988416SJim Ingham 
18175a988416SJim Ingham     CommandOptions m_options;
1818223383edSEnrico Granata };
1819223383edSEnrico Granata 
18200a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] =
18210a305db7SEnrico Granata {
18220a305db7SEnrico Granata     { eScriptedCommandSynchronicitySynchronous,      "synchronous",       "Run synchronous"},
18230a305db7SEnrico Granata     { eScriptedCommandSynchronicityAsynchronous,     "asynchronous",      "Run asynchronous"},
18240a305db7SEnrico Granata     { eScriptedCommandSynchronicityCurrentValue,     "current",           "Do not alter current setting"},
18250a305db7SEnrico Granata     { 0, NULL, NULL }
18260a305db7SEnrico Granata };
18270a305db7SEnrico Granata 
1828223383edSEnrico Granata OptionDefinition
1829223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] =
1830223383edSEnrico Granata {
1831*e2607b50SVirgile Bello     { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, NULL, 0, eArgTypePythonFunction,        "Name of the Python function to bind to this command name."},
1832*e2607b50SVirgile Bello     { LLDB_OPT_SET_1, false, "synchronicity", 's', OptionParser::eRequiredArgument, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity,        "Set the synchronicity of this command's executions with regard to LLDB event system."},
1833223383edSEnrico Granata     { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL }
1834223383edSEnrico Granata };
1835223383edSEnrico Granata 
1836223383edSEnrico Granata //-------------------------------------------------------------------------
1837223383edSEnrico Granata // CommandObjectCommandsScriptList
1838223383edSEnrico Granata //-------------------------------------------------------------------------
1839223383edSEnrico Granata 
18405a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed
1841223383edSEnrico Granata {
1842223383edSEnrico Granata private:
1843223383edSEnrico Granata 
1844223383edSEnrico Granata public:
1845223383edSEnrico Granata     CommandObjectCommandsScriptList(CommandInterpreter &interpreter) :
18465a988416SJim Ingham     CommandObjectParsed (interpreter,
1847223383edSEnrico Granata                    "command script list",
1848223383edSEnrico Granata                    "List defined scripted commands.",
1849223383edSEnrico Granata                    NULL)
1850223383edSEnrico Granata     {
1851223383edSEnrico Granata     }
1852223383edSEnrico Granata 
1853223383edSEnrico Granata     ~CommandObjectCommandsScriptList ()
1854223383edSEnrico Granata     {
1855223383edSEnrico Granata     }
1856223383edSEnrico Granata 
1857223383edSEnrico Granata     bool
18585a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1859223383edSEnrico Granata     {
1860223383edSEnrico Granata 
1861223383edSEnrico Granata         m_interpreter.GetHelp(result,
1862223383edSEnrico Granata                               CommandInterpreter::eCommandTypesUserDef);
1863223383edSEnrico Granata 
1864223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
1865223383edSEnrico Granata 
1866223383edSEnrico Granata         return true;
1867223383edSEnrico Granata 
1868223383edSEnrico Granata 
1869223383edSEnrico Granata     }
1870223383edSEnrico Granata };
1871223383edSEnrico Granata 
1872223383edSEnrico Granata //-------------------------------------------------------------------------
1873223383edSEnrico Granata // CommandObjectCommandsScriptClear
1874223383edSEnrico Granata //-------------------------------------------------------------------------
1875223383edSEnrico Granata 
18765a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed
1877223383edSEnrico Granata {
1878223383edSEnrico Granata private:
1879223383edSEnrico Granata 
1880223383edSEnrico Granata public:
1881223383edSEnrico Granata     CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) :
18825a988416SJim Ingham         CommandObjectParsed (interpreter,
1883223383edSEnrico Granata                              "command script clear",
1884223383edSEnrico Granata                              "Delete all scripted commands.",
1885223383edSEnrico Granata                              NULL)
1886223383edSEnrico Granata     {
1887223383edSEnrico Granata     }
1888223383edSEnrico Granata 
1889223383edSEnrico Granata     ~CommandObjectCommandsScriptClear ()
1890223383edSEnrico Granata     {
1891223383edSEnrico Granata     }
1892223383edSEnrico Granata 
18935a988416SJim Ingham protected:
1894223383edSEnrico Granata     bool
18955a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1896223383edSEnrico Granata     {
1897223383edSEnrico Granata 
1898223383edSEnrico Granata         m_interpreter.RemoveAllUser();
1899223383edSEnrico Granata 
1900223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
1901223383edSEnrico Granata 
1902223383edSEnrico Granata         return true;
1903223383edSEnrico Granata     }
1904223383edSEnrico Granata };
1905223383edSEnrico Granata 
1906223383edSEnrico Granata //-------------------------------------------------------------------------
1907223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1908223383edSEnrico Granata //-------------------------------------------------------------------------
1909223383edSEnrico Granata 
19105a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed
1911223383edSEnrico Granata {
1912223383edSEnrico Granata public:
1913223383edSEnrico Granata     CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) :
19145a988416SJim Ingham         CommandObjectParsed (interpreter,
1915223383edSEnrico Granata                              "command script delete",
1916223383edSEnrico Granata                              "Delete a scripted command.",
1917223383edSEnrico Granata                              NULL)
1918223383edSEnrico Granata     {
1919223383edSEnrico Granata         CommandArgumentEntry arg1;
1920223383edSEnrico Granata         CommandArgumentData cmd_arg;
1921223383edSEnrico Granata 
1922223383edSEnrico Granata         // Define the first (and only) variant of this arg.
1923223383edSEnrico Granata         cmd_arg.arg_type = eArgTypeCommandName;
1924223383edSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlain;
1925223383edSEnrico Granata 
1926223383edSEnrico Granata         // There is only one variant this argument could be; put it into the argument entry.
1927223383edSEnrico Granata         arg1.push_back (cmd_arg);
1928223383edSEnrico Granata 
1929223383edSEnrico Granata         // Push the data for the first argument into the m_arguments vector.
1930223383edSEnrico Granata         m_arguments.push_back (arg1);
1931223383edSEnrico Granata     }
1932223383edSEnrico Granata 
1933223383edSEnrico Granata     ~CommandObjectCommandsScriptDelete ()
1934223383edSEnrico Granata     {
1935223383edSEnrico Granata     }
1936223383edSEnrico Granata 
19375a988416SJim Ingham protected:
1938223383edSEnrico Granata     bool
19395a988416SJim Ingham     DoExecute (Args& command, CommandReturnObject &result)
1940223383edSEnrico Granata     {
1941223383edSEnrico Granata 
19425a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1943223383edSEnrico Granata 
1944223383edSEnrico Granata         if (argc != 1)
1945223383edSEnrico Granata         {
1946223383edSEnrico Granata             result.AppendError ("'command script delete' requires one argument");
1947223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1948223383edSEnrico Granata             return false;
1949223383edSEnrico Granata         }
1950223383edSEnrico Granata 
19515a988416SJim Ingham         const char* cmd_name = command.GetArgumentAtIndex(0);
1952223383edSEnrico Granata 
1953223383edSEnrico Granata         if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name))
1954223383edSEnrico Granata         {
1955223383edSEnrico Granata             m_interpreter.RemoveUser(cmd_name);
1956223383edSEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishResult);
1957223383edSEnrico Granata         }
1958223383edSEnrico Granata         else
1959223383edSEnrico Granata         {
1960223383edSEnrico Granata             result.AppendErrorWithFormat ("command %s not found", cmd_name);
1961223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1962223383edSEnrico Granata         }
1963223383edSEnrico Granata 
1964223383edSEnrico Granata         return result.Succeeded();
1965223383edSEnrico Granata 
1966223383edSEnrico Granata     }
1967223383edSEnrico Granata };
1968223383edSEnrico Granata 
1969223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1970223383edSEnrico Granata 
1971223383edSEnrico Granata //-------------------------------------------------------------------------
1972223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1973223383edSEnrico Granata //-------------------------------------------------------------------------
1974223383edSEnrico Granata 
1975223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword
1976223383edSEnrico Granata {
1977223383edSEnrico Granata public:
1978223383edSEnrico Granata     CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) :
1979223383edSEnrico Granata     CommandObjectMultiword (interpreter,
1980223383edSEnrico Granata                             "command script",
1981223383edSEnrico Granata                             "A set of commands for managing or customizing script commands.",
1982223383edSEnrico Granata                             "command script <subcommand> [<subcommand-options>]")
1983223383edSEnrico Granata     {
1984223383edSEnrico Granata         LoadSubCommand ("add",  CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter)));
1985223383edSEnrico Granata         LoadSubCommand ("delete",   CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter)));
1986223383edSEnrico Granata         LoadSubCommand ("clear", CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter)));
1987223383edSEnrico Granata         LoadSubCommand ("list",   CommandObjectSP (new CommandObjectCommandsScriptList (interpreter)));
1988a9dbf432SEnrico Granata         LoadSubCommand ("import",   CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter)));
1989223383edSEnrico Granata     }
1990223383edSEnrico Granata 
1991223383edSEnrico Granata     ~CommandObjectMultiwordCommandsScript ()
1992223383edSEnrico Granata     {
1993223383edSEnrico Granata     }
1994223383edSEnrico Granata 
1995223383edSEnrico Granata };
1996223383edSEnrico Granata 
1997223383edSEnrico Granata 
1998ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1999ebc09c36SJim Ingham 
2000ebc09c36SJim Ingham //-------------------------------------------------------------------------
2001ebc09c36SJim Ingham // CommandObjectMultiwordCommands
2002ebc09c36SJim Ingham //-------------------------------------------------------------------------
2003ebc09c36SJim Ingham 
2004ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) :
2005a7015092SGreg Clayton     CommandObjectMultiword (interpreter,
20060e5e5a79SGreg Clayton                             "command",
20073f4c09c1SCaroline Tice                             "A set of commands for managing or customizing the debugger commands.",
20080e5e5a79SGreg Clayton                             "command <subcommand> [<subcommand-options>]")
2009ebc09c36SJim Ingham {
2010a7015092SGreg Clayton     LoadSubCommand ("source",  CommandObjectSP (new CommandObjectCommandsSource (interpreter)));
2011a7015092SGreg Clayton     LoadSubCommand ("alias",   CommandObjectSP (new CommandObjectCommandsAlias (interpreter)));
2012a7015092SGreg Clayton     LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter)));
2013de164aaaSGreg Clayton     LoadSubCommand ("regex",   CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter)));
2014a5a97ebeSJim Ingham     LoadSubCommand ("history",   CommandObjectSP (new CommandObjectCommandsHistory (interpreter)));
2015223383edSEnrico Granata     LoadSubCommand ("script",   CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter)));
2016ebc09c36SJim Ingham }
2017ebc09c36SJim Ingham 
2018ebc09c36SJim Ingham CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands ()
2019ebc09c36SJim Ingham {
2020ebc09c36SJim Ingham }
2021ebc09c36SJim Ingham 
2022