1*9390b346SJonas Devlieghere //===-- CommandObjectScript.cpp -------------------------------------------===//
2*9390b346SJonas Devlieghere //
3*9390b346SJonas Devlieghere // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*9390b346SJonas Devlieghere // See https://llvm.org/LICENSE.txt for license information.
5*9390b346SJonas Devlieghere // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*9390b346SJonas Devlieghere //
7*9390b346SJonas Devlieghere //===----------------------------------------------------------------------===//
8*9390b346SJonas Devlieghere 
9*9390b346SJonas Devlieghere #include "CommandObjectScript.h"
10*9390b346SJonas Devlieghere #include "lldb/Core/Debugger.h"
11*9390b346SJonas Devlieghere #include "lldb/DataFormatters/DataVisualization.h"
12*9390b346SJonas Devlieghere #include "lldb/Host/Config.h"
13*9390b346SJonas Devlieghere #include "lldb/Interpreter/CommandInterpreter.h"
14*9390b346SJonas Devlieghere #include "lldb/Interpreter/CommandReturnObject.h"
15*9390b346SJonas Devlieghere #include "lldb/Interpreter/ScriptInterpreter.h"
16*9390b346SJonas Devlieghere #include "lldb/Utility/Args.h"
17*9390b346SJonas Devlieghere 
18*9390b346SJonas Devlieghere using namespace lldb;
19*9390b346SJonas Devlieghere using namespace lldb_private;
20*9390b346SJonas Devlieghere 
21*9390b346SJonas Devlieghere // CommandObjectScript
22*9390b346SJonas Devlieghere 
23*9390b346SJonas Devlieghere CommandObjectScript::CommandObjectScript(CommandInterpreter &interpreter)
24*9390b346SJonas Devlieghere     : CommandObjectRaw(
25*9390b346SJonas Devlieghere           interpreter, "script",
26*9390b346SJonas Devlieghere           "Invoke the script interpreter with provided code and display any "
27*9390b346SJonas Devlieghere           "results.  Start the interactive interpreter if no code is supplied.",
28*9390b346SJonas Devlieghere           "script [<script-code>]") {}
29*9390b346SJonas Devlieghere 
30*9390b346SJonas Devlieghere CommandObjectScript::~CommandObjectScript() {}
31*9390b346SJonas Devlieghere 
32*9390b346SJonas Devlieghere bool CommandObjectScript::DoExecute(llvm::StringRef command,
33*9390b346SJonas Devlieghere                                     CommandReturnObject &result) {
34*9390b346SJonas Devlieghere   if (m_interpreter.GetDebugger().GetScriptLanguage() ==
35*9390b346SJonas Devlieghere       lldb::eScriptLanguageNone) {
36*9390b346SJonas Devlieghere     result.AppendError(
37*9390b346SJonas Devlieghere         "the script-lang setting is set to none - scripting not available");
38*9390b346SJonas Devlieghere     result.SetStatus(eReturnStatusFailed);
39*9390b346SJonas Devlieghere     return false;
40*9390b346SJonas Devlieghere   }
41*9390b346SJonas Devlieghere 
42*9390b346SJonas Devlieghere   ScriptInterpreter *script_interpreter = GetDebugger().GetScriptInterpreter();
43*9390b346SJonas Devlieghere 
44*9390b346SJonas Devlieghere   if (script_interpreter == nullptr) {
45*9390b346SJonas Devlieghere     result.AppendError("no script interpreter");
46*9390b346SJonas Devlieghere     result.SetStatus(eReturnStatusFailed);
47*9390b346SJonas Devlieghere     return false;
48*9390b346SJonas Devlieghere   }
49*9390b346SJonas Devlieghere 
50*9390b346SJonas Devlieghere   // Script might change Python code we use for formatting. Make sure we keep
51*9390b346SJonas Devlieghere   // up to date with it.
52*9390b346SJonas Devlieghere   DataVisualization::ForceUpdate();
53*9390b346SJonas Devlieghere 
54*9390b346SJonas Devlieghere   if (command.empty()) {
55*9390b346SJonas Devlieghere     script_interpreter->ExecuteInterpreterLoop();
56*9390b346SJonas Devlieghere     result.SetStatus(eReturnStatusSuccessFinishNoResult);
57*9390b346SJonas Devlieghere     return result.Succeeded();
58*9390b346SJonas Devlieghere   }
59*9390b346SJonas Devlieghere 
60*9390b346SJonas Devlieghere   // We can do better when reporting the status of one-liner script execution.
61*9390b346SJonas Devlieghere   if (script_interpreter->ExecuteOneLine(command, &result))
62*9390b346SJonas Devlieghere     result.SetStatus(eReturnStatusSuccessFinishNoResult);
63*9390b346SJonas Devlieghere   else
64*9390b346SJonas Devlieghere     result.SetStatus(eReturnStatusFailed);
65*9390b346SJonas Devlieghere 
66*9390b346SJonas Devlieghere   return result.Succeeded();
67*9390b346SJonas Devlieghere }
68