144d93782SGreg Clayton //===-- IOHandler.cpp -------------------------------------------*- C++ -*-===//
244d93782SGreg Clayton //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
644d93782SGreg Clayton //
744d93782SGreg Clayton //===----------------------------------------------------------------------===//
844d93782SGreg Clayton 
92f3df613SZachary Turner #include "lldb/Core/IOHandler.h"
102f3df613SZachary Turner 
11315b6884SEugene Zelenko #ifndef LLDB_DISABLE_CURSES
1227801f4fSBruce Mitchener #include <curses.h>
13315b6884SEugene Zelenko #include <panel.h>
14315b6884SEugene Zelenko #endif
1544d93782SGreg Clayton 
167c9aa073STodd Fiala #if defined(__APPLE__)
177c9aa073STodd Fiala #include <deque>
187c9aa073STodd Fiala #endif
1944d93782SGreg Clayton #include <string>
2044d93782SGreg Clayton 
2144d93782SGreg Clayton #include "lldb/Core/Debugger.h"
2244d93782SGreg Clayton #include "lldb/Core/StreamFile.h"
23672d2c12SJonas Devlieghere #include "lldb/Host/File.h"
24672d2c12SJonas Devlieghere #include "lldb/Utility/Predicate.h"
25672d2c12SJonas Devlieghere #include "lldb/Utility/Status.h"
26672d2c12SJonas Devlieghere #include "lldb/Utility/StreamString.h"
27672d2c12SJonas Devlieghere #include "lldb/Utility/StringList.h"
28672d2c12SJonas Devlieghere #include "lldb/lldb-forward.h"
292f3df613SZachary Turner 
30cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
3144d93782SGreg Clayton #include "lldb/Host/Editline.h"
32cacde7dfSTodd Fiala #endif
3344d93782SGreg Clayton #include "lldb/Interpreter/CommandCompletions.h"
3444d93782SGreg Clayton #include "lldb/Interpreter/CommandInterpreter.h"
352f3df613SZachary Turner #ifndef LLDB_DISABLE_CURSES
362f3df613SZachary Turner #include "lldb/Breakpoint/BreakpointLocation.h"
372f3df613SZachary Turner #include "lldb/Core/Module.h"
382f3df613SZachary Turner #include "lldb/Core/ValueObject.h"
392f3df613SZachary Turner #include "lldb/Core/ValueObjectRegister.h"
4044d93782SGreg Clayton #include "lldb/Symbol/Block.h"
4144d93782SGreg Clayton #include "lldb/Symbol/Function.h"
4244d93782SGreg Clayton #include "lldb/Symbol/Symbol.h"
43c5dac77aSEugene Zelenko #include "lldb/Symbol/VariableList.h"
44c5dac77aSEugene Zelenko #include "lldb/Target/Process.h"
452f3df613SZachary Turner #include "lldb/Target/RegisterContext.h"
46c5dac77aSEugene Zelenko #include "lldb/Target/StackFrame.h"
472f3df613SZachary Turner #include "lldb/Target/StopInfo.h"
48b9c1b51eSKate Stone #include "lldb/Target/Target.h"
49b9c1b51eSKate Stone #include "lldb/Target/Thread.h"
50d821c997SPavel Labath #include "lldb/Utility/State.h"
51c5dac77aSEugene Zelenko #endif
527c9aa073STodd Fiala 
53672d2c12SJonas Devlieghere #include "llvm/ADT/StringRef.h"
542f3df613SZachary Turner 
555c82608dSHaibo Huang #ifdef _WIN32
56aaea8ee6SZachary Turner #include "lldb/Host/windows/windows.h"
57fab31220STed Woodward #endif
58fab31220STed Woodward 
59672d2c12SJonas Devlieghere #include <memory>
60672d2c12SJonas Devlieghere #include <mutex>
612f3df613SZachary Turner 
62672d2c12SJonas Devlieghere #include <assert.h>
63672d2c12SJonas Devlieghere #include <ctype.h>
64672d2c12SJonas Devlieghere #include <errno.h>
65672d2c12SJonas Devlieghere #include <locale.h>
66672d2c12SJonas Devlieghere #include <stdint.h>
67672d2c12SJonas Devlieghere #include <stdio.h>
68672d2c12SJonas Devlieghere #include <string.h>
69672d2c12SJonas Devlieghere #include <type_traits>
702f3df613SZachary Turner 
7144d93782SGreg Clayton using namespace lldb;
7244d93782SGreg Clayton using namespace lldb_private;
73*b3faa01fSLawrence D'Anna using llvm::None;
74*b3faa01fSLawrence D'Anna using llvm::Optional;
75*b3faa01fSLawrence D'Anna using llvm::StringRef;
76*b3faa01fSLawrence D'Anna 
7744d93782SGreg Clayton 
78b9c1b51eSKate Stone IOHandler::IOHandler(Debugger &debugger, IOHandler::Type type)
79b9c1b51eSKate Stone     : IOHandler(debugger, type,
807ca15ba7SLawrence D'Anna                 FileSP(),       // Adopt STDIN from top input reader
8144d93782SGreg Clayton                 StreamFileSP(), // Adopt STDOUT from top input reader
82340b0309SGreg Clayton                 StreamFileSP(), // Adopt STDERR from top input reader
83d77c2e09SJonas Devlieghere                 0,              // Flags
84d77c2e09SJonas Devlieghere                 nullptr         // Shadow file recorder
85d77c2e09SJonas Devlieghere       ) {}
8644d93782SGreg Clayton 
87b9c1b51eSKate Stone IOHandler::IOHandler(Debugger &debugger, IOHandler::Type type,
887ca15ba7SLawrence D'Anna                      const lldb::FileSP &input_sp,
8944d93782SGreg Clayton                      const lldb::StreamFileSP &output_sp,
90d77c2e09SJonas Devlieghere                      const lldb::StreamFileSP &error_sp, uint32_t flags,
91d77c2e09SJonas Devlieghere                      repro::DataRecorder *data_recorder)
92b9c1b51eSKate Stone     : m_debugger(debugger), m_input_sp(input_sp), m_output_sp(output_sp),
93d77c2e09SJonas Devlieghere       m_error_sp(error_sp), m_data_recorder(data_recorder), m_popped(false),
94d77c2e09SJonas Devlieghere       m_flags(flags), m_type(type), m_user_data(nullptr), m_done(false),
95d77c2e09SJonas Devlieghere       m_active(false) {
9644d93782SGreg Clayton   // If any files are not specified, then adopt them from the top input reader.
9744d93782SGreg Clayton   if (!m_input_sp || !m_output_sp || !m_error_sp)
98b9c1b51eSKate Stone     debugger.AdoptTopIOHandlerFilesIfInvalid(m_input_sp, m_output_sp,
9944d93782SGreg Clayton                                              m_error_sp);
10044d93782SGreg Clayton }
10144d93782SGreg Clayton 
102315b6884SEugene Zelenko IOHandler::~IOHandler() = default;
10344d93782SGreg Clayton 
104b9c1b51eSKate Stone int IOHandler::GetInputFD() {
1057ca15ba7SLawrence D'Anna   return (m_input_sp ? m_input_sp->GetDescriptor() : -1);
10644d93782SGreg Clayton }
10744d93782SGreg Clayton 
108b9c1b51eSKate Stone int IOHandler::GetOutputFD() {
109c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetDescriptor() : -1);
11044d93782SGreg Clayton }
11144d93782SGreg Clayton 
112b9c1b51eSKate Stone int IOHandler::GetErrorFD() {
113c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetDescriptor() : -1);
11444d93782SGreg Clayton }
11544d93782SGreg Clayton 
116b9c1b51eSKate Stone FILE *IOHandler::GetInputFILE() {
1177ca15ba7SLawrence D'Anna   return (m_input_sp ? m_input_sp->GetStream() : nullptr);
11844d93782SGreg Clayton }
11944d93782SGreg Clayton 
120b9c1b51eSKate Stone FILE *IOHandler::GetOutputFILE() {
121c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetStream() : nullptr);
12244d93782SGreg Clayton }
12344d93782SGreg Clayton 
124b9c1b51eSKate Stone FILE *IOHandler::GetErrorFILE() {
125c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetStream() : nullptr);
12644d93782SGreg Clayton }
12744d93782SGreg Clayton 
1287ca15ba7SLawrence D'Anna FileSP &IOHandler::GetInputFileSP() { return m_input_sp; }
12944d93782SGreg Clayton 
1307ca15ba7SLawrence D'Anna StreamFileSP &IOHandler::GetOutputStreamFileSP() { return m_output_sp; }
13144d93782SGreg Clayton 
1327ca15ba7SLawrence D'Anna StreamFileSP &IOHandler::GetErrorStreamFileSP() { return m_error_sp; }
13344d93782SGreg Clayton 
134b9c1b51eSKate Stone bool IOHandler::GetIsInteractive() {
1357ca15ba7SLawrence D'Anna   return GetInputFileSP() ? GetInputFileSP()->GetIsInteractive() : false;
136340b0309SGreg Clayton }
137340b0309SGreg Clayton 
138b9c1b51eSKate Stone bool IOHandler::GetIsRealTerminal() {
1397ca15ba7SLawrence D'Anna   return GetInputFileSP() ? GetInputFileSP()->GetIsRealTerminal() : false;
140340b0309SGreg Clayton }
14144d93782SGreg Clayton 
142b9c1b51eSKate Stone void IOHandler::SetPopped(bool b) { m_popped.SetValue(b, eBroadcastOnChange); }
143e30f11d9SKate Stone 
144b9c1b51eSKate Stone void IOHandler::WaitForPop() { m_popped.WaitForValueEqualTo(true); }
145e30f11d9SKate Stone 
146b9c1b51eSKate Stone void IOHandlerStack::PrintAsync(Stream *stream, const char *s, size_t len) {
147b9c1b51eSKate Stone   if (stream) {
14816ff8604SSaleem Abdulrasool     std::lock_guard<std::recursive_mutex> guard(m_mutex);
1494446487dSPavel Labath     if (m_top)
1504446487dSPavel Labath       m_top->PrintAsync(stream, s, len);
1514446487dSPavel Labath   }
1524446487dSPavel Labath }
1534446487dSPavel Labath 
1547a120c8bSZachary Turner IOHandlerConfirm::IOHandlerConfirm(Debugger &debugger, llvm::StringRef prompt,
155b9c1b51eSKate Stone                                    bool default_response)
156b9c1b51eSKate Stone     : IOHandlerEditline(
157b9c1b51eSKate Stone           debugger, IOHandler::Type::Confirm,
158c5dac77aSEugene Zelenko           nullptr, // nullptr editline_name means no history loaded/saved
159514d8cd8SZachary Turner           llvm::StringRef(), // No prompt
160514d8cd8SZachary Turner           llvm::StringRef(), // No continuation prompt
16144d93782SGreg Clayton           false,             // Multi-line
162e30f11d9SKate Stone           false, // Don't colorize the prompt (i.e. the confirm message.)
163d77c2e09SJonas Devlieghere           0, *this, nullptr),
164b9c1b51eSKate Stone       m_default_response(default_response), m_user_response(default_response) {
16544d93782SGreg Clayton   StreamString prompt_stream;
16644d93782SGreg Clayton   prompt_stream.PutCString(prompt);
16744d93782SGreg Clayton   if (m_default_response)
16844d93782SGreg Clayton     prompt_stream.Printf(": [Y/n] ");
16944d93782SGreg Clayton   else
17044d93782SGreg Clayton     prompt_stream.Printf(": [y/N] ");
17144d93782SGreg Clayton 
172514d8cd8SZachary Turner   SetPrompt(prompt_stream.GetString());
17344d93782SGreg Clayton }
17444d93782SGreg Clayton 
175315b6884SEugene Zelenko IOHandlerConfirm::~IOHandlerConfirm() = default;
17644d93782SGreg Clayton 
177ae34ed2cSRaphael Isemann void IOHandlerConfirm::IOHandlerComplete(IOHandler &io_handler,
1782fc20f65SRaphael Isemann                                          CompletionRequest &request) {
1791153dc96SRaphael Isemann   if (request.GetRawCursorPos() != 0)
1801153dc96SRaphael Isemann     return;
1811153dc96SRaphael Isemann   request.AddCompletion(m_default_response ? "y" : "n");
18244d93782SGreg Clayton }
18344d93782SGreg Clayton 
184b9c1b51eSKate Stone void IOHandlerConfirm::IOHandlerInputComplete(IOHandler &io_handler,
185b9c1b51eSKate Stone                                               std::string &line) {
186b9c1b51eSKate Stone   if (line.empty()) {
18744d93782SGreg Clayton     // User just hit enter, set the response to the default
18844d93782SGreg Clayton     m_user_response = m_default_response;
18944d93782SGreg Clayton     io_handler.SetIsDone(true);
19044d93782SGreg Clayton     return;
19144d93782SGreg Clayton   }
19244d93782SGreg Clayton 
193b9c1b51eSKate Stone   if (line.size() == 1) {
194b9c1b51eSKate Stone     switch (line[0]) {
19544d93782SGreg Clayton     case 'y':
19644d93782SGreg Clayton     case 'Y':
19744d93782SGreg Clayton       m_user_response = true;
19844d93782SGreg Clayton       io_handler.SetIsDone(true);
19944d93782SGreg Clayton       return;
20044d93782SGreg Clayton     case 'n':
20144d93782SGreg Clayton     case 'N':
20244d93782SGreg Clayton       m_user_response = false;
20344d93782SGreg Clayton       io_handler.SetIsDone(true);
20444d93782SGreg Clayton       return;
20544d93782SGreg Clayton     default:
20644d93782SGreg Clayton       break;
20744d93782SGreg Clayton     }
20844d93782SGreg Clayton   }
20944d93782SGreg Clayton 
210b9c1b51eSKate Stone   if (line == "yes" || line == "YES" || line == "Yes") {
21144d93782SGreg Clayton     m_user_response = true;
21244d93782SGreg Clayton     io_handler.SetIsDone(true);
213b9c1b51eSKate Stone   } else if (line == "no" || line == "NO" || line == "No") {
21444d93782SGreg Clayton     m_user_response = false;
21544d93782SGreg Clayton     io_handler.SetIsDone(true);
21644d93782SGreg Clayton   }
21744d93782SGreg Clayton }
21844d93782SGreg Clayton 
219ae34ed2cSRaphael Isemann void IOHandlerDelegate::IOHandlerComplete(IOHandler &io_handler,
2202fc20f65SRaphael Isemann                                           CompletionRequest &request) {
221b9c1b51eSKate Stone   switch (m_completion) {
22244d93782SGreg Clayton   case Completion::None:
22344d93782SGreg Clayton     break;
22444d93782SGreg Clayton   case Completion::LLDBCommand:
225ae34ed2cSRaphael Isemann     io_handler.GetDebugger().GetCommandInterpreter().HandleCompletion(request);
226ae34ed2cSRaphael Isemann     break;
227ae34ed2cSRaphael Isemann   case Completion::Expression:
228b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
229b9c1b51eSKate Stone         io_handler.GetDebugger().GetCommandInterpreter(),
230ae34ed2cSRaphael Isemann         CommandCompletions::eVariablePathCompletion, request, nullptr);
231ae34ed2cSRaphael Isemann     break;
23244d93782SGreg Clayton   }
23344d93782SGreg Clayton }
23444d93782SGreg Clayton 
235b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
236b9c1b51eSKate Stone     Debugger &debugger, IOHandler::Type type,
23744d93782SGreg Clayton     const char *editline_name, // Used for saving history files
238514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
239514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
240d77c2e09SJonas Devlieghere     IOHandlerDelegate &delegate, repro::DataRecorder *data_recorder)
241b9c1b51eSKate Stone     : IOHandlerEditline(debugger, type,
2427ca15ba7SLawrence D'Anna                         FileSP(),       // Inherit input from top input reader
24344d93782SGreg Clayton                         StreamFileSP(), // Inherit output from top input reader
24444d93782SGreg Clayton                         StreamFileSP(), // Inherit error from top input reader
245340b0309SGreg Clayton                         0,              // Flags
24644d93782SGreg Clayton                         editline_name,  // Used for saving history files
247b9c1b51eSKate Stone                         prompt, continuation_prompt, multi_line, color_prompts,
248d77c2e09SJonas Devlieghere                         line_number_start, delegate, data_recorder) {}
24944d93782SGreg Clayton 
250b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
2517ca15ba7SLawrence D'Anna     Debugger &debugger, IOHandler::Type type, const lldb::FileSP &input_sp,
2527ca15ba7SLawrence D'Anna     const lldb::StreamFileSP &output_sp, const lldb::StreamFileSP &error_sp,
2537ca15ba7SLawrence D'Anna     uint32_t flags,
25444d93782SGreg Clayton     const char *editline_name, // Used for saving history files
255514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
256514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
257d77c2e09SJonas Devlieghere     IOHandlerDelegate &delegate, repro::DataRecorder *data_recorder)
258d77c2e09SJonas Devlieghere     : IOHandler(debugger, type, input_sp, output_sp, error_sp, flags,
259d77c2e09SJonas Devlieghere                 data_recorder),
260cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
261d5b44036SJonas Devlieghere       m_editline_up(),
262cacde7dfSTodd Fiala #endif
263b9c1b51eSKate Stone       m_delegate(delegate), m_prompt(), m_continuation_prompt(),
264b9c1b51eSKate Stone       m_current_lines_ptr(nullptr), m_base_line_number(line_number_start),
265b9c1b51eSKate Stone       m_curr_line_idx(UINT32_MAX), m_multi_line(multi_line),
266b9c1b51eSKate Stone       m_color_prompts(color_prompts), m_interrupt_exits(true),
267b9c1b51eSKate Stone       m_editing(false) {
26844d93782SGreg Clayton   SetPrompt(prompt);
26944d93782SGreg Clayton 
270cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
271914b8d98SDeepak Panickal   bool use_editline = false;
272340b0309SGreg Clayton 
273609010d0SLawrence D'Anna   use_editline = GetInputFILE() && GetOutputFILE() && GetErrorFILE() &&
274609010d0SLawrence D'Anna                  m_input_sp && m_input_sp->GetIsRealTerminal();
27544d93782SGreg Clayton 
276b9c1b51eSKate Stone   if (use_editline) {
277d5b44036SJonas Devlieghere     m_editline_up.reset(new Editline(editline_name, GetInputFILE(),
278b9c1b51eSKate Stone                                      GetOutputFILE(), GetErrorFILE(),
279e30f11d9SKate Stone                                      m_color_prompts));
280d5b44036SJonas Devlieghere     m_editline_up->SetIsInputCompleteCallback(IsInputCompleteCallback, this);
281d5b44036SJonas Devlieghere     m_editline_up->SetAutoCompleteCallback(AutoCompleteCallback, this);
282e30f11d9SKate Stone     // See if the delegate supports fixing indentation
283e30f11d9SKate Stone     const char *indent_chars = delegate.IOHandlerGetFixIndentationCharacters();
284b9c1b51eSKate Stone     if (indent_chars) {
285b9c1b51eSKate Stone       // The delegate does support indentation, hook it up so when any
28605097246SAdrian Prantl       // indentation character is typed, the delegate gets a chance to fix it
287d5b44036SJonas Devlieghere       m_editline_up->SetFixIndentationCallback(FixIndentationCallback, this,
288b9c1b51eSKate Stone                                                indent_chars);
289e30f11d9SKate Stone     }
29044d93782SGreg Clayton   }
291cacde7dfSTodd Fiala #endif
292e30f11d9SKate Stone   SetBaseLineNumber(m_base_line_number);
293514d8cd8SZachary Turner   SetPrompt(prompt);
294e30f11d9SKate Stone   SetContinuationPrompt(continuation_prompt);
29544d93782SGreg Clayton }
29644d93782SGreg Clayton 
297b9c1b51eSKate Stone IOHandlerEditline::~IOHandlerEditline() {
298cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
299d5b44036SJonas Devlieghere   m_editline_up.reset();
300cacde7dfSTodd Fiala #endif
30144d93782SGreg Clayton }
30244d93782SGreg Clayton 
303b9c1b51eSKate Stone void IOHandlerEditline::Activate() {
304e30f11d9SKate Stone   IOHandler::Activate();
3050affb582SDave Lee   m_delegate.IOHandlerActivated(*this, GetIsInteractive());
306e30f11d9SKate Stone }
307e30f11d9SKate Stone 
308b9c1b51eSKate Stone void IOHandlerEditline::Deactivate() {
309e30f11d9SKate Stone   IOHandler::Deactivate();
310e30f11d9SKate Stone   m_delegate.IOHandlerDeactivated(*this);
311e30f11d9SKate Stone }
312e30f11d9SKate Stone 
313*b3faa01fSLawrence D'Anna // Split out a line from the buffer, if there is a full one to get.
314*b3faa01fSLawrence D'Anna static Optional<std::string> SplitLine(std::string &line_buffer) {
315*b3faa01fSLawrence D'Anna   size_t pos = line_buffer.find('\n');
316*b3faa01fSLawrence D'Anna   if (pos == std::string::npos)
317*b3faa01fSLawrence D'Anna     return None;
318*b3faa01fSLawrence D'Anna   std::string line = StringRef(line_buffer.c_str(), pos).rtrim("\n\r");
319*b3faa01fSLawrence D'Anna   line_buffer = line_buffer.substr(pos + 1);
320*b3faa01fSLawrence D'Anna   return line;
321*b3faa01fSLawrence D'Anna }
322*b3faa01fSLawrence D'Anna 
323*b3faa01fSLawrence D'Anna // If the final line of the file ends without a end-of-line, return
324*b3faa01fSLawrence D'Anna // it as a line anyway.
325*b3faa01fSLawrence D'Anna static Optional<std::string> SplitLineEOF(std::string &line_buffer) {
326*b3faa01fSLawrence D'Anna   if (llvm::all_of(line_buffer, isspace))
327*b3faa01fSLawrence D'Anna     return None;
328*b3faa01fSLawrence D'Anna   std::string line = std::move(line_buffer);
329*b3faa01fSLawrence D'Anna   line_buffer.clear();
330*b3faa01fSLawrence D'Anna   return line;
331*b3faa01fSLawrence D'Anna }
332*b3faa01fSLawrence D'Anna 
333b9c1b51eSKate Stone bool IOHandlerEditline::GetLine(std::string &line, bool &interrupted) {
334cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
335d5b44036SJonas Devlieghere   if (m_editline_up) {
336d77c2e09SJonas Devlieghere     bool b = m_editline_up->GetLine(line, interrupted);
337*b3faa01fSLawrence D'Anna     if (b && m_data_recorder)
338d77c2e09SJonas Devlieghere       m_data_recorder->Record(line, true);
339d77c2e09SJonas Devlieghere     return b;
340*b3faa01fSLawrence D'Anna   }
341cacde7dfSTodd Fiala #endif
342*b3faa01fSLawrence D'Anna 
34344d93782SGreg Clayton   line.clear();
34444d93782SGreg Clayton 
345b9c1b51eSKate Stone   if (GetIsInteractive()) {
346c5dac77aSEugene Zelenko     const char *prompt = nullptr;
347e30f11d9SKate Stone 
348e30f11d9SKate Stone     if (m_multi_line && m_curr_line_idx > 0)
349e30f11d9SKate Stone       prompt = GetContinuationPrompt();
350e30f11d9SKate Stone 
351c5dac77aSEugene Zelenko     if (prompt == nullptr)
352e30f11d9SKate Stone       prompt = GetPrompt();
353e30f11d9SKate Stone 
354b9c1b51eSKate Stone     if (prompt && prompt[0]) {
3555da2bc22SLawrence D'Anna       if (m_output_sp) {
3565da2bc22SLawrence D'Anna         m_output_sp->Printf("%s", prompt);
3575da2bc22SLawrence D'Anna         m_output_sp->Flush();
35844d93782SGreg Clayton       }
35944d93782SGreg Clayton     }
36044d93782SGreg Clayton   }
361*b3faa01fSLawrence D'Anna 
362*b3faa01fSLawrence D'Anna   Optional<std::string> got_line = SplitLine(m_line_buffer);
363*b3faa01fSLawrence D'Anna 
364*b3faa01fSLawrence D'Anna   if (!got_line && !m_input_sp) {
365*b3faa01fSLawrence D'Anna     // No more input file, we are done...
366*b3faa01fSLawrence D'Anna     SetIsDone(true);
367*b3faa01fSLawrence D'Anna     return false;
368*b3faa01fSLawrence D'Anna   }
369*b3faa01fSLawrence D'Anna 
370*b3faa01fSLawrence D'Anna   FILE *in = GetInputFILE();
37144d93782SGreg Clayton   char buffer[256];
372*b3faa01fSLawrence D'Anna 
373*b3faa01fSLawrence D'Anna   if (!got_line && !in && m_input_sp) {
374*b3faa01fSLawrence D'Anna     // there is no FILE*, fall back on just reading bytes from the stream.
375*b3faa01fSLawrence D'Anna     while (!got_line) {
376*b3faa01fSLawrence D'Anna       size_t bytes_read = sizeof(buffer);
377*b3faa01fSLawrence D'Anna       Status error = m_input_sp->Read((void *)buffer, bytes_read);
378*b3faa01fSLawrence D'Anna       if (error.Success() && !bytes_read) {
379*b3faa01fSLawrence D'Anna         got_line = SplitLineEOF(m_line_buffer);
380*b3faa01fSLawrence D'Anna         break;
381*b3faa01fSLawrence D'Anna       }
382*b3faa01fSLawrence D'Anna       if (error.Fail())
383*b3faa01fSLawrence D'Anna         break;
384*b3faa01fSLawrence D'Anna       m_line_buffer += StringRef(buffer, bytes_read);
385*b3faa01fSLawrence D'Anna       got_line = SplitLine(m_line_buffer);
386*b3faa01fSLawrence D'Anna     }
387*b3faa01fSLawrence D'Anna   }
388*b3faa01fSLawrence D'Anna 
389*b3faa01fSLawrence D'Anna   if (!got_line && in) {
390e034a04eSGreg Clayton     m_editing = true;
391*b3faa01fSLawrence D'Anna     while (!got_line) {
392*b3faa01fSLawrence D'Anna       char *r = fgets(buffer, sizeof(buffer), in);
393e7167908SNathan Lanza #ifdef _WIN32
394e7167908SNathan Lanza       // ReadFile on Windows is supposed to set ERROR_OPERATION_ABORTED
395e7167908SNathan Lanza       // according to the docs on MSDN. However, this has evidently been a
396e7167908SNathan Lanza       // known bug since Windows 8. Therefore, we can't detect if a signal
397e7167908SNathan Lanza       // interrupted in the fgets. So pressing ctrl-c causes the repl to end
398e7167908SNathan Lanza       // and the process to exit. A temporary workaround is just to attempt to
399e7167908SNathan Lanza       // fgets twice until this bug is fixed.
400*b3faa01fSLawrence D'Anna       if (r == nullptr)
401*b3faa01fSLawrence D'Anna         r = fgets(buffer, sizeof(buffer), in);
402cb305205SNathan Lanza       // this is the equivalent of EINTR for Windows
403*b3faa01fSLawrence D'Anna       if (r == nullptr && GetLastError() == ERROR_OPERATION_ABORTED)
404cb305205SNathan Lanza         continue;
405e7167908SNathan Lanza #endif
406*b3faa01fSLawrence D'Anna       if (r == nullptr) {
407*b3faa01fSLawrence D'Anna         if (ferror(in) && errno == EINTR)
408*b3faa01fSLawrence D'Anna           continue;
409c9cf5798SGreg Clayton         if (feof(in))
410*b3faa01fSLawrence D'Anna           got_line = SplitLineEOF(m_line_buffer);
41144d93782SGreg Clayton         break;
41244d93782SGreg Clayton       }
413*b3faa01fSLawrence D'Anna       m_line_buffer += buffer;
414*b3faa01fSLawrence D'Anna       got_line = SplitLine(m_line_buffer);
41544d93782SGreg Clayton     }
416e034a04eSGreg Clayton     m_editing = false;
417*b3faa01fSLawrence D'Anna   }
418*b3faa01fSLawrence D'Anna 
419*b3faa01fSLawrence D'Anna   if (got_line) {
420*b3faa01fSLawrence D'Anna     line = got_line.getValue();
421*b3faa01fSLawrence D'Anna     if (m_data_recorder)
422d77c2e09SJonas Devlieghere       m_data_recorder->Record(line, true);
42344d93782SGreg Clayton   }
424*b3faa01fSLawrence D'Anna 
425*b3faa01fSLawrence D'Anna   return (bool)got_line;
42644d93782SGreg Clayton }
42744d93782SGreg Clayton 
428cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
429b9c1b51eSKate Stone bool IOHandlerEditline::IsInputCompleteCallback(Editline *editline,
43044d93782SGreg Clayton                                                 StringList &lines,
431b9c1b51eSKate Stone                                                 void *baton) {
43244d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
433b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerIsInputComplete(*editline_reader,
434b9c1b51eSKate Stone                                                               lines);
435e30f11d9SKate Stone }
436e30f11d9SKate Stone 
437b9c1b51eSKate Stone int IOHandlerEditline::FixIndentationCallback(Editline *editline,
438e30f11d9SKate Stone                                               const StringList &lines,
439e30f11d9SKate Stone                                               int cursor_position,
440b9c1b51eSKate Stone                                               void *baton) {
441e30f11d9SKate Stone   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
442b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerFixIndentation(
443b9c1b51eSKate Stone       *editline_reader, lines, cursor_position);
44444d93782SGreg Clayton }
44544d93782SGreg Clayton 
446ae34ed2cSRaphael Isemann void IOHandlerEditline::AutoCompleteCallback(CompletionRequest &request,
4472fc20f65SRaphael Isemann                                              void *baton) {
44844d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
44944d93782SGreg Clayton   if (editline_reader)
450ae34ed2cSRaphael Isemann     editline_reader->m_delegate.IOHandlerComplete(*editline_reader, request);
45144d93782SGreg Clayton }
452cacde7dfSTodd Fiala #endif
45344d93782SGreg Clayton 
454b9c1b51eSKate Stone const char *IOHandlerEditline::GetPrompt() {
455cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
456d5b44036SJonas Devlieghere   if (m_editline_up) {
457d5b44036SJonas Devlieghere     return m_editline_up->GetPrompt();
458b9c1b51eSKate Stone   } else {
459cacde7dfSTodd Fiala #endif
460cacde7dfSTodd Fiala     if (m_prompt.empty())
461c5dac77aSEugene Zelenko       return nullptr;
462cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
463cacde7dfSTodd Fiala   }
464cacde7dfSTodd Fiala #endif
46544d93782SGreg Clayton   return m_prompt.c_str();
46644d93782SGreg Clayton }
46744d93782SGreg Clayton 
468514d8cd8SZachary Turner bool IOHandlerEditline::SetPrompt(llvm::StringRef prompt) {
469514d8cd8SZachary Turner   m_prompt = prompt;
470514d8cd8SZachary Turner 
471cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
472d5b44036SJonas Devlieghere   if (m_editline_up)
473d5b44036SJonas Devlieghere     m_editline_up->SetPrompt(m_prompt.empty() ? nullptr : m_prompt.c_str());
474cacde7dfSTodd Fiala #endif
47544d93782SGreg Clayton   return true;
47644d93782SGreg Clayton }
47744d93782SGreg Clayton 
478b9c1b51eSKate Stone const char *IOHandlerEditline::GetContinuationPrompt() {
479b9c1b51eSKate Stone   return (m_continuation_prompt.empty() ? nullptr
480b9c1b51eSKate Stone                                         : m_continuation_prompt.c_str());
481e30f11d9SKate Stone }
482e30f11d9SKate Stone 
483514d8cd8SZachary Turner void IOHandlerEditline::SetContinuationPrompt(llvm::StringRef prompt) {
484514d8cd8SZachary Turner   m_continuation_prompt = prompt;
485e30f11d9SKate Stone 
486d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
487d5b44036SJonas Devlieghere   if (m_editline_up)
488d5b44036SJonas Devlieghere     m_editline_up->SetContinuationPrompt(m_continuation_prompt.empty()
489b9c1b51eSKate Stone                                              ? nullptr
490b9c1b51eSKate Stone                                              : m_continuation_prompt.c_str());
491d553d00cSZachary Turner #endif
492e30f11d9SKate Stone }
493e30f11d9SKate Stone 
494b9c1b51eSKate Stone void IOHandlerEditline::SetBaseLineNumber(uint32_t line) {
495f6913cd7SGreg Clayton   m_base_line_number = line;
496f6913cd7SGreg Clayton }
497e30f11d9SKate Stone 
498b9c1b51eSKate Stone uint32_t IOHandlerEditline::GetCurrentLineIndex() const {
499d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
500d5b44036SJonas Devlieghere   if (m_editline_up)
501d5b44036SJonas Devlieghere     return m_editline_up->GetCurrentLine();
502e30f11d9SKate Stone #endif
503e30f11d9SKate Stone   return m_curr_line_idx;
504e30f11d9SKate Stone }
505e30f11d9SKate Stone 
506b9c1b51eSKate Stone bool IOHandlerEditline::GetLines(StringList &lines, bool &interrupted) {
507e30f11d9SKate Stone   m_current_lines_ptr = &lines;
508e30f11d9SKate Stone 
50944d93782SGreg Clayton   bool success = false;
510cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
511d5b44036SJonas Devlieghere   if (m_editline_up) {
512d5b44036SJonas Devlieghere     return m_editline_up->GetLines(m_base_line_number, lines, interrupted);
513b9c1b51eSKate Stone   } else {
514cacde7dfSTodd Fiala #endif
515e30f11d9SKate Stone     bool done = false;
51697206d57SZachary Turner     Status error;
51744d93782SGreg Clayton 
518b9c1b51eSKate Stone     while (!done) {
519f6913cd7SGreg Clayton       // Show line numbers if we are asked to
52044d93782SGreg Clayton       std::string line;
521b9c1b51eSKate Stone       if (m_base_line_number > 0 && GetIsInteractive()) {
5225da2bc22SLawrence D'Anna         if (m_output_sp) {
5235da2bc22SLawrence D'Anna           m_output_sp->Printf("%u%s",
5245da2bc22SLawrence D'Anna                               m_base_line_number + (uint32_t)lines.GetSize(),
525b9c1b51eSKate Stone                               GetPrompt() == nullptr ? " " : "");
526f6913cd7SGreg Clayton         }
5275da2bc22SLawrence D'Anna       }
528f6913cd7SGreg Clayton 
529e30f11d9SKate Stone       m_curr_line_idx = lines.GetSize();
530e30f11d9SKate Stone 
531f0066ad0SGreg Clayton       bool interrupted = false;
532b9c1b51eSKate Stone       if (GetLine(line, interrupted) && !interrupted) {
53344d93782SGreg Clayton         lines.AppendString(line);
534e30f11d9SKate Stone         done = m_delegate.IOHandlerIsInputComplete(*this, lines);
535b9c1b51eSKate Stone       } else {
536e30f11d9SKate Stone         done = true;
53744d93782SGreg Clayton       }
53844d93782SGreg Clayton     }
53944d93782SGreg Clayton     success = lines.GetSize() > 0;
540cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
54144d93782SGreg Clayton   }
542cacde7dfSTodd Fiala #endif
54344d93782SGreg Clayton   return success;
54444d93782SGreg Clayton }
54544d93782SGreg Clayton 
54605097246SAdrian Prantl // Each IOHandler gets to run until it is done. It should read data from the
54705097246SAdrian Prantl // "in" and place output into "out" and "err and return when done.
548b9c1b51eSKate Stone void IOHandlerEditline::Run() {
54944d93782SGreg Clayton   std::string line;
550b9c1b51eSKate Stone   while (IsActive()) {
551f0066ad0SGreg Clayton     bool interrupted = false;
552b9c1b51eSKate Stone     if (m_multi_line) {
55344d93782SGreg Clayton       StringList lines;
554b9c1b51eSKate Stone       if (GetLines(lines, interrupted)) {
555b9c1b51eSKate Stone         if (interrupted) {
556e30f11d9SKate Stone           m_done = m_interrupt_exits;
557e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
558e30f11d9SKate Stone 
559b9c1b51eSKate Stone         } else {
56044d93782SGreg Clayton           line = lines.CopyList();
56144d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
56244d93782SGreg Clayton         }
563b9c1b51eSKate Stone       } else {
56444d93782SGreg Clayton         m_done = true;
56544d93782SGreg Clayton       }
566b9c1b51eSKate Stone     } else {
567b9c1b51eSKate Stone       if (GetLine(line, interrupted)) {
568e30f11d9SKate Stone         if (interrupted)
569e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
570e30f11d9SKate Stone         else
57144d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
572b9c1b51eSKate Stone       } else {
57344d93782SGreg Clayton         m_done = true;
57444d93782SGreg Clayton       }
57544d93782SGreg Clayton     }
57644d93782SGreg Clayton   }
57744d93782SGreg Clayton }
57844d93782SGreg Clayton 
579b9c1b51eSKate Stone void IOHandlerEditline::Cancel() {
580cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
581d5b44036SJonas Devlieghere   if (m_editline_up)
582d5b44036SJonas Devlieghere     m_editline_up->Cancel();
583cacde7dfSTodd Fiala #endif
584e68f5d6bSGreg Clayton }
585e68f5d6bSGreg Clayton 
586b9c1b51eSKate Stone bool IOHandlerEditline::Interrupt() {
587f0066ad0SGreg Clayton   // Let the delgate handle it first
588f0066ad0SGreg Clayton   if (m_delegate.IOHandlerInterrupt(*this))
589f0066ad0SGreg Clayton     return true;
590f0066ad0SGreg Clayton 
591cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
592d5b44036SJonas Devlieghere   if (m_editline_up)
593d5b44036SJonas Devlieghere     return m_editline_up->Interrupt();
594cacde7dfSTodd Fiala #endif
595f0066ad0SGreg Clayton   return false;
59644d93782SGreg Clayton }
59744d93782SGreg Clayton 
598b9c1b51eSKate Stone void IOHandlerEditline::GotEOF() {
599cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
600d5b44036SJonas Devlieghere   if (m_editline_up)
601d5b44036SJonas Devlieghere     m_editline_up->Interrupt();
602cacde7dfSTodd Fiala #endif
60344d93782SGreg Clayton }
60444d93782SGreg Clayton 
605b9c1b51eSKate Stone void IOHandlerEditline::PrintAsync(Stream *stream, const char *s, size_t len) {
6064446487dSPavel Labath #ifndef LLDB_DISABLE_LIBEDIT
607d5b44036SJonas Devlieghere   if (m_editline_up)
608d5b44036SJonas Devlieghere     m_editline_up->PrintAsync(stream, s, len);
6094446487dSPavel Labath   else
6104446487dSPavel Labath #endif
611fab31220STed Woodward   {
6128b98f12aSMartin Storsjo #ifdef _WIN32
613341e4789SDawn Perchik     const char *prompt = GetPrompt();
614b9c1b51eSKate Stone     if (prompt) {
615fab31220STed Woodward       // Back up over previous prompt using Windows API
616fab31220STed Woodward       CONSOLE_SCREEN_BUFFER_INFO screen_buffer_info;
617fab31220STed Woodward       HANDLE console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
618fab31220STed Woodward       GetConsoleScreenBufferInfo(console_handle, &screen_buffer_info);
619fab31220STed Woodward       COORD coord = screen_buffer_info.dwCursorPosition;
620fab31220STed Woodward       coord.X -= strlen(prompt);
621fab31220STed Woodward       if (coord.X < 0)
622fab31220STed Woodward         coord.X = 0;
623fab31220STed Woodward       SetConsoleCursorPosition(console_handle, coord);
624fab31220STed Woodward     }
625fab31220STed Woodward #endif
6264446487dSPavel Labath     IOHandler::PrintAsync(stream, s, len);
6278b98f12aSMartin Storsjo #ifdef _WIN32
628fab31220STed Woodward     if (prompt)
6297ca15ba7SLawrence D'Anna       IOHandler::PrintAsync(GetOutputStreamFileSP().get(), prompt,
630b9c1b51eSKate Stone                             strlen(prompt));
631341e4789SDawn Perchik #endif
632fab31220STed Woodward   }
6334446487dSPavel Labath }
6344446487dSPavel Labath 
63505097246SAdrian Prantl // we may want curses to be disabled for some builds for instance, windows
636914b8d98SDeepak Panickal #ifndef LLDB_DISABLE_CURSES
637914b8d98SDeepak Panickal 
63844d93782SGreg Clayton #define KEY_RETURN 10
63944d93782SGreg Clayton #define KEY_ESCAPE 27
64044d93782SGreg Clayton 
641b9c1b51eSKate Stone namespace curses {
64244d93782SGreg Clayton class Menu;
64344d93782SGreg Clayton class MenuDelegate;
64444d93782SGreg Clayton class Window;
64544d93782SGreg Clayton class WindowDelegate;
64644d93782SGreg Clayton typedef std::shared_ptr<Menu> MenuSP;
64744d93782SGreg Clayton typedef std::shared_ptr<MenuDelegate> MenuDelegateSP;
64844d93782SGreg Clayton typedef std::shared_ptr<Window> WindowSP;
64944d93782SGreg Clayton typedef std::shared_ptr<WindowDelegate> WindowDelegateSP;
65044d93782SGreg Clayton typedef std::vector<MenuSP> Menus;
65144d93782SGreg Clayton typedef std::vector<WindowSP> Windows;
65244d93782SGreg Clayton typedef std::vector<WindowDelegateSP> WindowDelegates;
65344d93782SGreg Clayton 
65444d93782SGreg Clayton #if 0
65544d93782SGreg Clayton type summary add -s "x=${var.x}, y=${var.y}" curses::Point
65644d93782SGreg Clayton type summary add -s "w=${var.width}, h=${var.height}" curses::Size
65744d93782SGreg Clayton type summary add -s "${var.origin%S} ${var.size%S}" curses::Rect
65844d93782SGreg Clayton #endif
659315b6884SEugene Zelenko 
660b9c1b51eSKate Stone struct Point {
66144d93782SGreg Clayton   int x;
66244d93782SGreg Clayton   int y;
66344d93782SGreg Clayton 
664b9c1b51eSKate Stone   Point(int _x = 0, int _y = 0) : x(_x), y(_y) {}
66544d93782SGreg Clayton 
666b9c1b51eSKate Stone   void Clear() {
66744d93782SGreg Clayton     x = 0;
66844d93782SGreg Clayton     y = 0;
66944d93782SGreg Clayton   }
67044d93782SGreg Clayton 
671b9c1b51eSKate Stone   Point &operator+=(const Point &rhs) {
67244d93782SGreg Clayton     x += rhs.x;
67344d93782SGreg Clayton     y += rhs.y;
67444d93782SGreg Clayton     return *this;
67544d93782SGreg Clayton   }
67644d93782SGreg Clayton 
677b9c1b51eSKate Stone   void Dump() { printf("(x=%i, y=%i)\n", x, y); }
67844d93782SGreg Clayton };
67944d93782SGreg Clayton 
680b9c1b51eSKate Stone bool operator==(const Point &lhs, const Point &rhs) {
68144d93782SGreg Clayton   return lhs.x == rhs.x && lhs.y == rhs.y;
68244d93782SGreg Clayton }
683315b6884SEugene Zelenko 
684b9c1b51eSKate Stone bool operator!=(const Point &lhs, const Point &rhs) {
68544d93782SGreg Clayton   return lhs.x != rhs.x || lhs.y != rhs.y;
68644d93782SGreg Clayton }
68744d93782SGreg Clayton 
688b9c1b51eSKate Stone struct Size {
68944d93782SGreg Clayton   int width;
69044d93782SGreg Clayton   int height;
691b9c1b51eSKate Stone   Size(int w = 0, int h = 0) : width(w), height(h) {}
69244d93782SGreg Clayton 
693b9c1b51eSKate Stone   void Clear() {
69444d93782SGreg Clayton     width = 0;
69544d93782SGreg Clayton     height = 0;
69644d93782SGreg Clayton   }
69744d93782SGreg Clayton 
698b9c1b51eSKate Stone   void Dump() { printf("(w=%i, h=%i)\n", width, height); }
69944d93782SGreg Clayton };
70044d93782SGreg Clayton 
701b9c1b51eSKate Stone bool operator==(const Size &lhs, const Size &rhs) {
70244d93782SGreg Clayton   return lhs.width == rhs.width && lhs.height == rhs.height;
70344d93782SGreg Clayton }
704315b6884SEugene Zelenko 
705b9c1b51eSKate Stone bool operator!=(const Size &lhs, const Size &rhs) {
70644d93782SGreg Clayton   return lhs.width != rhs.width || lhs.height != rhs.height;
70744d93782SGreg Clayton }
70844d93782SGreg Clayton 
709b9c1b51eSKate Stone struct Rect {
71044d93782SGreg Clayton   Point origin;
71144d93782SGreg Clayton   Size size;
71244d93782SGreg Clayton 
713b9c1b51eSKate Stone   Rect() : origin(), size() {}
71444d93782SGreg Clayton 
715b9c1b51eSKate Stone   Rect(const Point &p, const Size &s) : origin(p), size(s) {}
71644d93782SGreg Clayton 
717b9c1b51eSKate Stone   void Clear() {
71844d93782SGreg Clayton     origin.Clear();
71944d93782SGreg Clayton     size.Clear();
72044d93782SGreg Clayton   }
72144d93782SGreg Clayton 
722b9c1b51eSKate Stone   void Dump() {
723b9c1b51eSKate Stone     printf("(x=%i, y=%i), w=%i, h=%i)\n", origin.x, origin.y, size.width,
724b9c1b51eSKate Stone            size.height);
72544d93782SGreg Clayton   }
72644d93782SGreg Clayton 
727b9c1b51eSKate Stone   void Inset(int w, int h) {
72844d93782SGreg Clayton     if (size.width > w * 2)
72944d93782SGreg Clayton       size.width -= w * 2;
73044d93782SGreg Clayton     origin.x += w;
73144d93782SGreg Clayton 
73244d93782SGreg Clayton     if (size.height > h * 2)
73344d93782SGreg Clayton       size.height -= h * 2;
73444d93782SGreg Clayton     origin.y += h;
73544d93782SGreg Clayton   }
736315b6884SEugene Zelenko 
73705097246SAdrian Prantl   // Return a status bar rectangle which is the last line of this rectangle.
73805097246SAdrian Prantl   // This rectangle will be modified to not include the status bar area.
739b9c1b51eSKate Stone   Rect MakeStatusBar() {
74044d93782SGreg Clayton     Rect status_bar;
741b9c1b51eSKate Stone     if (size.height > 1) {
74244d93782SGreg Clayton       status_bar.origin.x = origin.x;
74344d93782SGreg Clayton       status_bar.origin.y = size.height;
74444d93782SGreg Clayton       status_bar.size.width = size.width;
74544d93782SGreg Clayton       status_bar.size.height = 1;
74644d93782SGreg Clayton       --size.height;
74744d93782SGreg Clayton     }
74844d93782SGreg Clayton     return status_bar;
74944d93782SGreg Clayton   }
75044d93782SGreg Clayton 
75105097246SAdrian Prantl   // Return a menubar rectangle which is the first line of this rectangle. This
75205097246SAdrian Prantl   // rectangle will be modified to not include the menubar area.
753b9c1b51eSKate Stone   Rect MakeMenuBar() {
75444d93782SGreg Clayton     Rect menubar;
755b9c1b51eSKate Stone     if (size.height > 1) {
75644d93782SGreg Clayton       menubar.origin.x = origin.x;
75744d93782SGreg Clayton       menubar.origin.y = origin.y;
75844d93782SGreg Clayton       menubar.size.width = size.width;
75944d93782SGreg Clayton       menubar.size.height = 1;
76044d93782SGreg Clayton       ++origin.y;
76144d93782SGreg Clayton       --size.height;
76244d93782SGreg Clayton     }
76344d93782SGreg Clayton     return menubar;
76444d93782SGreg Clayton   }
76544d93782SGreg Clayton 
766b9c1b51eSKate Stone   void HorizontalSplitPercentage(float top_percentage, Rect &top,
767b9c1b51eSKate Stone                                  Rect &bottom) const {
76844d93782SGreg Clayton     float top_height = top_percentage * size.height;
76944d93782SGreg Clayton     HorizontalSplit(top_height, top, bottom);
77044d93782SGreg Clayton   }
77144d93782SGreg Clayton 
772b9c1b51eSKate Stone   void HorizontalSplit(int top_height, Rect &top, Rect &bottom) const {
77344d93782SGreg Clayton     top = *this;
774b9c1b51eSKate Stone     if (top_height < size.height) {
77544d93782SGreg Clayton       top.size.height = top_height;
77644d93782SGreg Clayton       bottom.origin.x = origin.x;
77744d93782SGreg Clayton       bottom.origin.y = origin.y + top.size.height;
77844d93782SGreg Clayton       bottom.size.width = size.width;
77944d93782SGreg Clayton       bottom.size.height = size.height - top.size.height;
780b9c1b51eSKate Stone     } else {
78144d93782SGreg Clayton       bottom.Clear();
78244d93782SGreg Clayton     }
78344d93782SGreg Clayton   }
78444d93782SGreg Clayton 
785b9c1b51eSKate Stone   void VerticalSplitPercentage(float left_percentage, Rect &left,
786b9c1b51eSKate Stone                                Rect &right) const {
78744d93782SGreg Clayton     float left_width = left_percentage * size.width;
78844d93782SGreg Clayton     VerticalSplit(left_width, left, right);
78944d93782SGreg Clayton   }
79044d93782SGreg Clayton 
791b9c1b51eSKate Stone   void VerticalSplit(int left_width, Rect &left, Rect &right) const {
79244d93782SGreg Clayton     left = *this;
793b9c1b51eSKate Stone     if (left_width < size.width) {
79444d93782SGreg Clayton       left.size.width = left_width;
79544d93782SGreg Clayton       right.origin.x = origin.x + left.size.width;
79644d93782SGreg Clayton       right.origin.y = origin.y;
79744d93782SGreg Clayton       right.size.width = size.width - left.size.width;
79844d93782SGreg Clayton       right.size.height = size.height;
799b9c1b51eSKate Stone     } else {
80044d93782SGreg Clayton       right.Clear();
80144d93782SGreg Clayton     }
80244d93782SGreg Clayton   }
80344d93782SGreg Clayton };
80444d93782SGreg Clayton 
805b9c1b51eSKate Stone bool operator==(const Rect &lhs, const Rect &rhs) {
80644d93782SGreg Clayton   return lhs.origin == rhs.origin && lhs.size == rhs.size;
80744d93782SGreg Clayton }
808315b6884SEugene Zelenko 
809b9c1b51eSKate Stone bool operator!=(const Rect &lhs, const Rect &rhs) {
81044d93782SGreg Clayton   return lhs.origin != rhs.origin || lhs.size != rhs.size;
81144d93782SGreg Clayton }
81244d93782SGreg Clayton 
813b9c1b51eSKate Stone enum HandleCharResult {
81444d93782SGreg Clayton   eKeyNotHandled = 0,
81544d93782SGreg Clayton   eKeyHandled = 1,
81644d93782SGreg Clayton   eQuitApplication = 2
81744d93782SGreg Clayton };
81844d93782SGreg Clayton 
819b9c1b51eSKate Stone enum class MenuActionResult {
82044d93782SGreg Clayton   Handled,
82144d93782SGreg Clayton   NotHandled,
82244d93782SGreg Clayton   Quit // Exit all menus and quit
82344d93782SGreg Clayton };
82444d93782SGreg Clayton 
825b9c1b51eSKate Stone struct KeyHelp {
82644d93782SGreg Clayton   int ch;
82744d93782SGreg Clayton   const char *description;
82844d93782SGreg Clayton };
82944d93782SGreg Clayton 
830b9c1b51eSKate Stone class WindowDelegate {
83144d93782SGreg Clayton public:
832b9c1b51eSKate Stone   virtual ~WindowDelegate() = default;
83344d93782SGreg Clayton 
834b9c1b51eSKate Stone   virtual bool WindowDelegateDraw(Window &window, bool force) {
83544d93782SGreg Clayton     return false; // Drawing not handled
83644d93782SGreg Clayton   }
83744d93782SGreg Clayton 
838b9c1b51eSKate Stone   virtual HandleCharResult WindowDelegateHandleChar(Window &window, int key) {
83944d93782SGreg Clayton     return eKeyNotHandled;
84044d93782SGreg Clayton   }
84144d93782SGreg Clayton 
842b9c1b51eSKate Stone   virtual const char *WindowDelegateGetHelpText() { return nullptr; }
84344d93782SGreg Clayton 
844b9c1b51eSKate Stone   virtual KeyHelp *WindowDelegateGetKeyHelp() { return nullptr; }
84544d93782SGreg Clayton };
84644d93782SGreg Clayton 
847b9c1b51eSKate Stone class HelpDialogDelegate : public WindowDelegate {
84844d93782SGreg Clayton public:
84944d93782SGreg Clayton   HelpDialogDelegate(const char *text, KeyHelp *key_help_array);
85044d93782SGreg Clayton 
851bd5ae6b4SGreg Clayton   ~HelpDialogDelegate() override;
85244d93782SGreg Clayton 
853b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
85444d93782SGreg Clayton 
855b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
85644d93782SGreg Clayton 
857b9c1b51eSKate Stone   size_t GetNumLines() const { return m_text.GetSize(); }
85844d93782SGreg Clayton 
859b9c1b51eSKate Stone   size_t GetMaxLineLength() const { return m_text.GetMaxStringLength(); }
86044d93782SGreg Clayton 
86144d93782SGreg Clayton protected:
86244d93782SGreg Clayton   StringList m_text;
86344d93782SGreg Clayton   int m_first_visible_line;
86444d93782SGreg Clayton };
86544d93782SGreg Clayton 
866b9c1b51eSKate Stone class Window {
86744d93782SGreg Clayton public:
868b9c1b51eSKate Stone   Window(const char *name)
869b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
870b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
871b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(false),
872b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {}
87344d93782SGreg Clayton 
874b9c1b51eSKate Stone   Window(const char *name, WINDOW *w, bool del = true)
875b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
876b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
877b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(del),
878b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
87944d93782SGreg Clayton     if (w)
88044d93782SGreg Clayton       Reset(w);
88144d93782SGreg Clayton   }
88244d93782SGreg Clayton 
883b9c1b51eSKate Stone   Window(const char *name, const Rect &bounds)
884b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_parent(nullptr), m_subwindows(),
885b9c1b51eSKate Stone         m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
886b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(true),
887b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
888b9c1b51eSKate Stone     Reset(::newwin(bounds.size.height, bounds.size.width, bounds.origin.y,
889b9c1b51eSKate Stone                    bounds.origin.y));
89044d93782SGreg Clayton   }
89144d93782SGreg Clayton 
892b9c1b51eSKate Stone   virtual ~Window() {
89344d93782SGreg Clayton     RemoveSubWindows();
89444d93782SGreg Clayton     Reset();
89544d93782SGreg Clayton   }
89644d93782SGreg Clayton 
897b9c1b51eSKate Stone   void Reset(WINDOW *w = nullptr, bool del = true) {
89844d93782SGreg Clayton     if (m_window == w)
89944d93782SGreg Clayton       return;
90044d93782SGreg Clayton 
901b9c1b51eSKate Stone     if (m_panel) {
90244d93782SGreg Clayton       ::del_panel(m_panel);
903c5dac77aSEugene Zelenko       m_panel = nullptr;
90444d93782SGreg Clayton     }
905b9c1b51eSKate Stone     if (m_window && m_delete) {
90644d93782SGreg Clayton       ::delwin(m_window);
907c5dac77aSEugene Zelenko       m_window = nullptr;
90844d93782SGreg Clayton       m_delete = false;
90944d93782SGreg Clayton     }
910b9c1b51eSKate Stone     if (w) {
91144d93782SGreg Clayton       m_window = w;
91244d93782SGreg Clayton       m_panel = ::new_panel(m_window);
91344d93782SGreg Clayton       m_delete = del;
91444d93782SGreg Clayton     }
91544d93782SGreg Clayton   }
91644d93782SGreg Clayton 
91744d93782SGreg Clayton   void AttributeOn(attr_t attr) { ::wattron(m_window, attr); }
91844d93782SGreg Clayton   void AttributeOff(attr_t attr) { ::wattroff(m_window, attr); }
919b9c1b51eSKate Stone   void Box(chtype v_char = ACS_VLINE, chtype h_char = ACS_HLINE) {
920b9c1b51eSKate Stone     ::box(m_window, v_char, h_char);
921b9c1b51eSKate Stone   }
92244d93782SGreg Clayton   void Clear() { ::wclear(m_window); }
92344d93782SGreg Clayton   void Erase() { ::werase(m_window); }
924b9c1b51eSKate Stone   Rect GetBounds() {
925b9c1b51eSKate Stone     return Rect(GetParentOrigin(), GetSize());
926b9c1b51eSKate Stone   } // Get the rectangle in our parent window
92744d93782SGreg Clayton   int GetChar() { return ::wgetch(m_window); }
92844d93782SGreg Clayton   int GetCursorX() { return getcurx(m_window); }
92944d93782SGreg Clayton   int GetCursorY() { return getcury(m_window); }
930b9c1b51eSKate Stone   Rect GetFrame() {
931b9c1b51eSKate Stone     return Rect(Point(), GetSize());
932b9c1b51eSKate Stone   } // Get our rectangle in our own coordinate system
93344d93782SGreg Clayton   Point GetParentOrigin() { return Point(GetParentX(), GetParentY()); }
93444d93782SGreg Clayton   Size GetSize() { return Size(GetWidth(), GetHeight()); }
93544d93782SGreg Clayton   int GetParentX() { return getparx(m_window); }
93644d93782SGreg Clayton   int GetParentY() { return getpary(m_window); }
93744d93782SGreg Clayton   int GetMaxX() { return getmaxx(m_window); }
93844d93782SGreg Clayton   int GetMaxY() { return getmaxy(m_window); }
93944d93782SGreg Clayton   int GetWidth() { return GetMaxX(); }
94044d93782SGreg Clayton   int GetHeight() { return GetMaxY(); }
94144d93782SGreg Clayton   void MoveCursor(int x, int y) { ::wmove(m_window, y, x); }
94244d93782SGreg Clayton   void MoveWindow(int x, int y) { MoveWindow(Point(x, y)); }
94344d93782SGreg Clayton   void Resize(int w, int h) { ::wresize(m_window, h, w); }
944b9c1b51eSKate Stone   void Resize(const Size &size) {
945b9c1b51eSKate Stone     ::wresize(m_window, size.height, size.width);
946b9c1b51eSKate Stone   }
94744d93782SGreg Clayton   void PutChar(int ch) { ::waddch(m_window, ch); }
94844d93782SGreg Clayton   void PutCString(const char *s, int len = -1) { ::waddnstr(m_window, s, len); }
949b9c1b51eSKate Stone   void SetBackground(int color_pair_idx) {
950b9c1b51eSKate Stone     ::wbkgd(m_window, COLOR_PAIR(color_pair_idx));
951b9c1b51eSKate Stone   }
95244d93782SGreg Clayton 
953b9c1b51eSKate Stone   void PutCStringTruncated(const char *s, int right_pad) {
95444d93782SGreg Clayton     int bytes_left = GetWidth() - GetCursorX();
955b9c1b51eSKate Stone     if (bytes_left > right_pad) {
95644d93782SGreg Clayton       bytes_left -= right_pad;
95744d93782SGreg Clayton       ::waddnstr(m_window, s, bytes_left);
95844d93782SGreg Clayton     }
95944d93782SGreg Clayton   }
96044d93782SGreg Clayton 
961b9c1b51eSKate Stone   void MoveWindow(const Point &origin) {
96244d93782SGreg Clayton     const bool moving_window = origin != GetParentOrigin();
963b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
96444d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
96544d93782SGreg Clayton       Size size = GetSize();
966b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, size.height, size.width, origin.y,
967b9c1b51eSKate Stone                      origin.x),
968b9c1b51eSKate Stone             true);
969b9c1b51eSKate Stone     } else {
97044d93782SGreg Clayton       ::mvwin(m_window, origin.y, origin.x);
97144d93782SGreg Clayton     }
97244d93782SGreg Clayton   }
97344d93782SGreg Clayton 
974b9c1b51eSKate Stone   void SetBounds(const Rect &bounds) {
97544d93782SGreg Clayton     const bool moving_window = bounds.origin != GetParentOrigin();
976b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
97744d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
978b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, bounds.size.height, bounds.size.width,
979b9c1b51eSKate Stone                      bounds.origin.y, bounds.origin.x),
980b9c1b51eSKate Stone             true);
981b9c1b51eSKate Stone     } else {
98244d93782SGreg Clayton       if (moving_window)
98344d93782SGreg Clayton         MoveWindow(bounds.origin);
98444d93782SGreg Clayton       Resize(bounds.size);
98544d93782SGreg Clayton     }
98644d93782SGreg Clayton   }
98744d93782SGreg Clayton 
988b9c1b51eSKate Stone   void Printf(const char *format, ...) __attribute__((format(printf, 2, 3))) {
98944d93782SGreg Clayton     va_list args;
99044d93782SGreg Clayton     va_start(args, format);
99144d93782SGreg Clayton     vwprintw(m_window, format, args);
99244d93782SGreg Clayton     va_end(args);
99344d93782SGreg Clayton   }
99444d93782SGreg Clayton 
995b9c1b51eSKate Stone   void Touch() {
99644d93782SGreg Clayton     ::touchwin(m_window);
99744d93782SGreg Clayton     if (m_parent)
99844d93782SGreg Clayton       m_parent->Touch();
99944d93782SGreg Clayton   }
100044d93782SGreg Clayton 
1001b9c1b51eSKate Stone   WindowSP CreateSubWindow(const char *name, const Rect &bounds,
1002b9c1b51eSKate Stone                            bool make_active) {
1003c6091d2bSJonas Devlieghere     auto get_window = [this, &bounds]() {
1004c6091d2bSJonas Devlieghere       return m_window
1005c6091d2bSJonas Devlieghere                  ? ::subwin(m_window, bounds.size.height, bounds.size.width,
1006c6091d2bSJonas Devlieghere                             bounds.origin.y, bounds.origin.x)
1007c6091d2bSJonas Devlieghere                  : ::newwin(bounds.size.height, bounds.size.width,
1008c6091d2bSJonas Devlieghere                             bounds.origin.y, bounds.origin.x);
1009c6091d2bSJonas Devlieghere     };
1010c6091d2bSJonas Devlieghere     WindowSP subwindow_sp = std::make_shared<Window>(name, get_window(), true);
1011c6091d2bSJonas Devlieghere     subwindow_sp->m_is_subwin = subwindow_sp.operator bool();
101244d93782SGreg Clayton     subwindow_sp->m_parent = this;
1013b9c1b51eSKate Stone     if (make_active) {
101444d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
101544d93782SGreg Clayton       m_curr_active_window_idx = m_subwindows.size();
101644d93782SGreg Clayton     }
101744d93782SGreg Clayton     m_subwindows.push_back(subwindow_sp);
101844d93782SGreg Clayton     ::top_panel(subwindow_sp->m_panel);
101944d93782SGreg Clayton     m_needs_update = true;
102044d93782SGreg Clayton     return subwindow_sp;
102144d93782SGreg Clayton   }
102244d93782SGreg Clayton 
1023b9c1b51eSKate Stone   bool RemoveSubWindow(Window *window) {
102444d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
102544d93782SGreg Clayton     size_t i = 0;
1026b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
1027b9c1b51eSKate Stone       if ((*pos).get() == window) {
102844d93782SGreg Clayton         if (m_prev_active_window_idx == i)
102944d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1030b9c1b51eSKate Stone         else if (m_prev_active_window_idx != UINT32_MAX &&
1031b9c1b51eSKate Stone                  m_prev_active_window_idx > i)
103244d93782SGreg Clayton           --m_prev_active_window_idx;
103344d93782SGreg Clayton 
103444d93782SGreg Clayton         if (m_curr_active_window_idx == i)
103544d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
1036b9c1b51eSKate Stone         else if (m_curr_active_window_idx != UINT32_MAX &&
1037b9c1b51eSKate Stone                  m_curr_active_window_idx > i)
103844d93782SGreg Clayton           --m_curr_active_window_idx;
103944d93782SGreg Clayton         window->Erase();
104044d93782SGreg Clayton         m_subwindows.erase(pos);
104144d93782SGreg Clayton         m_needs_update = true;
104244d93782SGreg Clayton         if (m_parent)
104344d93782SGreg Clayton           m_parent->Touch();
104444d93782SGreg Clayton         else
104544d93782SGreg Clayton           ::touchwin(stdscr);
104644d93782SGreg Clayton         return true;
104744d93782SGreg Clayton       }
104844d93782SGreg Clayton     }
104944d93782SGreg Clayton     return false;
105044d93782SGreg Clayton   }
105144d93782SGreg Clayton 
1052b9c1b51eSKate Stone   WindowSP FindSubWindow(const char *name) {
105344d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
105444d93782SGreg Clayton     size_t i = 0;
1055b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
10568d20cfdfSJonas Devlieghere       if ((*pos)->m_name == name)
105744d93782SGreg Clayton         return *pos;
105844d93782SGreg Clayton     }
105944d93782SGreg Clayton     return WindowSP();
106044d93782SGreg Clayton   }
106144d93782SGreg Clayton 
1062b9c1b51eSKate Stone   void RemoveSubWindows() {
106344d93782SGreg Clayton     m_curr_active_window_idx = UINT32_MAX;
106444d93782SGreg Clayton     m_prev_active_window_idx = UINT32_MAX;
106544d93782SGreg Clayton     for (Windows::iterator pos = m_subwindows.begin();
1066b9c1b51eSKate Stone          pos != m_subwindows.end(); pos = m_subwindows.erase(pos)) {
106744d93782SGreg Clayton       (*pos)->Erase();
106844d93782SGreg Clayton     }
106944d93782SGreg Clayton     if (m_parent)
107044d93782SGreg Clayton       m_parent->Touch();
107144d93782SGreg Clayton     else
107244d93782SGreg Clayton       ::touchwin(stdscr);
107344d93782SGreg Clayton   }
107444d93782SGreg Clayton 
1075b9c1b51eSKate Stone   WINDOW *get() { return m_window; }
107644d93782SGreg Clayton 
1077b9c1b51eSKate Stone   operator WINDOW *() { return m_window; }
107844d93782SGreg Clayton 
107944d93782SGreg Clayton   // Window drawing utilities
1080b9c1b51eSKate Stone   void DrawTitleBox(const char *title, const char *bottom_message = nullptr) {
108144d93782SGreg Clayton     attr_t attr = 0;
108244d93782SGreg Clayton     if (IsActive())
108344d93782SGreg Clayton       attr = A_BOLD | COLOR_PAIR(2);
108444d93782SGreg Clayton     else
108544d93782SGreg Clayton       attr = 0;
108644d93782SGreg Clayton     if (attr)
108744d93782SGreg Clayton       AttributeOn(attr);
108844d93782SGreg Clayton 
108944d93782SGreg Clayton     Box();
109044d93782SGreg Clayton     MoveCursor(3, 0);
109144d93782SGreg Clayton 
1092b9c1b51eSKate Stone     if (title && title[0]) {
109344d93782SGreg Clayton       PutChar('<');
109444d93782SGreg Clayton       PutCString(title);
109544d93782SGreg Clayton       PutChar('>');
109644d93782SGreg Clayton     }
109744d93782SGreg Clayton 
1098b9c1b51eSKate Stone     if (bottom_message && bottom_message[0]) {
109944d93782SGreg Clayton       int bottom_message_length = strlen(bottom_message);
110044d93782SGreg Clayton       int x = GetWidth() - 3 - (bottom_message_length + 2);
110144d93782SGreg Clayton 
1102b9c1b51eSKate Stone       if (x > 0) {
110344d93782SGreg Clayton         MoveCursor(x, GetHeight() - 1);
110444d93782SGreg Clayton         PutChar('[');
110544d93782SGreg Clayton         PutCString(bottom_message);
110644d93782SGreg Clayton         PutChar(']');
1107b9c1b51eSKate Stone       } else {
110844d93782SGreg Clayton         MoveCursor(1, GetHeight() - 1);
110944d93782SGreg Clayton         PutChar('[');
111044d93782SGreg Clayton         PutCStringTruncated(bottom_message, 1);
111144d93782SGreg Clayton       }
111244d93782SGreg Clayton     }
111344d93782SGreg Clayton     if (attr)
111444d93782SGreg Clayton       AttributeOff(attr);
111544d93782SGreg Clayton   }
111644d93782SGreg Clayton 
1117b9c1b51eSKate Stone   virtual void Draw(bool force) {
111844d93782SGreg Clayton     if (m_delegate_sp && m_delegate_sp->WindowDelegateDraw(*this, force))
111944d93782SGreg Clayton       return;
112044d93782SGreg Clayton 
112144d93782SGreg Clayton     for (auto &subwindow_sp : m_subwindows)
112244d93782SGreg Clayton       subwindow_sp->Draw(force);
112344d93782SGreg Clayton   }
112444d93782SGreg Clayton 
1125b9c1b51eSKate Stone   bool CreateHelpSubwindow() {
1126b9c1b51eSKate Stone     if (m_delegate_sp) {
112744d93782SGreg Clayton       const char *text = m_delegate_sp->WindowDelegateGetHelpText();
112844d93782SGreg Clayton       KeyHelp *key_help = m_delegate_sp->WindowDelegateGetKeyHelp();
1129b9c1b51eSKate Stone       if ((text && text[0]) || key_help) {
1130d5b44036SJonas Devlieghere         std::unique_ptr<HelpDialogDelegate> help_delegate_up(
1131b9c1b51eSKate Stone             new HelpDialogDelegate(text, key_help));
1132d5b44036SJonas Devlieghere         const size_t num_lines = help_delegate_up->GetNumLines();
1133d5b44036SJonas Devlieghere         const size_t max_length = help_delegate_up->GetMaxLineLength();
113444d93782SGreg Clayton         Rect bounds = GetBounds();
113544d93782SGreg Clayton         bounds.Inset(1, 1);
1136b9c1b51eSKate Stone         if (max_length + 4 < static_cast<size_t>(bounds.size.width)) {
113744d93782SGreg Clayton           bounds.origin.x += (bounds.size.width - max_length + 4) / 2;
113844d93782SGreg Clayton           bounds.size.width = max_length + 4;
1139b9c1b51eSKate Stone         } else {
1140b9c1b51eSKate Stone           if (bounds.size.width > 100) {
114144d93782SGreg Clayton             const int inset_w = bounds.size.width / 4;
114244d93782SGreg Clayton             bounds.origin.x += inset_w;
114344d93782SGreg Clayton             bounds.size.width -= 2 * inset_w;
114444d93782SGreg Clayton           }
114544d93782SGreg Clayton         }
114644d93782SGreg Clayton 
1147b9c1b51eSKate Stone         if (num_lines + 2 < static_cast<size_t>(bounds.size.height)) {
114844d93782SGreg Clayton           bounds.origin.y += (bounds.size.height - num_lines + 2) / 2;
114944d93782SGreg Clayton           bounds.size.height = num_lines + 2;
1150b9c1b51eSKate Stone         } else {
1151b9c1b51eSKate Stone           if (bounds.size.height > 100) {
115244d93782SGreg Clayton             const int inset_h = bounds.size.height / 4;
115344d93782SGreg Clayton             bounds.origin.y += inset_h;
115444d93782SGreg Clayton             bounds.size.height -= 2 * inset_h;
115544d93782SGreg Clayton           }
115644d93782SGreg Clayton         }
11575fdb09bbSGreg Clayton         WindowSP help_window_sp;
11585fdb09bbSGreg Clayton         Window *parent_window = GetParent();
11595fdb09bbSGreg Clayton         if (parent_window)
11605fdb09bbSGreg Clayton           help_window_sp = parent_window->CreateSubWindow("Help", bounds, true);
11615fdb09bbSGreg Clayton         else
11625fdb09bbSGreg Clayton           help_window_sp = CreateSubWindow("Help", bounds, true);
1163b9c1b51eSKate Stone         help_window_sp->SetDelegate(
1164d5b44036SJonas Devlieghere             WindowDelegateSP(help_delegate_up.release()));
116544d93782SGreg Clayton         return true;
116644d93782SGreg Clayton       }
116744d93782SGreg Clayton     }
116844d93782SGreg Clayton     return false;
116944d93782SGreg Clayton   }
117044d93782SGreg Clayton 
1171b9c1b51eSKate Stone   virtual HandleCharResult HandleChar(int key) {
117244d93782SGreg Clayton     // Always check the active window first
117344d93782SGreg Clayton     HandleCharResult result = eKeyNotHandled;
117444d93782SGreg Clayton     WindowSP active_window_sp = GetActiveWindow();
1175b9c1b51eSKate Stone     if (active_window_sp) {
117644d93782SGreg Clayton       result = active_window_sp->HandleChar(key);
117744d93782SGreg Clayton       if (result != eKeyNotHandled)
117844d93782SGreg Clayton         return result;
117944d93782SGreg Clayton     }
118044d93782SGreg Clayton 
1181b9c1b51eSKate Stone     if (m_delegate_sp) {
118244d93782SGreg Clayton       result = m_delegate_sp->WindowDelegateHandleChar(*this, key);
118344d93782SGreg Clayton       if (result != eKeyNotHandled)
118444d93782SGreg Clayton         return result;
118544d93782SGreg Clayton     }
118644d93782SGreg Clayton 
118705097246SAdrian Prantl     // Then check for any windows that want any keys that weren't handled. This
118805097246SAdrian Prantl     // is typically only for a menubar. Make a copy of the subwindows in case
118905097246SAdrian Prantl     // any HandleChar() functions muck with the subwindows. If we don't do
119005097246SAdrian Prantl     // this, we can crash when iterating over the subwindows.
119144d93782SGreg Clayton     Windows subwindows(m_subwindows);
1192b9c1b51eSKate Stone     for (auto subwindow_sp : subwindows) {
1193b9c1b51eSKate Stone       if (!subwindow_sp->m_can_activate) {
119444d93782SGreg Clayton         HandleCharResult result = subwindow_sp->HandleChar(key);
119544d93782SGreg Clayton         if (result != eKeyNotHandled)
119644d93782SGreg Clayton           return result;
119744d93782SGreg Clayton       }
119844d93782SGreg Clayton     }
119944d93782SGreg Clayton 
120044d93782SGreg Clayton     return eKeyNotHandled;
120144d93782SGreg Clayton   }
120244d93782SGreg Clayton 
1203b9c1b51eSKate Stone   WindowSP GetActiveWindow() {
1204b9c1b51eSKate Stone     if (!m_subwindows.empty()) {
1205b9c1b51eSKate Stone       if (m_curr_active_window_idx >= m_subwindows.size()) {
1206b9c1b51eSKate Stone         if (m_prev_active_window_idx < m_subwindows.size()) {
120744d93782SGreg Clayton           m_curr_active_window_idx = m_prev_active_window_idx;
120844d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1209b9c1b51eSKate Stone         } else if (IsActive()) {
121044d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
121144d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
121244d93782SGreg Clayton 
121344d93782SGreg Clayton           // Find first window that wants to be active if this window is active
121444d93782SGreg Clayton           const size_t num_subwindows = m_subwindows.size();
1215b9c1b51eSKate Stone           for (size_t i = 0; i < num_subwindows; ++i) {
1216b9c1b51eSKate Stone             if (m_subwindows[i]->GetCanBeActive()) {
121744d93782SGreg Clayton               m_curr_active_window_idx = i;
121844d93782SGreg Clayton               break;
121944d93782SGreg Clayton             }
122044d93782SGreg Clayton           }
122144d93782SGreg Clayton         }
122244d93782SGreg Clayton       }
122344d93782SGreg Clayton 
122444d93782SGreg Clayton       if (m_curr_active_window_idx < m_subwindows.size())
122544d93782SGreg Clayton         return m_subwindows[m_curr_active_window_idx];
122644d93782SGreg Clayton     }
122744d93782SGreg Clayton     return WindowSP();
122844d93782SGreg Clayton   }
122944d93782SGreg Clayton 
1230b9c1b51eSKate Stone   bool GetCanBeActive() const { return m_can_activate; }
123144d93782SGreg Clayton 
1232b9c1b51eSKate Stone   void SetCanBeActive(bool b) { m_can_activate = b; }
123344d93782SGreg Clayton 
1234b9c1b51eSKate Stone   void SetDelegate(const WindowDelegateSP &delegate_sp) {
123544d93782SGreg Clayton     m_delegate_sp = delegate_sp;
123644d93782SGreg Clayton   }
123744d93782SGreg Clayton 
1238b9c1b51eSKate Stone   Window *GetParent() const { return m_parent; }
123944d93782SGreg Clayton 
1240b9c1b51eSKate Stone   bool IsActive() const {
124144d93782SGreg Clayton     if (m_parent)
124244d93782SGreg Clayton       return m_parent->GetActiveWindow().get() == this;
124344d93782SGreg Clayton     else
124444d93782SGreg Clayton       return true; // Top level window is always active
124544d93782SGreg Clayton   }
124644d93782SGreg Clayton 
1247b9c1b51eSKate Stone   void SelectNextWindowAsActive() {
124844d93782SGreg Clayton     // Move active focus to next window
124944d93782SGreg Clayton     const size_t num_subwindows = m_subwindows.size();
1250b9c1b51eSKate Stone     if (m_curr_active_window_idx == UINT32_MAX) {
125144d93782SGreg Clayton       uint32_t idx = 0;
1252b9c1b51eSKate Stone       for (auto subwindow_sp : m_subwindows) {
1253b9c1b51eSKate Stone         if (subwindow_sp->GetCanBeActive()) {
125444d93782SGreg Clayton           m_curr_active_window_idx = idx;
125544d93782SGreg Clayton           break;
125644d93782SGreg Clayton         }
125744d93782SGreg Clayton         ++idx;
125844d93782SGreg Clayton       }
1259b9c1b51eSKate Stone     } else if (m_curr_active_window_idx + 1 < num_subwindows) {
126044d93782SGreg Clayton       bool handled = false;
126144d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1262b9c1b51eSKate Stone       for (size_t idx = m_curr_active_window_idx + 1; idx < num_subwindows;
1263b9c1b51eSKate Stone            ++idx) {
1264b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
126544d93782SGreg Clayton           m_curr_active_window_idx = idx;
126644d93782SGreg Clayton           handled = true;
126744d93782SGreg Clayton           break;
126844d93782SGreg Clayton         }
126944d93782SGreg Clayton       }
1270b9c1b51eSKate Stone       if (!handled) {
1271b9c1b51eSKate Stone         for (size_t idx = 0; idx <= m_prev_active_window_idx; ++idx) {
1272b9c1b51eSKate Stone           if (m_subwindows[idx]->GetCanBeActive()) {
127344d93782SGreg Clayton             m_curr_active_window_idx = idx;
127444d93782SGreg Clayton             break;
127544d93782SGreg Clayton           }
127644d93782SGreg Clayton         }
127744d93782SGreg Clayton       }
1278b9c1b51eSKate Stone     } else {
127944d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1280b9c1b51eSKate Stone       for (size_t idx = 0; idx < num_subwindows; ++idx) {
1281b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
128244d93782SGreg Clayton           m_curr_active_window_idx = idx;
128344d93782SGreg Clayton           break;
128444d93782SGreg Clayton         }
128544d93782SGreg Clayton       }
128644d93782SGreg Clayton     }
128744d93782SGreg Clayton   }
128844d93782SGreg Clayton 
1289b9c1b51eSKate Stone   const char *GetName() const { return m_name.c_str(); }
1290315b6884SEugene Zelenko 
129144d93782SGreg Clayton protected:
129244d93782SGreg Clayton   std::string m_name;
129344d93782SGreg Clayton   WINDOW *m_window;
129444d93782SGreg Clayton   PANEL *m_panel;
129544d93782SGreg Clayton   Window *m_parent;
129644d93782SGreg Clayton   Windows m_subwindows;
129744d93782SGreg Clayton   WindowDelegateSP m_delegate_sp;
129844d93782SGreg Clayton   uint32_t m_curr_active_window_idx;
129944d93782SGreg Clayton   uint32_t m_prev_active_window_idx;
130044d93782SGreg Clayton   bool m_delete;
130144d93782SGreg Clayton   bool m_needs_update;
130244d93782SGreg Clayton   bool m_can_activate;
130344d93782SGreg Clayton   bool m_is_subwin;
130444d93782SGreg Clayton 
130544d93782SGreg Clayton private:
130644d93782SGreg Clayton   DISALLOW_COPY_AND_ASSIGN(Window);
130744d93782SGreg Clayton };
130844d93782SGreg Clayton 
1309b9c1b51eSKate Stone class MenuDelegate {
131044d93782SGreg Clayton public:
1311315b6884SEugene Zelenko   virtual ~MenuDelegate() = default;
131244d93782SGreg Clayton 
1313b9c1b51eSKate Stone   virtual MenuActionResult MenuDelegateAction(Menu &menu) = 0;
131444d93782SGreg Clayton };
131544d93782SGreg Clayton 
1316b9c1b51eSKate Stone class Menu : public WindowDelegate {
131744d93782SGreg Clayton public:
1318b9c1b51eSKate Stone   enum class Type { Invalid, Bar, Item, Separator };
131944d93782SGreg Clayton 
132044d93782SGreg Clayton   // Menubar or separator constructor
132144d93782SGreg Clayton   Menu(Type type);
132244d93782SGreg Clayton 
132344d93782SGreg Clayton   // Menuitem constructor
1324b9c1b51eSKate Stone   Menu(const char *name, const char *key_name, int key_value,
132544d93782SGreg Clayton        uint64_t identifier);
132644d93782SGreg Clayton 
1327315b6884SEugene Zelenko   ~Menu() override = default;
132844d93782SGreg Clayton 
1329b9c1b51eSKate Stone   const MenuDelegateSP &GetDelegate() const { return m_delegate_sp; }
133044d93782SGreg Clayton 
1331b9c1b51eSKate Stone   void SetDelegate(const MenuDelegateSP &delegate_sp) {
133244d93782SGreg Clayton     m_delegate_sp = delegate_sp;
133344d93782SGreg Clayton   }
133444d93782SGreg Clayton 
1335b9c1b51eSKate Stone   void RecalculateNameLengths();
133644d93782SGreg Clayton 
1337b9c1b51eSKate Stone   void AddSubmenu(const MenuSP &menu_sp);
133844d93782SGreg Clayton 
1339b9c1b51eSKate Stone   int DrawAndRunMenu(Window &window);
134044d93782SGreg Clayton 
1341b9c1b51eSKate Stone   void DrawMenuTitle(Window &window, bool highlight);
134244d93782SGreg Clayton 
1343b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
134444d93782SGreg Clayton 
1345b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
134644d93782SGreg Clayton 
1347b9c1b51eSKate Stone   MenuActionResult ActionPrivate(Menu &menu) {
134844d93782SGreg Clayton     MenuActionResult result = MenuActionResult::NotHandled;
1349b9c1b51eSKate Stone     if (m_delegate_sp) {
135044d93782SGreg Clayton       result = m_delegate_sp->MenuDelegateAction(menu);
135144d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
135244d93782SGreg Clayton         return result;
1353b9c1b51eSKate Stone     } else if (m_parent) {
135444d93782SGreg Clayton       result = m_parent->ActionPrivate(menu);
135544d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
135644d93782SGreg Clayton         return result;
135744d93782SGreg Clayton     }
135844d93782SGreg Clayton     return m_canned_result;
135944d93782SGreg Clayton   }
136044d93782SGreg Clayton 
1361b9c1b51eSKate Stone   MenuActionResult Action() {
136205097246SAdrian Prantl     // Call the recursive action so it can try to handle it with the menu
136305097246SAdrian Prantl     // delegate, and if not, try our parent menu
136444d93782SGreg Clayton     return ActionPrivate(*this);
136544d93782SGreg Clayton   }
136644d93782SGreg Clayton 
1367b9c1b51eSKate Stone   void SetCannedResult(MenuActionResult result) { m_canned_result = result; }
136844d93782SGreg Clayton 
1369b9c1b51eSKate Stone   Menus &GetSubmenus() { return m_submenus; }
137044d93782SGreg Clayton 
1371b9c1b51eSKate Stone   const Menus &GetSubmenus() const { return m_submenus; }
137244d93782SGreg Clayton 
1373b9c1b51eSKate Stone   int GetSelectedSubmenuIndex() const { return m_selected; }
137444d93782SGreg Clayton 
1375b9c1b51eSKate Stone   void SetSelectedSubmenuIndex(int idx) { m_selected = idx; }
137644d93782SGreg Clayton 
1377b9c1b51eSKate Stone   Type GetType() const { return m_type; }
137844d93782SGreg Clayton 
1379b9c1b51eSKate Stone   int GetStartingColumn() const { return m_start_col; }
138044d93782SGreg Clayton 
1381b9c1b51eSKate Stone   void SetStartingColumn(int col) { m_start_col = col; }
138244d93782SGreg Clayton 
1383b9c1b51eSKate Stone   int GetKeyValue() const { return m_key_value; }
138444d93782SGreg Clayton 
1385b9c1b51eSKate Stone   std::string &GetName() { return m_name; }
138644d93782SGreg Clayton 
1387b9c1b51eSKate Stone   int GetDrawWidth() const {
138844d93782SGreg Clayton     return m_max_submenu_name_length + m_max_submenu_key_name_length + 8;
138944d93782SGreg Clayton   }
139044d93782SGreg Clayton 
1391b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
139244d93782SGreg Clayton 
1393b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
139444d93782SGreg Clayton 
139544d93782SGreg Clayton protected:
139644d93782SGreg Clayton   std::string m_name;
139744d93782SGreg Clayton   std::string m_key_name;
139844d93782SGreg Clayton   uint64_t m_identifier;
139944d93782SGreg Clayton   Type m_type;
140044d93782SGreg Clayton   int m_key_value;
140144d93782SGreg Clayton   int m_start_col;
140244d93782SGreg Clayton   int m_max_submenu_name_length;
140344d93782SGreg Clayton   int m_max_submenu_key_name_length;
140444d93782SGreg Clayton   int m_selected;
140544d93782SGreg Clayton   Menu *m_parent;
140644d93782SGreg Clayton   Menus m_submenus;
140744d93782SGreg Clayton   WindowSP m_menu_window_sp;
140844d93782SGreg Clayton   MenuActionResult m_canned_result;
140944d93782SGreg Clayton   MenuDelegateSP m_delegate_sp;
141044d93782SGreg Clayton };
141144d93782SGreg Clayton 
141244d93782SGreg Clayton // Menubar or separator constructor
1413b9c1b51eSKate Stone Menu::Menu(Type type)
1414b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(0), m_type(type), m_key_value(0),
1415b9c1b51eSKate Stone       m_start_col(0), m_max_submenu_name_length(0),
1416b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1417b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1418b9c1b51eSKate Stone       m_delegate_sp() {}
141944d93782SGreg Clayton 
142044d93782SGreg Clayton // Menuitem constructor
1421b9c1b51eSKate Stone Menu::Menu(const char *name, const char *key_name, int key_value,
1422b9c1b51eSKate Stone            uint64_t identifier)
1423b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(identifier), m_type(Type::Invalid),
1424b9c1b51eSKate Stone       m_key_value(key_value), m_start_col(0), m_max_submenu_name_length(0),
1425b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1426b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1427b9c1b51eSKate Stone       m_delegate_sp() {
1428b9c1b51eSKate Stone   if (name && name[0]) {
142944d93782SGreg Clayton     m_name = name;
143044d93782SGreg Clayton     m_type = Type::Item;
143144d93782SGreg Clayton     if (key_name && key_name[0])
143244d93782SGreg Clayton       m_key_name = key_name;
1433b9c1b51eSKate Stone   } else {
143444d93782SGreg Clayton     m_type = Type::Separator;
143544d93782SGreg Clayton   }
143644d93782SGreg Clayton }
143744d93782SGreg Clayton 
1438b9c1b51eSKate Stone void Menu::RecalculateNameLengths() {
143944d93782SGreg Clayton   m_max_submenu_name_length = 0;
144044d93782SGreg Clayton   m_max_submenu_key_name_length = 0;
144144d93782SGreg Clayton   Menus &submenus = GetSubmenus();
144244d93782SGreg Clayton   const size_t num_submenus = submenus.size();
1443b9c1b51eSKate Stone   for (size_t i = 0; i < num_submenus; ++i) {
144444d93782SGreg Clayton     Menu *submenu = submenus[i].get();
14453985c8c6SSaleem Abdulrasool     if (static_cast<size_t>(m_max_submenu_name_length) < submenu->m_name.size())
144644d93782SGreg Clayton       m_max_submenu_name_length = submenu->m_name.size();
1447b9c1b51eSKate Stone     if (static_cast<size_t>(m_max_submenu_key_name_length) <
1448b9c1b51eSKate Stone         submenu->m_key_name.size())
144944d93782SGreg Clayton       m_max_submenu_key_name_length = submenu->m_key_name.size();
145044d93782SGreg Clayton   }
145144d93782SGreg Clayton }
145244d93782SGreg Clayton 
1453b9c1b51eSKate Stone void Menu::AddSubmenu(const MenuSP &menu_sp) {
145444d93782SGreg Clayton   menu_sp->m_parent = this;
14553985c8c6SSaleem Abdulrasool   if (static_cast<size_t>(m_max_submenu_name_length) < menu_sp->m_name.size())
145644d93782SGreg Clayton     m_max_submenu_name_length = menu_sp->m_name.size();
1457b9c1b51eSKate Stone   if (static_cast<size_t>(m_max_submenu_key_name_length) <
1458b9c1b51eSKate Stone       menu_sp->m_key_name.size())
145944d93782SGreg Clayton     m_max_submenu_key_name_length = menu_sp->m_key_name.size();
146044d93782SGreg Clayton   m_submenus.push_back(menu_sp);
146144d93782SGreg Clayton }
146244d93782SGreg Clayton 
1463b9c1b51eSKate Stone void Menu::DrawMenuTitle(Window &window, bool highlight) {
1464b9c1b51eSKate Stone   if (m_type == Type::Separator) {
146544d93782SGreg Clayton     window.MoveCursor(0, window.GetCursorY());
146644d93782SGreg Clayton     window.PutChar(ACS_LTEE);
146744d93782SGreg Clayton     int width = window.GetWidth();
1468b9c1b51eSKate Stone     if (width > 2) {
146944d93782SGreg Clayton       width -= 2;
14703985c8c6SSaleem Abdulrasool       for (int i = 0; i < width; ++i)
147144d93782SGreg Clayton         window.PutChar(ACS_HLINE);
147244d93782SGreg Clayton     }
147344d93782SGreg Clayton     window.PutChar(ACS_RTEE);
1474b9c1b51eSKate Stone   } else {
147544d93782SGreg Clayton     const int shortcut_key = m_key_value;
147644d93782SGreg Clayton     bool underlined_shortcut = false;
147744d93782SGreg Clayton     const attr_t hilgight_attr = A_REVERSE;
147844d93782SGreg Clayton     if (highlight)
147944d93782SGreg Clayton       window.AttributeOn(hilgight_attr);
1480b9c1b51eSKate Stone     if (isprint(shortcut_key)) {
148144d93782SGreg Clayton       size_t lower_pos = m_name.find(tolower(shortcut_key));
148244d93782SGreg Clayton       size_t upper_pos = m_name.find(toupper(shortcut_key));
148344d93782SGreg Clayton       const char *name = m_name.c_str();
148444d93782SGreg Clayton       size_t pos = std::min<size_t>(lower_pos, upper_pos);
1485b9c1b51eSKate Stone       if (pos != std::string::npos) {
148644d93782SGreg Clayton         underlined_shortcut = true;
1487b9c1b51eSKate Stone         if (pos > 0) {
148844d93782SGreg Clayton           window.PutCString(name, pos);
148944d93782SGreg Clayton           name += pos;
149044d93782SGreg Clayton         }
149144d93782SGreg Clayton         const attr_t shortcut_attr = A_UNDERLINE | A_BOLD;
149244d93782SGreg Clayton         window.AttributeOn(shortcut_attr);
149344d93782SGreg Clayton         window.PutChar(name[0]);
149444d93782SGreg Clayton         window.AttributeOff(shortcut_attr);
149544d93782SGreg Clayton         name++;
149644d93782SGreg Clayton         if (name[0])
149744d93782SGreg Clayton           window.PutCString(name);
149844d93782SGreg Clayton       }
149944d93782SGreg Clayton     }
150044d93782SGreg Clayton 
1501b9c1b51eSKate Stone     if (!underlined_shortcut) {
150244d93782SGreg Clayton       window.PutCString(m_name.c_str());
150344d93782SGreg Clayton     }
150444d93782SGreg Clayton 
150544d93782SGreg Clayton     if (highlight)
150644d93782SGreg Clayton       window.AttributeOff(hilgight_attr);
150744d93782SGreg Clayton 
1508b9c1b51eSKate Stone     if (m_key_name.empty()) {
1509b9c1b51eSKate Stone       if (!underlined_shortcut && isprint(m_key_value)) {
151044d93782SGreg Clayton         window.AttributeOn(COLOR_PAIR(3));
151144d93782SGreg Clayton         window.Printf(" (%c)", m_key_value);
151244d93782SGreg Clayton         window.AttributeOff(COLOR_PAIR(3));
151344d93782SGreg Clayton       }
1514b9c1b51eSKate Stone     } else {
151544d93782SGreg Clayton       window.AttributeOn(COLOR_PAIR(3));
151644d93782SGreg Clayton       window.Printf(" (%s)", m_key_name.c_str());
151744d93782SGreg Clayton       window.AttributeOff(COLOR_PAIR(3));
151844d93782SGreg Clayton     }
151944d93782SGreg Clayton   }
152044d93782SGreg Clayton }
152144d93782SGreg Clayton 
1522b9c1b51eSKate Stone bool Menu::WindowDelegateDraw(Window &window, bool force) {
152344d93782SGreg Clayton   Menus &submenus = GetSubmenus();
152444d93782SGreg Clayton   const size_t num_submenus = submenus.size();
152544d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
152644d93782SGreg Clayton   Menu::Type menu_type = GetType();
1527b9c1b51eSKate Stone   switch (menu_type) {
1528b9c1b51eSKate Stone   case Menu::Type::Bar: {
152944d93782SGreg Clayton     window.SetBackground(2);
153044d93782SGreg Clayton     window.MoveCursor(0, 0);
1531b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
153244d93782SGreg Clayton       Menu *menu = submenus[i].get();
153344d93782SGreg Clayton       if (i > 0)
153444d93782SGreg Clayton         window.PutChar(' ');
153544d93782SGreg Clayton       menu->SetStartingColumn(window.GetCursorX());
153644d93782SGreg Clayton       window.PutCString("| ");
153744d93782SGreg Clayton       menu->DrawMenuTitle(window, false);
153844d93782SGreg Clayton     }
153944d93782SGreg Clayton     window.PutCString(" |");
1540b9c1b51eSKate Stone   } break;
154144d93782SGreg Clayton 
1542b9c1b51eSKate Stone   case Menu::Type::Item: {
154344d93782SGreg Clayton     int y = 1;
154444d93782SGreg Clayton     int x = 3;
154544d93782SGreg Clayton     // Draw the menu
154644d93782SGreg Clayton     int cursor_x = 0;
154744d93782SGreg Clayton     int cursor_y = 0;
154844d93782SGreg Clayton     window.Erase();
154944d93782SGreg Clayton     window.SetBackground(2);
155044d93782SGreg Clayton     window.Box();
1551b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
1552b9c1b51eSKate Stone       const bool is_selected = (i == static_cast<size_t>(selected_idx));
155344d93782SGreg Clayton       window.MoveCursor(x, y + i);
1554b9c1b51eSKate Stone       if (is_selected) {
155544d93782SGreg Clayton         // Remember where we want the cursor to be
155644d93782SGreg Clayton         cursor_x = x - 1;
155744d93782SGreg Clayton         cursor_y = y + i;
155844d93782SGreg Clayton       }
155944d93782SGreg Clayton       submenus[i]->DrawMenuTitle(window, is_selected);
156044d93782SGreg Clayton     }
156144d93782SGreg Clayton     window.MoveCursor(cursor_x, cursor_y);
1562b9c1b51eSKate Stone   } break;
156344d93782SGreg Clayton 
156444d93782SGreg Clayton   default:
156544d93782SGreg Clayton   case Menu::Type::Separator:
156644d93782SGreg Clayton     break;
156744d93782SGreg Clayton   }
156844d93782SGreg Clayton   return true; // Drawing handled...
156944d93782SGreg Clayton }
157044d93782SGreg Clayton 
1571b9c1b51eSKate Stone HandleCharResult Menu::WindowDelegateHandleChar(Window &window, int key) {
157244d93782SGreg Clayton   HandleCharResult result = eKeyNotHandled;
157344d93782SGreg Clayton 
157444d93782SGreg Clayton   Menus &submenus = GetSubmenus();
157544d93782SGreg Clayton   const size_t num_submenus = submenus.size();
157644d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
157744d93782SGreg Clayton   Menu::Type menu_type = GetType();
1578b9c1b51eSKate Stone   if (menu_type == Menu::Type::Bar) {
157944d93782SGreg Clayton     MenuSP run_menu_sp;
1580b9c1b51eSKate Stone     switch (key) {
158144d93782SGreg Clayton     case KEY_DOWN:
158244d93782SGreg Clayton     case KEY_UP:
158344d93782SGreg Clayton       // Show last menu or first menu
15843985c8c6SSaleem Abdulrasool       if (selected_idx < static_cast<int>(num_submenus))
158544d93782SGreg Clayton         run_menu_sp = submenus[selected_idx];
158644d93782SGreg Clayton       else if (!submenus.empty())
158744d93782SGreg Clayton         run_menu_sp = submenus.front();
158844d93782SGreg Clayton       result = eKeyHandled;
158944d93782SGreg Clayton       break;
159044d93782SGreg Clayton 
159144d93782SGreg Clayton     case KEY_RIGHT:
159244d93782SGreg Clayton       ++m_selected;
15933985c8c6SSaleem Abdulrasool       if (m_selected >= static_cast<int>(num_submenus))
159444d93782SGreg Clayton         m_selected = 0;
15953985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
159644d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
159744d93782SGreg Clayton       else if (!submenus.empty())
159844d93782SGreg Clayton         run_menu_sp = submenus.front();
159944d93782SGreg Clayton       result = eKeyHandled;
160044d93782SGreg Clayton       break;
160144d93782SGreg Clayton 
160244d93782SGreg Clayton     case KEY_LEFT:
160344d93782SGreg Clayton       --m_selected;
160444d93782SGreg Clayton       if (m_selected < 0)
160544d93782SGreg Clayton         m_selected = num_submenus - 1;
16063985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
160744d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
160844d93782SGreg Clayton       else if (!submenus.empty())
160944d93782SGreg Clayton         run_menu_sp = submenus.front();
161044d93782SGreg Clayton       result = eKeyHandled;
161144d93782SGreg Clayton       break;
161244d93782SGreg Clayton 
161344d93782SGreg Clayton     default:
1614b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
1615b9c1b51eSKate Stone         if (submenus[i]->GetKeyValue() == key) {
161644d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
161744d93782SGreg Clayton           run_menu_sp = submenus[i];
161844d93782SGreg Clayton           result = eKeyHandled;
161944d93782SGreg Clayton           break;
162044d93782SGreg Clayton         }
162144d93782SGreg Clayton       }
162244d93782SGreg Clayton       break;
162344d93782SGreg Clayton     }
162444d93782SGreg Clayton 
1625b9c1b51eSKate Stone     if (run_menu_sp) {
162605097246SAdrian Prantl       // Run the action on this menu in case we need to populate the menu with
162705097246SAdrian Prantl       // dynamic content and also in case check marks, and any other menu
162805097246SAdrian Prantl       // decorations need to be calculated
162944d93782SGreg Clayton       if (run_menu_sp->Action() == MenuActionResult::Quit)
163044d93782SGreg Clayton         return eQuitApplication;
163144d93782SGreg Clayton 
163244d93782SGreg Clayton       Rect menu_bounds;
163344d93782SGreg Clayton       menu_bounds.origin.x = run_menu_sp->GetStartingColumn();
163444d93782SGreg Clayton       menu_bounds.origin.y = 1;
163544d93782SGreg Clayton       menu_bounds.size.width = run_menu_sp->GetDrawWidth();
163644d93782SGreg Clayton       menu_bounds.size.height = run_menu_sp->GetSubmenus().size() + 2;
163744d93782SGreg Clayton       if (m_menu_window_sp)
163844d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(m_menu_window_sp.get());
163944d93782SGreg Clayton 
1640b9c1b51eSKate Stone       m_menu_window_sp = window.GetParent()->CreateSubWindow(
1641b9c1b51eSKate Stone           run_menu_sp->GetName().c_str(), menu_bounds, true);
164244d93782SGreg Clayton       m_menu_window_sp->SetDelegate(run_menu_sp);
164344d93782SGreg Clayton     }
1644b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Item) {
1645b9c1b51eSKate Stone     switch (key) {
164644d93782SGreg Clayton     case KEY_DOWN:
1647b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
164844d93782SGreg Clayton         const int start_select = m_selected;
1649b9c1b51eSKate Stone         while (++m_selected != start_select) {
16503985c8c6SSaleem Abdulrasool           if (static_cast<size_t>(m_selected) >= num_submenus)
165144d93782SGreg Clayton             m_selected = 0;
165244d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
165344d93782SGreg Clayton             continue;
165444d93782SGreg Clayton           else
165544d93782SGreg Clayton             break;
165644d93782SGreg Clayton         }
165744d93782SGreg Clayton         return eKeyHandled;
165844d93782SGreg Clayton       }
165944d93782SGreg Clayton       break;
166044d93782SGreg Clayton 
166144d93782SGreg Clayton     case KEY_UP:
1662b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
166344d93782SGreg Clayton         const int start_select = m_selected;
1664b9c1b51eSKate Stone         while (--m_selected != start_select) {
16653985c8c6SSaleem Abdulrasool           if (m_selected < static_cast<int>(0))
166644d93782SGreg Clayton             m_selected = num_submenus - 1;
166744d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
166844d93782SGreg Clayton             continue;
166944d93782SGreg Clayton           else
167044d93782SGreg Clayton             break;
167144d93782SGreg Clayton         }
167244d93782SGreg Clayton         return eKeyHandled;
167344d93782SGreg Clayton       }
167444d93782SGreg Clayton       break;
167544d93782SGreg Clayton 
167644d93782SGreg Clayton     case KEY_RETURN:
1677b9c1b51eSKate Stone       if (static_cast<size_t>(selected_idx) < num_submenus) {
167844d93782SGreg Clayton         if (submenus[selected_idx]->Action() == MenuActionResult::Quit)
167944d93782SGreg Clayton           return eQuitApplication;
168044d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(&window);
168144d93782SGreg Clayton         return eKeyHandled;
168244d93782SGreg Clayton       }
168344d93782SGreg Clayton       break;
168444d93782SGreg Clayton 
1685b9c1b51eSKate Stone     case KEY_ESCAPE: // Beware: pressing escape key has 1 to 2 second delay in
1686b9c1b51eSKate Stone                      // case other chars are entered for escaped sequences
168744d93782SGreg Clayton       window.GetParent()->RemoveSubWindow(&window);
168844d93782SGreg Clayton       return eKeyHandled;
168944d93782SGreg Clayton 
169044d93782SGreg Clayton     default:
1691b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
169244d93782SGreg Clayton         Menu *menu = submenus[i].get();
1693b9c1b51eSKate Stone         if (menu->GetKeyValue() == key) {
169444d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
169544d93782SGreg Clayton           window.GetParent()->RemoveSubWindow(&window);
169644d93782SGreg Clayton           if (menu->Action() == MenuActionResult::Quit)
169744d93782SGreg Clayton             return eQuitApplication;
169844d93782SGreg Clayton           return eKeyHandled;
169944d93782SGreg Clayton         }
170044d93782SGreg Clayton       }
170144d93782SGreg Clayton       break;
170244d93782SGreg Clayton     }
1703b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Separator) {
170444d93782SGreg Clayton   }
170544d93782SGreg Clayton   return result;
170644d93782SGreg Clayton }
170744d93782SGreg Clayton 
1708b9c1b51eSKate Stone class Application {
170944d93782SGreg Clayton public:
1710b9c1b51eSKate Stone   Application(FILE *in, FILE *out)
1711b9c1b51eSKate Stone       : m_window_sp(), m_screen(nullptr), m_in(in), m_out(out) {}
171244d93782SGreg Clayton 
1713b9c1b51eSKate Stone   ~Application() {
171444d93782SGreg Clayton     m_window_delegates.clear();
171544d93782SGreg Clayton     m_window_sp.reset();
1716b9c1b51eSKate Stone     if (m_screen) {
171744d93782SGreg Clayton       ::delscreen(m_screen);
1718c5dac77aSEugene Zelenko       m_screen = nullptr;
171944d93782SGreg Clayton     }
172044d93782SGreg Clayton   }
172144d93782SGreg Clayton 
1722b9c1b51eSKate Stone   void Initialize() {
172344d93782SGreg Clayton     ::setlocale(LC_ALL, "");
172444d93782SGreg Clayton     ::setlocale(LC_CTYPE, "");
1725c5dac77aSEugene Zelenko     m_screen = ::newterm(nullptr, m_out, m_in);
172644d93782SGreg Clayton     ::start_color();
172744d93782SGreg Clayton     ::curs_set(0);
172844d93782SGreg Clayton     ::noecho();
172944d93782SGreg Clayton     ::keypad(stdscr, TRUE);
173044d93782SGreg Clayton   }
173144d93782SGreg Clayton 
1732b9c1b51eSKate Stone   void Terminate() { ::endwin(); }
173344d93782SGreg Clayton 
1734b9c1b51eSKate Stone   void Run(Debugger &debugger) {
173544d93782SGreg Clayton     bool done = false;
173644d93782SGreg Clayton     int delay_in_tenths_of_a_second = 1;
173744d93782SGreg Clayton 
173805097246SAdrian Prantl     // Alas the threading model in curses is a bit lame so we need to resort to
173905097246SAdrian Prantl     // polling every 0.5 seconds. We could poll for stdin ourselves and then
174005097246SAdrian Prantl     // pass the keys down but then we need to translate all of the escape
174105097246SAdrian Prantl     // sequences ourselves. So we resort to polling for input because we need
174205097246SAdrian Prantl     // to receive async process events while in this loop.
174344d93782SGreg Clayton 
1744b9c1b51eSKate Stone     halfdelay(delay_in_tenths_of_a_second); // Poll using some number of tenths
1745b9c1b51eSKate Stone                                             // of seconds seconds when calling
1746b9c1b51eSKate Stone                                             // Window::GetChar()
174744d93782SGreg Clayton 
1748b9c1b51eSKate Stone     ListenerSP listener_sp(
1749b9c1b51eSKate Stone         Listener::MakeListener("lldb.IOHandler.curses.Application"));
175044d93782SGreg Clayton     ConstString broadcaster_class_target(Target::GetStaticBroadcasterClass());
175144d93782SGreg Clayton     ConstString broadcaster_class_process(Process::GetStaticBroadcasterClass());
175244d93782SGreg Clayton     ConstString broadcaster_class_thread(Thread::GetStaticBroadcasterClass());
175344d93782SGreg Clayton     debugger.EnableForwardEvents(listener_sp);
175444d93782SGreg Clayton 
175544d93782SGreg Clayton     bool update = true;
175644d93782SGreg Clayton #if defined(__APPLE__)
175744d93782SGreg Clayton     std::deque<int> escape_chars;
175844d93782SGreg Clayton #endif
175944d93782SGreg Clayton 
1760b9c1b51eSKate Stone     while (!done) {
1761b9c1b51eSKate Stone       if (update) {
176244d93782SGreg Clayton         m_window_sp->Draw(false);
176305097246SAdrian Prantl         // All windows should be calling Window::DeferredRefresh() instead of
176405097246SAdrian Prantl         // Window::Refresh() so we can do a single update and avoid any screen
176505097246SAdrian Prantl         // blinking
176644d93782SGreg Clayton         update_panels();
176744d93782SGreg Clayton 
1768b9c1b51eSKate Stone         // Cursor hiding isn't working on MacOSX, so hide it in the top left
1769b9c1b51eSKate Stone         // corner
177044d93782SGreg Clayton         m_window_sp->MoveCursor(0, 0);
177144d93782SGreg Clayton 
177244d93782SGreg Clayton         doupdate();
177344d93782SGreg Clayton         update = false;
177444d93782SGreg Clayton       }
177544d93782SGreg Clayton 
177644d93782SGreg Clayton #if defined(__APPLE__)
177705097246SAdrian Prantl       // Terminal.app doesn't map its function keys correctly, F1-F4 default
177805097246SAdrian Prantl       // to: \033OP, \033OQ, \033OR, \033OS, so lets take care of this here if
1779b9c1b51eSKate Stone       // possible
178044d93782SGreg Clayton       int ch;
178144d93782SGreg Clayton       if (escape_chars.empty())
178244d93782SGreg Clayton         ch = m_window_sp->GetChar();
1783b9c1b51eSKate Stone       else {
178444d93782SGreg Clayton         ch = escape_chars.front();
178544d93782SGreg Clayton         escape_chars.pop_front();
178644d93782SGreg Clayton       }
1787b9c1b51eSKate Stone       if (ch == KEY_ESCAPE) {
178844d93782SGreg Clayton         int ch2 = m_window_sp->GetChar();
1789b9c1b51eSKate Stone         if (ch2 == 'O') {
179044d93782SGreg Clayton           int ch3 = m_window_sp->GetChar();
1791b9c1b51eSKate Stone           switch (ch3) {
1792b9c1b51eSKate Stone           case 'P':
1793b9c1b51eSKate Stone             ch = KEY_F(1);
1794b9c1b51eSKate Stone             break;
1795b9c1b51eSKate Stone           case 'Q':
1796b9c1b51eSKate Stone             ch = KEY_F(2);
1797b9c1b51eSKate Stone             break;
1798b9c1b51eSKate Stone           case 'R':
1799b9c1b51eSKate Stone             ch = KEY_F(3);
1800b9c1b51eSKate Stone             break;
1801b9c1b51eSKate Stone           case 'S':
1802b9c1b51eSKate Stone             ch = KEY_F(4);
1803b9c1b51eSKate Stone             break;
180444d93782SGreg Clayton           default:
180544d93782SGreg Clayton             escape_chars.push_back(ch2);
180644d93782SGreg Clayton             if (ch3 != -1)
180744d93782SGreg Clayton               escape_chars.push_back(ch3);
180844d93782SGreg Clayton             break;
180944d93782SGreg Clayton           }
1810b9c1b51eSKate Stone         } else if (ch2 != -1)
181144d93782SGreg Clayton           escape_chars.push_back(ch2);
181244d93782SGreg Clayton       }
181344d93782SGreg Clayton #else
181444d93782SGreg Clayton       int ch = m_window_sp->GetChar();
181544d93782SGreg Clayton 
181644d93782SGreg Clayton #endif
1817b9c1b51eSKate Stone       if (ch == -1) {
1818b9c1b51eSKate Stone         if (feof(m_in) || ferror(m_in)) {
181944d93782SGreg Clayton           done = true;
1820b9c1b51eSKate Stone         } else {
182144d93782SGreg Clayton           // Just a timeout from using halfdelay(), check for events
182244d93782SGreg Clayton           EventSP event_sp;
1823b9c1b51eSKate Stone           while (listener_sp->PeekAtNextEvent()) {
1824d35031e1SPavel Labath             listener_sp->GetEvent(event_sp, std::chrono::seconds(0));
182544d93782SGreg Clayton 
1826b9c1b51eSKate Stone             if (event_sp) {
182744d93782SGreg Clayton               Broadcaster *broadcaster = event_sp->GetBroadcaster();
1828b9c1b51eSKate Stone               if (broadcaster) {
182944d93782SGreg Clayton                 // uint32_t event_type = event_sp->GetType();
1830b9c1b51eSKate Stone                 ConstString broadcaster_class(
1831b9c1b51eSKate Stone                     broadcaster->GetBroadcasterClass());
1832b9c1b51eSKate Stone                 if (broadcaster_class == broadcaster_class_process) {
1833b9c1b51eSKate Stone                   debugger.GetCommandInterpreter().UpdateExecutionContext(
1834b9c1b51eSKate Stone                       nullptr);
183544d93782SGreg Clayton                   update = true;
183644d93782SGreg Clayton                   continue; // Don't get any key, just update our view
183744d93782SGreg Clayton                 }
183844d93782SGreg Clayton               }
183944d93782SGreg Clayton             }
184044d93782SGreg Clayton           }
184144d93782SGreg Clayton         }
1842b9c1b51eSKate Stone       } else {
184344d93782SGreg Clayton         HandleCharResult key_result = m_window_sp->HandleChar(ch);
1844b9c1b51eSKate Stone         switch (key_result) {
184544d93782SGreg Clayton         case eKeyHandled:
1846c5dac77aSEugene Zelenko           debugger.GetCommandInterpreter().UpdateExecutionContext(nullptr);
184744d93782SGreg Clayton           update = true;
184844d93782SGreg Clayton           break;
184944d93782SGreg Clayton         case eKeyNotHandled:
185044d93782SGreg Clayton           break;
185144d93782SGreg Clayton         case eQuitApplication:
185244d93782SGreg Clayton           done = true;
185344d93782SGreg Clayton           break;
185444d93782SGreg Clayton         }
185544d93782SGreg Clayton       }
185644d93782SGreg Clayton     }
185744d93782SGreg Clayton 
185844d93782SGreg Clayton     debugger.CancelForwardEvents(listener_sp);
185944d93782SGreg Clayton   }
186044d93782SGreg Clayton 
1861b9c1b51eSKate Stone   WindowSP &GetMainWindow() {
186244d93782SGreg Clayton     if (!m_window_sp)
1863796ac80bSJonas Devlieghere       m_window_sp = std::make_shared<Window>("main", stdscr, false);
186444d93782SGreg Clayton     return m_window_sp;
186544d93782SGreg Clayton   }
186644d93782SGreg Clayton 
186744d93782SGreg Clayton protected:
186844d93782SGreg Clayton   WindowSP m_window_sp;
186944d93782SGreg Clayton   WindowDelegates m_window_delegates;
187044d93782SGreg Clayton   SCREEN *m_screen;
187144d93782SGreg Clayton   FILE *m_in;
187244d93782SGreg Clayton   FILE *m_out;
187344d93782SGreg Clayton };
187444d93782SGreg Clayton 
187544d93782SGreg Clayton } // namespace curses
187644d93782SGreg Clayton 
187744d93782SGreg Clayton using namespace curses;
187844d93782SGreg Clayton 
1879b9c1b51eSKate Stone struct Row {
18808369b28dSGreg Clayton   ValueObjectManager value;
188144d93782SGreg Clayton   Row *parent;
18828369b28dSGreg Clayton   // The process stop ID when the children were calculated.
18838369b28dSGreg Clayton   uint32_t children_stop_id;
188444d93782SGreg Clayton   int row_idx;
188544d93782SGreg Clayton   int x;
188644d93782SGreg Clayton   int y;
188744d93782SGreg Clayton   bool might_have_children;
188844d93782SGreg Clayton   bool expanded;
188944d93782SGreg Clayton   bool calculated_children;
189044d93782SGreg Clayton   std::vector<Row> children;
189144d93782SGreg Clayton 
1892b9c1b51eSKate Stone   Row(const ValueObjectSP &v, Row *p)
18938369b28dSGreg Clayton       : value(v, lldb::eDynamicDontRunTarget, true), parent(p), row_idx(0),
18948369b28dSGreg Clayton         x(1), y(1), might_have_children(v ? v->MightHaveChildren() : false),
1895b9c1b51eSKate Stone         expanded(false), calculated_children(false), children() {}
189644d93782SGreg Clayton 
1897b9c1b51eSKate Stone   size_t GetDepth() const {
189844d93782SGreg Clayton     if (parent)
189944d93782SGreg Clayton       return 1 + parent->GetDepth();
190044d93782SGreg Clayton     return 0;
190144d93782SGreg Clayton   }
190244d93782SGreg Clayton 
1903171dd2e6SJonas Devlieghere   void Expand() { expanded = true; }
19048369b28dSGreg Clayton 
19058369b28dSGreg Clayton   std::vector<Row> &GetChildren() {
19068369b28dSGreg Clayton     ProcessSP process_sp = value.GetProcessSP();
19078369b28dSGreg Clayton     auto stop_id = process_sp->GetStopID();
19088369b28dSGreg Clayton     if (process_sp && stop_id != children_stop_id) {
19098369b28dSGreg Clayton       children_stop_id = stop_id;
19108369b28dSGreg Clayton       calculated_children = false;
19118369b28dSGreg Clayton     }
1912b9c1b51eSKate Stone     if (!calculated_children) {
19138369b28dSGreg Clayton       children.clear();
191444d93782SGreg Clayton       calculated_children = true;
19158369b28dSGreg Clayton       ValueObjectSP valobj = value.GetSP();
1916b9c1b51eSKate Stone       if (valobj) {
191744d93782SGreg Clayton         const size_t num_children = valobj->GetNumChildren();
1918b9c1b51eSKate Stone         for (size_t i = 0; i < num_children; ++i) {
191944d93782SGreg Clayton           children.push_back(Row(valobj->GetChildAtIndex(i, true), this));
192044d93782SGreg Clayton         }
192144d93782SGreg Clayton       }
192244d93782SGreg Clayton     }
19238369b28dSGreg Clayton     return children;
192444d93782SGreg Clayton   }
192544d93782SGreg Clayton 
19268369b28dSGreg Clayton   void Unexpand() {
19278369b28dSGreg Clayton     expanded = false;
19288369b28dSGreg Clayton     calculated_children = false;
19298369b28dSGreg Clayton     children.clear();
19308369b28dSGreg Clayton   }
193144d93782SGreg Clayton 
1932b9c1b51eSKate Stone   void DrawTree(Window &window) {
193344d93782SGreg Clayton     if (parent)
193444d93782SGreg Clayton       parent->DrawTreeForChild(window, this, 0);
193544d93782SGreg Clayton 
1936b9c1b51eSKate Stone     if (might_have_children) {
193705097246SAdrian Prantl       // It we can get UTF8 characters to work we should try to use the
193805097246SAdrian Prantl       // "symbol" UTF8 string below
193944d93782SGreg Clayton       //            const char *symbol = "";
194044d93782SGreg Clayton       //            if (row.expanded)
194144d93782SGreg Clayton       //                symbol = "\xe2\x96\xbd ";
194244d93782SGreg Clayton       //            else
194344d93782SGreg Clayton       //                symbol = "\xe2\x96\xb7 ";
194444d93782SGreg Clayton       //            window.PutCString (symbol);
194544d93782SGreg Clayton 
194605097246SAdrian Prantl       // The ACS_DARROW and ACS_RARROW don't look very nice they are just a 'v'
194705097246SAdrian Prantl       // or '>' character...
194844d93782SGreg Clayton       //            if (expanded)
194944d93782SGreg Clayton       //                window.PutChar (ACS_DARROW);
195044d93782SGreg Clayton       //            else
195144d93782SGreg Clayton       //                window.PutChar (ACS_RARROW);
195205097246SAdrian Prantl       // Since we can't find any good looking right arrow/down arrow symbols,
195305097246SAdrian Prantl       // just use a diamond...
195444d93782SGreg Clayton       window.PutChar(ACS_DIAMOND);
195544d93782SGreg Clayton       window.PutChar(ACS_HLINE);
195644d93782SGreg Clayton     }
195744d93782SGreg Clayton   }
195844d93782SGreg Clayton 
1959b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, Row *child, uint32_t reverse_depth) {
196044d93782SGreg Clayton     if (parent)
196144d93782SGreg Clayton       parent->DrawTreeForChild(window, this, reverse_depth + 1);
196244d93782SGreg Clayton 
19638369b28dSGreg Clayton     if (&GetChildren().back() == child) {
196444d93782SGreg Clayton       // Last child
1965b9c1b51eSKate Stone       if (reverse_depth == 0) {
196644d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
196744d93782SGreg Clayton         window.PutChar(ACS_HLINE);
1968b9c1b51eSKate Stone       } else {
196944d93782SGreg Clayton         window.PutChar(' ');
197044d93782SGreg Clayton         window.PutChar(' ');
197144d93782SGreg Clayton       }
1972b9c1b51eSKate Stone     } else {
1973b9c1b51eSKate Stone       if (reverse_depth == 0) {
197444d93782SGreg Clayton         window.PutChar(ACS_LTEE);
197544d93782SGreg Clayton         window.PutChar(ACS_HLINE);
1976b9c1b51eSKate Stone       } else {
197744d93782SGreg Clayton         window.PutChar(ACS_VLINE);
197844d93782SGreg Clayton         window.PutChar(' ');
197944d93782SGreg Clayton       }
198044d93782SGreg Clayton     }
198144d93782SGreg Clayton   }
198244d93782SGreg Clayton };
198344d93782SGreg Clayton 
1984b9c1b51eSKate Stone struct DisplayOptions {
198544d93782SGreg Clayton   bool show_types;
198644d93782SGreg Clayton };
198744d93782SGreg Clayton 
198844d93782SGreg Clayton class TreeItem;
198944d93782SGreg Clayton 
1990b9c1b51eSKate Stone class TreeDelegate {
199144d93782SGreg Clayton public:
1992c5dac77aSEugene Zelenko   TreeDelegate() = default;
1993315b6884SEugene Zelenko   virtual ~TreeDelegate() = default;
1994315b6884SEugene Zelenko 
199544d93782SGreg Clayton   virtual void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) = 0;
199644d93782SGreg Clayton   virtual void TreeDelegateGenerateChildren(TreeItem &item) = 0;
1997b9c1b51eSKate Stone   virtual bool TreeDelegateItemSelected(
1998b9c1b51eSKate Stone       TreeItem &item) = 0; // Return true if we need to update views
199944d93782SGreg Clayton };
2000315b6884SEugene Zelenko 
200144d93782SGreg Clayton typedef std::shared_ptr<TreeDelegate> TreeDelegateSP;
200244d93782SGreg Clayton 
2003b9c1b51eSKate Stone class TreeItem {
200444d93782SGreg Clayton public:
2005b9c1b51eSKate Stone   TreeItem(TreeItem *parent, TreeDelegate &delegate, bool might_have_children)
2006b9c1b51eSKate Stone       : m_parent(parent), m_delegate(delegate), m_user_data(nullptr),
2007b9c1b51eSKate Stone         m_identifier(0), m_row_idx(-1), m_children(),
2008b9c1b51eSKate Stone         m_might_have_children(might_have_children), m_is_expanded(false) {}
200944d93782SGreg Clayton 
2010b9c1b51eSKate Stone   TreeItem &operator=(const TreeItem &rhs) {
2011b9c1b51eSKate Stone     if (this != &rhs) {
201244d93782SGreg Clayton       m_parent = rhs.m_parent;
201344d93782SGreg Clayton       m_delegate = rhs.m_delegate;
2014ec990867SGreg Clayton       m_user_data = rhs.m_user_data;
201544d93782SGreg Clayton       m_identifier = rhs.m_identifier;
201644d93782SGreg Clayton       m_row_idx = rhs.m_row_idx;
201744d93782SGreg Clayton       m_children = rhs.m_children;
201844d93782SGreg Clayton       m_might_have_children = rhs.m_might_have_children;
201944d93782SGreg Clayton       m_is_expanded = rhs.m_is_expanded;
202044d93782SGreg Clayton     }
202144d93782SGreg Clayton     return *this;
202244d93782SGreg Clayton   }
202344d93782SGreg Clayton 
2024b9c1b51eSKate Stone   size_t GetDepth() const {
202544d93782SGreg Clayton     if (m_parent)
202644d93782SGreg Clayton       return 1 + m_parent->GetDepth();
202744d93782SGreg Clayton     return 0;
202844d93782SGreg Clayton   }
202944d93782SGreg Clayton 
2030b9c1b51eSKate Stone   int GetRowIndex() const { return m_row_idx; }
203144d93782SGreg Clayton 
2032b9c1b51eSKate Stone   void ClearChildren() { m_children.clear(); }
203344d93782SGreg Clayton 
2034b9c1b51eSKate Stone   void Resize(size_t n, const TreeItem &t) { m_children.resize(n, t); }
203544d93782SGreg Clayton 
2036b9c1b51eSKate Stone   TreeItem &operator[](size_t i) { return m_children[i]; }
203744d93782SGreg Clayton 
2038b9c1b51eSKate Stone   void SetRowIndex(int row_idx) { m_row_idx = row_idx; }
203944d93782SGreg Clayton 
2040b9c1b51eSKate Stone   size_t GetNumChildren() {
204144d93782SGreg Clayton     m_delegate.TreeDelegateGenerateChildren(*this);
204244d93782SGreg Clayton     return m_children.size();
204344d93782SGreg Clayton   }
204444d93782SGreg Clayton 
2045b9c1b51eSKate Stone   void ItemWasSelected() { m_delegate.TreeDelegateItemSelected(*this); }
2046315b6884SEugene Zelenko 
2047b9c1b51eSKate Stone   void CalculateRowIndexes(int &row_idx) {
204844d93782SGreg Clayton     SetRowIndex(row_idx);
204944d93782SGreg Clayton     ++row_idx;
205044d93782SGreg Clayton 
2051ec990867SGreg Clayton     const bool expanded = IsExpanded();
2052ec990867SGreg Clayton 
205305097246SAdrian Prantl     // The root item must calculate its children, or we must calculate the
205405097246SAdrian Prantl     // number of children if the item is expanded
2055c5dac77aSEugene Zelenko     if (m_parent == nullptr || expanded)
205644d93782SGreg Clayton       GetNumChildren();
205744d93782SGreg Clayton 
2058b9c1b51eSKate Stone     for (auto &item : m_children) {
205944d93782SGreg Clayton       if (expanded)
206044d93782SGreg Clayton         item.CalculateRowIndexes(row_idx);
206144d93782SGreg Clayton       else
206244d93782SGreg Clayton         item.SetRowIndex(-1);
206344d93782SGreg Clayton     }
206444d93782SGreg Clayton   }
206544d93782SGreg Clayton 
2066b9c1b51eSKate Stone   TreeItem *GetParent() { return m_parent; }
206744d93782SGreg Clayton 
2068b9c1b51eSKate Stone   bool IsExpanded() const { return m_is_expanded; }
206944d93782SGreg Clayton 
2070b9c1b51eSKate Stone   void Expand() { m_is_expanded = true; }
207144d93782SGreg Clayton 
2072b9c1b51eSKate Stone   void Unexpand() { m_is_expanded = false; }
207344d93782SGreg Clayton 
2074b9c1b51eSKate Stone   bool Draw(Window &window, const int first_visible_row,
2075b9c1b51eSKate Stone             const uint32_t selected_row_idx, int &row_idx, int &num_rows_left) {
207644d93782SGreg Clayton     if (num_rows_left <= 0)
207744d93782SGreg Clayton       return false;
207844d93782SGreg Clayton 
2079b9c1b51eSKate Stone     if (m_row_idx >= first_visible_row) {
208044d93782SGreg Clayton       window.MoveCursor(2, row_idx + 1);
208144d93782SGreg Clayton 
208244d93782SGreg Clayton       if (m_parent)
208344d93782SGreg Clayton         m_parent->DrawTreeForChild(window, this, 0);
208444d93782SGreg Clayton 
2085b9c1b51eSKate Stone       if (m_might_have_children) {
2086b9c1b51eSKate Stone         // It we can get UTF8 characters to work we should try to use the
208705097246SAdrian Prantl         // "symbol" UTF8 string below
208844d93782SGreg Clayton         //            const char *symbol = "";
208944d93782SGreg Clayton         //            if (row.expanded)
209044d93782SGreg Clayton         //                symbol = "\xe2\x96\xbd ";
209144d93782SGreg Clayton         //            else
209244d93782SGreg Clayton         //                symbol = "\xe2\x96\xb7 ";
209344d93782SGreg Clayton         //            window.PutCString (symbol);
209444d93782SGreg Clayton 
209544d93782SGreg Clayton         // The ACS_DARROW and ACS_RARROW don't look very nice they are just a
209644d93782SGreg Clayton         // 'v' or '>' character...
209744d93782SGreg Clayton         //            if (expanded)
209844d93782SGreg Clayton         //                window.PutChar (ACS_DARROW);
209944d93782SGreg Clayton         //            else
210044d93782SGreg Clayton         //                window.PutChar (ACS_RARROW);
210105097246SAdrian Prantl         // Since we can't find any good looking right arrow/down arrow symbols,
210205097246SAdrian Prantl         // just use a diamond...
210344d93782SGreg Clayton         window.PutChar(ACS_DIAMOND);
210444d93782SGreg Clayton         window.PutChar(ACS_HLINE);
210544d93782SGreg Clayton       }
2106b9c1b51eSKate Stone       bool highlight = (selected_row_idx == static_cast<size_t>(m_row_idx)) &&
2107b9c1b51eSKate Stone                        window.IsActive();
210844d93782SGreg Clayton 
210944d93782SGreg Clayton       if (highlight)
211044d93782SGreg Clayton         window.AttributeOn(A_REVERSE);
211144d93782SGreg Clayton 
211244d93782SGreg Clayton       m_delegate.TreeDelegateDrawTreeItem(*this, window);
211344d93782SGreg Clayton 
211444d93782SGreg Clayton       if (highlight)
211544d93782SGreg Clayton         window.AttributeOff(A_REVERSE);
211644d93782SGreg Clayton       ++row_idx;
211744d93782SGreg Clayton       --num_rows_left;
211844d93782SGreg Clayton     }
211944d93782SGreg Clayton 
212044d93782SGreg Clayton     if (num_rows_left <= 0)
212144d93782SGreg Clayton       return false; // We are done drawing...
212244d93782SGreg Clayton 
2123b9c1b51eSKate Stone     if (IsExpanded()) {
2124b9c1b51eSKate Stone       for (auto &item : m_children) {
212505097246SAdrian Prantl         // If we displayed all the rows and item.Draw() returns false we are
212605097246SAdrian Prantl         // done drawing and can exit this for loop
2127b9c1b51eSKate Stone         if (!item.Draw(window, first_visible_row, selected_row_idx, row_idx,
2128b9c1b51eSKate Stone                        num_rows_left))
212944d93782SGreg Clayton           break;
213044d93782SGreg Clayton       }
213144d93782SGreg Clayton     }
213244d93782SGreg Clayton     return num_rows_left >= 0; // Return true if not done drawing yet
213344d93782SGreg Clayton   }
213444d93782SGreg Clayton 
2135b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, TreeItem *child,
2136b9c1b51eSKate Stone                         uint32_t reverse_depth) {
213744d93782SGreg Clayton     if (m_parent)
213844d93782SGreg Clayton       m_parent->DrawTreeForChild(window, this, reverse_depth + 1);
213944d93782SGreg Clayton 
2140b9c1b51eSKate Stone     if (&m_children.back() == child) {
214144d93782SGreg Clayton       // Last child
2142b9c1b51eSKate Stone       if (reverse_depth == 0) {
214344d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
214444d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2145b9c1b51eSKate Stone       } else {
214644d93782SGreg Clayton         window.PutChar(' ');
214744d93782SGreg Clayton         window.PutChar(' ');
214844d93782SGreg Clayton       }
2149b9c1b51eSKate Stone     } else {
2150b9c1b51eSKate Stone       if (reverse_depth == 0) {
215144d93782SGreg Clayton         window.PutChar(ACS_LTEE);
215244d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2153b9c1b51eSKate Stone       } else {
215444d93782SGreg Clayton         window.PutChar(ACS_VLINE);
215544d93782SGreg Clayton         window.PutChar(' ');
215644d93782SGreg Clayton       }
215744d93782SGreg Clayton     }
215844d93782SGreg Clayton   }
215944d93782SGreg Clayton 
2160b9c1b51eSKate Stone   TreeItem *GetItemForRowIndex(uint32_t row_idx) {
21613985c8c6SSaleem Abdulrasool     if (static_cast<uint32_t>(m_row_idx) == row_idx)
216244d93782SGreg Clayton       return this;
216344d93782SGreg Clayton     if (m_children.empty())
2164c5dac77aSEugene Zelenko       return nullptr;
2165b9c1b51eSKate Stone     if (IsExpanded()) {
2166b9c1b51eSKate Stone       for (auto &item : m_children) {
216744d93782SGreg Clayton         TreeItem *selected_item_ptr = item.GetItemForRowIndex(row_idx);
216844d93782SGreg Clayton         if (selected_item_ptr)
216944d93782SGreg Clayton           return selected_item_ptr;
217044d93782SGreg Clayton       }
217144d93782SGreg Clayton     }
2172c5dac77aSEugene Zelenko     return nullptr;
217344d93782SGreg Clayton   }
217444d93782SGreg Clayton 
2175b9c1b51eSKate Stone   void *GetUserData() const { return m_user_data; }
2176ec990867SGreg Clayton 
2177b9c1b51eSKate Stone   void SetUserData(void *user_data) { m_user_data = user_data; }
2178ec990867SGreg Clayton 
2179b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
218044d93782SGreg Clayton 
2181b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
218244d93782SGreg Clayton 
2183b9c1b51eSKate Stone   void SetMightHaveChildren(bool b) { m_might_have_children = b; }
2184ec990867SGreg Clayton 
218544d93782SGreg Clayton protected:
218644d93782SGreg Clayton   TreeItem *m_parent;
218744d93782SGreg Clayton   TreeDelegate &m_delegate;
2188ec990867SGreg Clayton   void *m_user_data;
218944d93782SGreg Clayton   uint64_t m_identifier;
2190b9c1b51eSKate Stone   int m_row_idx; // Zero based visible row index, -1 if not visible or for the
2191b9c1b51eSKate Stone                  // root item
219244d93782SGreg Clayton   std::vector<TreeItem> m_children;
219344d93782SGreg Clayton   bool m_might_have_children;
219444d93782SGreg Clayton   bool m_is_expanded;
219544d93782SGreg Clayton };
219644d93782SGreg Clayton 
2197b9c1b51eSKate Stone class TreeWindowDelegate : public WindowDelegate {
219844d93782SGreg Clayton public:
2199b9c1b51eSKate Stone   TreeWindowDelegate(Debugger &debugger, const TreeDelegateSP &delegate_sp)
2200b9c1b51eSKate Stone       : m_debugger(debugger), m_delegate_sp(delegate_sp),
2201b9c1b51eSKate Stone         m_root(nullptr, *delegate_sp, true), m_selected_item(nullptr),
2202b9c1b51eSKate Stone         m_num_rows(0), m_selected_row_idx(0), m_first_visible_row(0),
2203b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
220444d93782SGreg Clayton 
2205b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
220644d93782SGreg Clayton 
2207b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
2208b9c1b51eSKate Stone     ExecutionContext exe_ctx(
2209b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
221044d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
221144d93782SGreg Clayton 
221244d93782SGreg Clayton     bool display_content = false;
2213b9c1b51eSKate Stone     if (process) {
221444d93782SGreg Clayton       StateType state = process->GetState();
2215b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
221644d93782SGreg Clayton         // We are stopped, so it is ok to
221744d93782SGreg Clayton         display_content = true;
2218b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
221944d93782SGreg Clayton         return true; // Don't do any updating when we are running
222044d93782SGreg Clayton       }
222144d93782SGreg Clayton     }
222244d93782SGreg Clayton 
222344d93782SGreg Clayton     m_min_x = 2;
222444d93782SGreg Clayton     m_min_y = 1;
222544d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
222644d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
222744d93782SGreg Clayton 
222844d93782SGreg Clayton     window.Erase();
222944d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
223044d93782SGreg Clayton 
2231b9c1b51eSKate Stone     if (display_content) {
223244d93782SGreg Clayton       const int num_visible_rows = NumVisibleRows();
223344d93782SGreg Clayton       m_num_rows = 0;
223444d93782SGreg Clayton       m_root.CalculateRowIndexes(m_num_rows);
223544d93782SGreg Clayton 
223605097246SAdrian Prantl       // If we unexpanded while having something selected our total number of
223705097246SAdrian Prantl       // rows is less than the num visible rows, then make sure we show all the
223805097246SAdrian Prantl       // rows by setting the first visible row accordingly.
223944d93782SGreg Clayton       if (m_first_visible_row > 0 && m_num_rows < num_visible_rows)
224044d93782SGreg Clayton         m_first_visible_row = 0;
224144d93782SGreg Clayton 
224244d93782SGreg Clayton       // Make sure the selected row is always visible
224344d93782SGreg Clayton       if (m_selected_row_idx < m_first_visible_row)
224444d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx;
224544d93782SGreg Clayton       else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
224644d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
224744d93782SGreg Clayton 
224844d93782SGreg Clayton       int row_idx = 0;
224944d93782SGreg Clayton       int num_rows_left = num_visible_rows;
2250b9c1b51eSKate Stone       m_root.Draw(window, m_first_visible_row, m_selected_row_idx, row_idx,
2251b9c1b51eSKate Stone                   num_rows_left);
225244d93782SGreg Clayton       // Get the selected row
225344d93782SGreg Clayton       m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
2254b9c1b51eSKate Stone     } else {
2255c5dac77aSEugene Zelenko       m_selected_item = nullptr;
225644d93782SGreg Clayton     }
225744d93782SGreg Clayton 
225844d93782SGreg Clayton     return true; // Drawing handled
225944d93782SGreg Clayton   }
226044d93782SGreg Clayton 
2261b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
226244d93782SGreg Clayton     return "Thread window keyboard shortcuts:";
226344d93782SGreg Clayton   }
226444d93782SGreg Clayton 
2265b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
226644d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
226744d93782SGreg Clayton         {KEY_UP, "Select previous item"},
226844d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
226944d93782SGreg Clayton         {KEY_RIGHT, "Expand the selected item"},
2270b9c1b51eSKate Stone         {KEY_LEFT,
2271b9c1b51eSKate Stone          "Unexpand the selected item or select parent if not expanded"},
227244d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
227344d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
227444d93782SGreg Clayton         {'h', "Show help dialog"},
227544d93782SGreg Clayton         {' ', "Toggle item expansion"},
227644d93782SGreg Clayton         {',', "Page up"},
227744d93782SGreg Clayton         {'.', "Page down"},
2278b9c1b51eSKate Stone         {'\0', nullptr}};
227944d93782SGreg Clayton     return g_source_view_key_help;
228044d93782SGreg Clayton   }
228144d93782SGreg Clayton 
2282b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2283b9c1b51eSKate Stone     switch (c) {
228444d93782SGreg Clayton     case ',':
228544d93782SGreg Clayton     case KEY_PPAGE:
228644d93782SGreg Clayton       // Page up key
2287b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
228844d93782SGreg Clayton         if (m_first_visible_row > m_max_y)
228944d93782SGreg Clayton           m_first_visible_row -= m_max_y;
229044d93782SGreg Clayton         else
229144d93782SGreg Clayton           m_first_visible_row = 0;
229244d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
229344d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
229444d93782SGreg Clayton         if (m_selected_item)
229544d93782SGreg Clayton           m_selected_item->ItemWasSelected();
229644d93782SGreg Clayton       }
229744d93782SGreg Clayton       return eKeyHandled;
229844d93782SGreg Clayton 
229944d93782SGreg Clayton     case '.':
230044d93782SGreg Clayton     case KEY_NPAGE:
230144d93782SGreg Clayton       // Page down key
2302b9c1b51eSKate Stone       if (m_num_rows > m_max_y) {
2303b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
230444d93782SGreg Clayton           m_first_visible_row += m_max_y;
230544d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
230644d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
230744d93782SGreg Clayton           if (m_selected_item)
230844d93782SGreg Clayton             m_selected_item->ItemWasSelected();
230944d93782SGreg Clayton         }
231044d93782SGreg Clayton       }
231144d93782SGreg Clayton       return eKeyHandled;
231244d93782SGreg Clayton 
231344d93782SGreg Clayton     case KEY_UP:
2314b9c1b51eSKate Stone       if (m_selected_row_idx > 0) {
231544d93782SGreg Clayton         --m_selected_row_idx;
231644d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
231744d93782SGreg Clayton         if (m_selected_item)
231844d93782SGreg Clayton           m_selected_item->ItemWasSelected();
231944d93782SGreg Clayton       }
232044d93782SGreg Clayton       return eKeyHandled;
2321315b6884SEugene Zelenko 
232244d93782SGreg Clayton     case KEY_DOWN:
2323b9c1b51eSKate Stone       if (m_selected_row_idx + 1 < m_num_rows) {
232444d93782SGreg Clayton         ++m_selected_row_idx;
232544d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
232644d93782SGreg Clayton         if (m_selected_item)
232744d93782SGreg Clayton           m_selected_item->ItemWasSelected();
232844d93782SGreg Clayton       }
232944d93782SGreg Clayton       return eKeyHandled;
233044d93782SGreg Clayton 
233144d93782SGreg Clayton     case KEY_RIGHT:
2332b9c1b51eSKate Stone       if (m_selected_item) {
233344d93782SGreg Clayton         if (!m_selected_item->IsExpanded())
233444d93782SGreg Clayton           m_selected_item->Expand();
233544d93782SGreg Clayton       }
233644d93782SGreg Clayton       return eKeyHandled;
233744d93782SGreg Clayton 
233844d93782SGreg Clayton     case KEY_LEFT:
2339b9c1b51eSKate Stone       if (m_selected_item) {
234044d93782SGreg Clayton         if (m_selected_item->IsExpanded())
234144d93782SGreg Clayton           m_selected_item->Unexpand();
2342b9c1b51eSKate Stone         else if (m_selected_item->GetParent()) {
234344d93782SGreg Clayton           m_selected_row_idx = m_selected_item->GetParent()->GetRowIndex();
234444d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
234544d93782SGreg Clayton           if (m_selected_item)
234644d93782SGreg Clayton             m_selected_item->ItemWasSelected();
234744d93782SGreg Clayton         }
234844d93782SGreg Clayton       }
234944d93782SGreg Clayton       return eKeyHandled;
235044d93782SGreg Clayton 
235144d93782SGreg Clayton     case ' ':
235244d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2353b9c1b51eSKate Stone       if (m_selected_item) {
235444d93782SGreg Clayton         if (m_selected_item->IsExpanded())
235544d93782SGreg Clayton           m_selected_item->Unexpand();
235644d93782SGreg Clayton         else
235744d93782SGreg Clayton           m_selected_item->Expand();
235844d93782SGreg Clayton       }
235944d93782SGreg Clayton       return eKeyHandled;
236044d93782SGreg Clayton 
236144d93782SGreg Clayton     case 'h':
236244d93782SGreg Clayton       window.CreateHelpSubwindow();
236344d93782SGreg Clayton       return eKeyHandled;
236444d93782SGreg Clayton 
236544d93782SGreg Clayton     default:
236644d93782SGreg Clayton       break;
236744d93782SGreg Clayton     }
236844d93782SGreg Clayton     return eKeyNotHandled;
236944d93782SGreg Clayton   }
237044d93782SGreg Clayton 
237144d93782SGreg Clayton protected:
237244d93782SGreg Clayton   Debugger &m_debugger;
237344d93782SGreg Clayton   TreeDelegateSP m_delegate_sp;
237444d93782SGreg Clayton   TreeItem m_root;
237544d93782SGreg Clayton   TreeItem *m_selected_item;
237644d93782SGreg Clayton   int m_num_rows;
237744d93782SGreg Clayton   int m_selected_row_idx;
237844d93782SGreg Clayton   int m_first_visible_row;
237944d93782SGreg Clayton   int m_min_x;
238044d93782SGreg Clayton   int m_min_y;
238144d93782SGreg Clayton   int m_max_x;
238244d93782SGreg Clayton   int m_max_y;
238344d93782SGreg Clayton };
238444d93782SGreg Clayton 
2385b9c1b51eSKate Stone class FrameTreeDelegate : public TreeDelegate {
238644d93782SGreg Clayton public:
2387b9c1b51eSKate Stone   FrameTreeDelegate() : TreeDelegate() {
2388b9c1b51eSKate Stone     FormatEntity::Parse(
2389b9c1b51eSKate Stone         "frame #${frame.index}: {${function.name}${function.pc-offset}}}",
2390554f68d3SGreg Clayton         m_format);
239144d93782SGreg Clayton   }
239244d93782SGreg Clayton 
2393315b6884SEugene Zelenko   ~FrameTreeDelegate() override = default;
239444d93782SGreg Clayton 
2395b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2396ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2397b9c1b51eSKate Stone     if (thread) {
239844d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2399ec990867SGreg Clayton       StackFrameSP frame_sp = thread->GetStackFrameAtIndex(frame_idx);
2400b9c1b51eSKate Stone       if (frame_sp) {
240144d93782SGreg Clayton         StreamString strm;
2402b9c1b51eSKate Stone         const SymbolContext &sc =
2403b9c1b51eSKate Stone             frame_sp->GetSymbolContext(eSymbolContextEverything);
240444d93782SGreg Clayton         ExecutionContext exe_ctx(frame_sp);
2405b9c1b51eSKate Stone         if (FormatEntity::Format(m_format, strm, &sc, &exe_ctx, nullptr,
2406b9c1b51eSKate Stone                                  nullptr, false, false)) {
240744d93782SGreg Clayton           int right_pad = 1;
2408c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
240944d93782SGreg Clayton         }
241044d93782SGreg Clayton       }
241144d93782SGreg Clayton     }
241244d93782SGreg Clayton   }
2413315b6884SEugene Zelenko 
2414b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
241544d93782SGreg Clayton     // No children for frames yet...
241644d93782SGreg Clayton   }
241744d93782SGreg Clayton 
2418b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2419ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2420b9c1b51eSKate Stone     if (thread) {
2421b9c1b51eSKate Stone       thread->GetProcess()->GetThreadList().SetSelectedThreadByID(
2422b9c1b51eSKate Stone           thread->GetID());
242344d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2424ec990867SGreg Clayton       thread->SetSelectedFrameByIndex(frame_idx);
242544d93782SGreg Clayton       return true;
242644d93782SGreg Clayton     }
242744d93782SGreg Clayton     return false;
242844d93782SGreg Clayton   }
2429315b6884SEugene Zelenko 
2430554f68d3SGreg Clayton protected:
2431554f68d3SGreg Clayton   FormatEntity::Entry m_format;
243244d93782SGreg Clayton };
243344d93782SGreg Clayton 
2434b9c1b51eSKate Stone class ThreadTreeDelegate : public TreeDelegate {
243544d93782SGreg Clayton public:
2436b9c1b51eSKate Stone   ThreadTreeDelegate(Debugger &debugger)
2437b9c1b51eSKate Stone       : TreeDelegate(), m_debugger(debugger), m_tid(LLDB_INVALID_THREAD_ID),
2438b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2439b9c1b51eSKate Stone     FormatEntity::Parse("thread #${thread.index}: tid = ${thread.id}{, stop "
2440b9c1b51eSKate Stone                         "reason = ${thread.stop-reason}}",
2441554f68d3SGreg Clayton                         m_format);
244244d93782SGreg Clayton   }
244344d93782SGreg Clayton 
2444315b6884SEugene Zelenko   ~ThreadTreeDelegate() override = default;
244544d93782SGreg Clayton 
2446b9c1b51eSKate Stone   ProcessSP GetProcess() {
2447b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2448b9c1b51eSKate Stone         .GetExecutionContext()
2449b9c1b51eSKate Stone         .GetProcessSP();
2450ec990867SGreg Clayton   }
2451ec990867SGreg Clayton 
2452b9c1b51eSKate Stone   ThreadSP GetThread(const TreeItem &item) {
2453ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2454ec990867SGreg Clayton     if (process_sp)
2455ec990867SGreg Clayton       return process_sp->GetThreadList().FindThreadByID(item.GetIdentifier());
2456ec990867SGreg Clayton     return ThreadSP();
2457ec990867SGreg Clayton   }
2458ec990867SGreg Clayton 
2459b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2460ec990867SGreg Clayton     ThreadSP thread_sp = GetThread(item);
2461b9c1b51eSKate Stone     if (thread_sp) {
246244d93782SGreg Clayton       StreamString strm;
246344d93782SGreg Clayton       ExecutionContext exe_ctx(thread_sp);
2464b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2465b9c1b51eSKate Stone                                nullptr, false, false)) {
246644d93782SGreg Clayton         int right_pad = 1;
2467c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
246844d93782SGreg Clayton       }
246944d93782SGreg Clayton     }
247044d93782SGreg Clayton   }
2471315b6884SEugene Zelenko 
2472b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2473ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2474b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
247544d93782SGreg Clayton       StateType state = process_sp->GetState();
2476b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2477ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2478b9c1b51eSKate Stone         if (thread_sp) {
2479b9c1b51eSKate Stone           if (m_stop_id == process_sp->GetStopID() &&
2480b9c1b51eSKate Stone               thread_sp->GetID() == m_tid)
248144d93782SGreg Clayton             return; // Children are already up to date
2482b9c1b51eSKate Stone           if (!m_frame_delegate_sp) {
248344d93782SGreg Clayton             // Always expand the thread item the first time we show it
2484796ac80bSJonas Devlieghere             m_frame_delegate_sp = std::make_shared<FrameTreeDelegate>();
248544d93782SGreg Clayton           }
248644d93782SGreg Clayton 
248744d93782SGreg Clayton           m_stop_id = process_sp->GetStopID();
248844d93782SGreg Clayton           m_tid = thread_sp->GetID();
248944d93782SGreg Clayton 
249044d93782SGreg Clayton           TreeItem t(&item, *m_frame_delegate_sp, false);
249144d93782SGreg Clayton           size_t num_frames = thread_sp->GetStackFrameCount();
249244d93782SGreg Clayton           item.Resize(num_frames, t);
2493b9c1b51eSKate Stone           for (size_t i = 0; i < num_frames; ++i) {
2494ec990867SGreg Clayton             item[i].SetUserData(thread_sp.get());
249544d93782SGreg Clayton             item[i].SetIdentifier(i);
249644d93782SGreg Clayton           }
249744d93782SGreg Clayton         }
249844d93782SGreg Clayton         return;
249944d93782SGreg Clayton       }
250044d93782SGreg Clayton     }
250144d93782SGreg Clayton     item.ClearChildren();
250244d93782SGreg Clayton   }
250344d93782SGreg Clayton 
2504b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2505ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2506b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2507ec990867SGreg Clayton       StateType state = process_sp->GetState();
2508b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2509ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2510b9c1b51eSKate Stone         if (thread_sp) {
251144d93782SGreg Clayton           ThreadList &thread_list = thread_sp->GetProcess()->GetThreadList();
2512bb19a13cSSaleem Abdulrasool           std::lock_guard<std::recursive_mutex> guard(thread_list.GetMutex());
251344d93782SGreg Clayton           ThreadSP selected_thread_sp = thread_list.GetSelectedThread();
2514b9c1b51eSKate Stone           if (selected_thread_sp->GetID() != thread_sp->GetID()) {
251544d93782SGreg Clayton             thread_list.SetSelectedThreadByID(thread_sp->GetID());
251644d93782SGreg Clayton             return true;
251744d93782SGreg Clayton           }
251844d93782SGreg Clayton         }
2519ec990867SGreg Clayton       }
2520ec990867SGreg Clayton     }
252144d93782SGreg Clayton     return false;
252244d93782SGreg Clayton   }
252344d93782SGreg Clayton 
252444d93782SGreg Clayton protected:
252544d93782SGreg Clayton   Debugger &m_debugger;
252644d93782SGreg Clayton   std::shared_ptr<FrameTreeDelegate> m_frame_delegate_sp;
252744d93782SGreg Clayton   lldb::user_id_t m_tid;
252844d93782SGreg Clayton   uint32_t m_stop_id;
2529554f68d3SGreg Clayton   FormatEntity::Entry m_format;
253044d93782SGreg Clayton };
253144d93782SGreg Clayton 
2532b9c1b51eSKate Stone class ThreadsTreeDelegate : public TreeDelegate {
2533ec990867SGreg Clayton public:
2534b9c1b51eSKate Stone   ThreadsTreeDelegate(Debugger &debugger)
2535b9c1b51eSKate Stone       : TreeDelegate(), m_thread_delegate_sp(), m_debugger(debugger),
2536b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2537554f68d3SGreg Clayton     FormatEntity::Parse("process ${process.id}{, name = ${process.name}}",
2538554f68d3SGreg Clayton                         m_format);
2539ec990867SGreg Clayton   }
2540ec990867SGreg Clayton 
2541315b6884SEugene Zelenko   ~ThreadsTreeDelegate() override = default;
2542ec990867SGreg Clayton 
2543b9c1b51eSKate Stone   ProcessSP GetProcess() {
2544b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2545b9c1b51eSKate Stone         .GetExecutionContext()
2546b9c1b51eSKate Stone         .GetProcessSP();
2547ec990867SGreg Clayton   }
2548ec990867SGreg Clayton 
2549b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2550ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2551b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2552ec990867SGreg Clayton       StreamString strm;
2553ec990867SGreg Clayton       ExecutionContext exe_ctx(process_sp);
2554b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2555b9c1b51eSKate Stone                                nullptr, false, false)) {
2556ec990867SGreg Clayton         int right_pad = 1;
2557c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
2558ec990867SGreg Clayton       }
2559ec990867SGreg Clayton     }
2560ec990867SGreg Clayton   }
2561ec990867SGreg Clayton 
2562b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2563ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2564b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2565ec990867SGreg Clayton       StateType state = process_sp->GetState();
2566b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2567ec990867SGreg Clayton         const uint32_t stop_id = process_sp->GetStopID();
2568ec990867SGreg Clayton         if (m_stop_id == stop_id)
2569ec990867SGreg Clayton           return; // Children are already up to date
2570ec990867SGreg Clayton 
2571ec990867SGreg Clayton         m_stop_id = stop_id;
2572ec990867SGreg Clayton 
2573b9c1b51eSKate Stone         if (!m_thread_delegate_sp) {
2574ec990867SGreg Clayton           // Always expand the thread item the first time we show it
2575ec990867SGreg Clayton           // item.Expand();
2576796ac80bSJonas Devlieghere           m_thread_delegate_sp =
2577796ac80bSJonas Devlieghere               std::make_shared<ThreadTreeDelegate>(m_debugger);
2578ec990867SGreg Clayton         }
2579ec990867SGreg Clayton 
2580ec990867SGreg Clayton         TreeItem t(&item, *m_thread_delegate_sp, false);
2581ec990867SGreg Clayton         ThreadList &threads = process_sp->GetThreadList();
2582bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
2583ec990867SGreg Clayton         size_t num_threads = threads.GetSize();
2584ec990867SGreg Clayton         item.Resize(num_threads, t);
2585b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
2586ec990867SGreg Clayton           item[i].SetIdentifier(threads.GetThreadAtIndex(i)->GetID());
2587ec990867SGreg Clayton           item[i].SetMightHaveChildren(true);
2588ec990867SGreg Clayton         }
2589ec990867SGreg Clayton         return;
2590ec990867SGreg Clayton       }
2591ec990867SGreg Clayton     }
2592ec990867SGreg Clayton     item.ClearChildren();
2593ec990867SGreg Clayton   }
2594ec990867SGreg Clayton 
2595b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override { return false; }
2596ec990867SGreg Clayton 
2597ec990867SGreg Clayton protected:
2598ec990867SGreg Clayton   std::shared_ptr<ThreadTreeDelegate> m_thread_delegate_sp;
2599ec990867SGreg Clayton   Debugger &m_debugger;
2600ec990867SGreg Clayton   uint32_t m_stop_id;
2601554f68d3SGreg Clayton   FormatEntity::Entry m_format;
2602ec990867SGreg Clayton };
2603ec990867SGreg Clayton 
2604b9c1b51eSKate Stone class ValueObjectListDelegate : public WindowDelegate {
260544d93782SGreg Clayton public:
2606b9c1b51eSKate Stone   ValueObjectListDelegate()
2607171dd2e6SJonas Devlieghere       : m_rows(), m_selected_row(nullptr), m_selected_row_idx(0),
2608171dd2e6SJonas Devlieghere         m_first_visible_row(0), m_num_rows(0), m_max_x(0), m_max_y(0) {}
260944d93782SGreg Clayton 
2610b9c1b51eSKate Stone   ValueObjectListDelegate(ValueObjectList &valobj_list)
2611171dd2e6SJonas Devlieghere       : m_rows(), m_selected_row(nullptr), m_selected_row_idx(0),
2612171dd2e6SJonas Devlieghere         m_first_visible_row(0), m_num_rows(0), m_max_x(0), m_max_y(0) {
261344d93782SGreg Clayton     SetValues(valobj_list);
261444d93782SGreg Clayton   }
261544d93782SGreg Clayton 
2616315b6884SEugene Zelenko   ~ValueObjectListDelegate() override = default;
261744d93782SGreg Clayton 
2618b9c1b51eSKate Stone   void SetValues(ValueObjectList &valobj_list) {
2619c5dac77aSEugene Zelenko     m_selected_row = nullptr;
262044d93782SGreg Clayton     m_selected_row_idx = 0;
262144d93782SGreg Clayton     m_first_visible_row = 0;
262244d93782SGreg Clayton     m_num_rows = 0;
262344d93782SGreg Clayton     m_rows.clear();
26248369b28dSGreg Clayton     for (auto &valobj_sp : valobj_list.GetObjects())
26258369b28dSGreg Clayton       m_rows.push_back(Row(valobj_sp, nullptr));
262644d93782SGreg Clayton   }
262744d93782SGreg Clayton 
2628b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
262944d93782SGreg Clayton     m_num_rows = 0;
263044d93782SGreg Clayton     m_min_x = 2;
263144d93782SGreg Clayton     m_min_y = 1;
263244d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
263344d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
263444d93782SGreg Clayton 
263544d93782SGreg Clayton     window.Erase();
263644d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
263744d93782SGreg Clayton 
263844d93782SGreg Clayton     const int num_visible_rows = NumVisibleRows();
263944d93782SGreg Clayton     const int num_rows = CalculateTotalNumberRows(m_rows);
264044d93782SGreg Clayton 
264105097246SAdrian Prantl     // If we unexpanded while having something selected our total number of
264205097246SAdrian Prantl     // rows is less than the num visible rows, then make sure we show all the
264305097246SAdrian Prantl     // rows by setting the first visible row accordingly.
264444d93782SGreg Clayton     if (m_first_visible_row > 0 && num_rows < num_visible_rows)
264544d93782SGreg Clayton       m_first_visible_row = 0;
264644d93782SGreg Clayton 
264744d93782SGreg Clayton     // Make sure the selected row is always visible
264844d93782SGreg Clayton     if (m_selected_row_idx < m_first_visible_row)
264944d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx;
265044d93782SGreg Clayton     else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
265144d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
265244d93782SGreg Clayton 
265344d93782SGreg Clayton     DisplayRows(window, m_rows, g_options);
265444d93782SGreg Clayton 
265544d93782SGreg Clayton     // Get the selected row
265644d93782SGreg Clayton     m_selected_row = GetRowForRowIndex(m_selected_row_idx);
265705097246SAdrian Prantl     // Keep the cursor on the selected row so the highlight and the cursor are
265805097246SAdrian Prantl     // always on the same line
265944d93782SGreg Clayton     if (m_selected_row)
2660b9c1b51eSKate Stone       window.MoveCursor(m_selected_row->x, m_selected_row->y);
266144d93782SGreg Clayton 
266244d93782SGreg Clayton     return true; // Drawing handled
266344d93782SGreg Clayton   }
266444d93782SGreg Clayton 
2665b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
266644d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
266744d93782SGreg Clayton         {KEY_UP, "Select previous item"},
266844d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
266944d93782SGreg Clayton         {KEY_RIGHT, "Expand selected item"},
267044d93782SGreg Clayton         {KEY_LEFT, "Unexpand selected item or select parent if not expanded"},
267144d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
267244d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
267344d93782SGreg Clayton         {'A', "Format as annotated address"},
267444d93782SGreg Clayton         {'b', "Format as binary"},
267544d93782SGreg Clayton         {'B', "Format as hex bytes with ASCII"},
267644d93782SGreg Clayton         {'c', "Format as character"},
267744d93782SGreg Clayton         {'d', "Format as a signed integer"},
267844d93782SGreg Clayton         {'D', "Format selected value using the default format for the type"},
267944d93782SGreg Clayton         {'f', "Format as float"},
268044d93782SGreg Clayton         {'h', "Show help dialog"},
268144d93782SGreg Clayton         {'i', "Format as instructions"},
268244d93782SGreg Clayton         {'o', "Format as octal"},
268344d93782SGreg Clayton         {'p', "Format as pointer"},
268444d93782SGreg Clayton         {'s', "Format as C string"},
268544d93782SGreg Clayton         {'t', "Toggle showing/hiding type names"},
268644d93782SGreg Clayton         {'u', "Format as an unsigned integer"},
268744d93782SGreg Clayton         {'x', "Format as hex"},
268844d93782SGreg Clayton         {'X', "Format as uppercase hex"},
268944d93782SGreg Clayton         {' ', "Toggle item expansion"},
269044d93782SGreg Clayton         {',', "Page up"},
269144d93782SGreg Clayton         {'.', "Page down"},
2692b9c1b51eSKate Stone         {'\0', nullptr}};
269344d93782SGreg Clayton     return g_source_view_key_help;
269444d93782SGreg Clayton   }
269544d93782SGreg Clayton 
2696b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2697b9c1b51eSKate Stone     switch (c) {
269844d93782SGreg Clayton     case 'x':
269944d93782SGreg Clayton     case 'X':
270044d93782SGreg Clayton     case 'o':
270144d93782SGreg Clayton     case 's':
270244d93782SGreg Clayton     case 'u':
270344d93782SGreg Clayton     case 'd':
270444d93782SGreg Clayton     case 'D':
270544d93782SGreg Clayton     case 'i':
270644d93782SGreg Clayton     case 'A':
270744d93782SGreg Clayton     case 'p':
270844d93782SGreg Clayton     case 'c':
270944d93782SGreg Clayton     case 'b':
271044d93782SGreg Clayton     case 'B':
271144d93782SGreg Clayton     case 'f':
271244d93782SGreg Clayton       // Change the format for the currently selected item
27138369b28dSGreg Clayton       if (m_selected_row) {
27148369b28dSGreg Clayton         auto valobj_sp = m_selected_row->value.GetSP();
27158369b28dSGreg Clayton         if (valobj_sp)
27168369b28dSGreg Clayton           valobj_sp->SetFormat(FormatForChar(c));
27178369b28dSGreg Clayton       }
271844d93782SGreg Clayton       return eKeyHandled;
271944d93782SGreg Clayton 
272044d93782SGreg Clayton     case 't':
272144d93782SGreg Clayton       // Toggle showing type names
272244d93782SGreg Clayton       g_options.show_types = !g_options.show_types;
272344d93782SGreg Clayton       return eKeyHandled;
272444d93782SGreg Clayton 
272544d93782SGreg Clayton     case ',':
272644d93782SGreg Clayton     case KEY_PPAGE:
272744d93782SGreg Clayton       // Page up key
2728b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
27293985c8c6SSaleem Abdulrasool         if (static_cast<int>(m_first_visible_row) > m_max_y)
273044d93782SGreg Clayton           m_first_visible_row -= m_max_y;
273144d93782SGreg Clayton         else
273244d93782SGreg Clayton           m_first_visible_row = 0;
273344d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
273444d93782SGreg Clayton       }
273544d93782SGreg Clayton       return eKeyHandled;
273644d93782SGreg Clayton 
273744d93782SGreg Clayton     case '.':
273844d93782SGreg Clayton     case KEY_NPAGE:
273944d93782SGreg Clayton       // Page down key
2740b9c1b51eSKate Stone       if (m_num_rows > static_cast<size_t>(m_max_y)) {
2741b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
274244d93782SGreg Clayton           m_first_visible_row += m_max_y;
274344d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
274444d93782SGreg Clayton         }
274544d93782SGreg Clayton       }
274644d93782SGreg Clayton       return eKeyHandled;
274744d93782SGreg Clayton 
274844d93782SGreg Clayton     case KEY_UP:
274944d93782SGreg Clayton       if (m_selected_row_idx > 0)
275044d93782SGreg Clayton         --m_selected_row_idx;
275144d93782SGreg Clayton       return eKeyHandled;
2752315b6884SEugene Zelenko 
275344d93782SGreg Clayton     case KEY_DOWN:
275444d93782SGreg Clayton       if (m_selected_row_idx + 1 < m_num_rows)
275544d93782SGreg Clayton         ++m_selected_row_idx;
275644d93782SGreg Clayton       return eKeyHandled;
275744d93782SGreg Clayton 
275844d93782SGreg Clayton     case KEY_RIGHT:
2759b9c1b51eSKate Stone       if (m_selected_row) {
276044d93782SGreg Clayton         if (!m_selected_row->expanded)
276144d93782SGreg Clayton           m_selected_row->Expand();
276244d93782SGreg Clayton       }
276344d93782SGreg Clayton       return eKeyHandled;
276444d93782SGreg Clayton 
276544d93782SGreg Clayton     case KEY_LEFT:
2766b9c1b51eSKate Stone       if (m_selected_row) {
276744d93782SGreg Clayton         if (m_selected_row->expanded)
276844d93782SGreg Clayton           m_selected_row->Unexpand();
276944d93782SGreg Clayton         else if (m_selected_row->parent)
277044d93782SGreg Clayton           m_selected_row_idx = m_selected_row->parent->row_idx;
277144d93782SGreg Clayton       }
277244d93782SGreg Clayton       return eKeyHandled;
277344d93782SGreg Clayton 
277444d93782SGreg Clayton     case ' ':
277544d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2776b9c1b51eSKate Stone       if (m_selected_row) {
277744d93782SGreg Clayton         if (m_selected_row->expanded)
277844d93782SGreg Clayton           m_selected_row->Unexpand();
277944d93782SGreg Clayton         else
278044d93782SGreg Clayton           m_selected_row->Expand();
278144d93782SGreg Clayton       }
278244d93782SGreg Clayton       return eKeyHandled;
278344d93782SGreg Clayton 
278444d93782SGreg Clayton     case 'h':
278544d93782SGreg Clayton       window.CreateHelpSubwindow();
278644d93782SGreg Clayton       return eKeyHandled;
278744d93782SGreg Clayton 
278844d93782SGreg Clayton     default:
278944d93782SGreg Clayton       break;
279044d93782SGreg Clayton     }
279144d93782SGreg Clayton     return eKeyNotHandled;
279244d93782SGreg Clayton   }
279344d93782SGreg Clayton 
279444d93782SGreg Clayton protected:
279544d93782SGreg Clayton   std::vector<Row> m_rows;
279644d93782SGreg Clayton   Row *m_selected_row;
279744d93782SGreg Clayton   uint32_t m_selected_row_idx;
279844d93782SGreg Clayton   uint32_t m_first_visible_row;
279944d93782SGreg Clayton   uint32_t m_num_rows;
280044d93782SGreg Clayton   int m_min_x;
280144d93782SGreg Clayton   int m_min_y;
280244d93782SGreg Clayton   int m_max_x;
280344d93782SGreg Clayton   int m_max_y;
280444d93782SGreg Clayton 
2805b9c1b51eSKate Stone   static Format FormatForChar(int c) {
2806b9c1b51eSKate Stone     switch (c) {
2807b9c1b51eSKate Stone     case 'x':
2808b9c1b51eSKate Stone       return eFormatHex;
2809b9c1b51eSKate Stone     case 'X':
2810b9c1b51eSKate Stone       return eFormatHexUppercase;
2811b9c1b51eSKate Stone     case 'o':
2812b9c1b51eSKate Stone       return eFormatOctal;
2813b9c1b51eSKate Stone     case 's':
2814b9c1b51eSKate Stone       return eFormatCString;
2815b9c1b51eSKate Stone     case 'u':
2816b9c1b51eSKate Stone       return eFormatUnsigned;
2817b9c1b51eSKate Stone     case 'd':
2818b9c1b51eSKate Stone       return eFormatDecimal;
2819b9c1b51eSKate Stone     case 'D':
2820b9c1b51eSKate Stone       return eFormatDefault;
2821b9c1b51eSKate Stone     case 'i':
2822b9c1b51eSKate Stone       return eFormatInstruction;
2823b9c1b51eSKate Stone     case 'A':
2824b9c1b51eSKate Stone       return eFormatAddressInfo;
2825b9c1b51eSKate Stone     case 'p':
2826b9c1b51eSKate Stone       return eFormatPointer;
2827b9c1b51eSKate Stone     case 'c':
2828b9c1b51eSKate Stone       return eFormatChar;
2829b9c1b51eSKate Stone     case 'b':
2830b9c1b51eSKate Stone       return eFormatBinary;
2831b9c1b51eSKate Stone     case 'B':
2832b9c1b51eSKate Stone       return eFormatBytesWithASCII;
2833b9c1b51eSKate Stone     case 'f':
2834b9c1b51eSKate Stone       return eFormatFloat;
283544d93782SGreg Clayton     }
283644d93782SGreg Clayton     return eFormatDefault;
283744d93782SGreg Clayton   }
283844d93782SGreg Clayton 
2839b9c1b51eSKate Stone   bool DisplayRowObject(Window &window, Row &row, DisplayOptions &options,
2840b9c1b51eSKate Stone                         bool highlight, bool last_child) {
28418369b28dSGreg Clayton     ValueObject *valobj = row.value.GetSP().get();
284244d93782SGreg Clayton 
2843c5dac77aSEugene Zelenko     if (valobj == nullptr)
284444d93782SGreg Clayton       return false;
284544d93782SGreg Clayton 
2846b9c1b51eSKate Stone     const char *type_name =
2847b9c1b51eSKate Stone         options.show_types ? valobj->GetTypeName().GetCString() : nullptr;
284844d93782SGreg Clayton     const char *name = valobj->GetName().GetCString();
284944d93782SGreg Clayton     const char *value = valobj->GetValueAsCString();
285044d93782SGreg Clayton     const char *summary = valobj->GetSummaryAsCString();
285144d93782SGreg Clayton 
285244d93782SGreg Clayton     window.MoveCursor(row.x, row.y);
285344d93782SGreg Clayton 
285444d93782SGreg Clayton     row.DrawTree(window);
285544d93782SGreg Clayton 
285644d93782SGreg Clayton     if (highlight)
285744d93782SGreg Clayton       window.AttributeOn(A_REVERSE);
285844d93782SGreg Clayton 
285944d93782SGreg Clayton     if (type_name && type_name[0])
286044d93782SGreg Clayton       window.Printf("(%s) ", type_name);
286144d93782SGreg Clayton 
286244d93782SGreg Clayton     if (name && name[0])
286344d93782SGreg Clayton       window.PutCString(name);
286444d93782SGreg Clayton 
286544d93782SGreg Clayton     attr_t changd_attr = 0;
286644d93782SGreg Clayton     if (valobj->GetValueDidChange())
286744d93782SGreg Clayton       changd_attr = COLOR_PAIR(5) | A_BOLD;
286844d93782SGreg Clayton 
2869b9c1b51eSKate Stone     if (value && value[0]) {
287044d93782SGreg Clayton       window.PutCString(" = ");
287144d93782SGreg Clayton       if (changd_attr)
287244d93782SGreg Clayton         window.AttributeOn(changd_attr);
287344d93782SGreg Clayton       window.PutCString(value);
287444d93782SGreg Clayton       if (changd_attr)
287544d93782SGreg Clayton         window.AttributeOff(changd_attr);
287644d93782SGreg Clayton     }
287744d93782SGreg Clayton 
2878b9c1b51eSKate Stone     if (summary && summary[0]) {
287944d93782SGreg Clayton       window.PutChar(' ');
288044d93782SGreg Clayton       if (changd_attr)
288144d93782SGreg Clayton         window.AttributeOn(changd_attr);
288244d93782SGreg Clayton       window.PutCString(summary);
288344d93782SGreg Clayton       if (changd_attr)
288444d93782SGreg Clayton         window.AttributeOff(changd_attr);
288544d93782SGreg Clayton     }
288644d93782SGreg Clayton 
288744d93782SGreg Clayton     if (highlight)
288844d93782SGreg Clayton       window.AttributeOff(A_REVERSE);
288944d93782SGreg Clayton 
289044d93782SGreg Clayton     return true;
289144d93782SGreg Clayton   }
2892315b6884SEugene Zelenko 
2893b9c1b51eSKate Stone   void DisplayRows(Window &window, std::vector<Row> &rows,
2894b9c1b51eSKate Stone                    DisplayOptions &options) {
289544d93782SGreg Clayton     // >   0x25B7
289644d93782SGreg Clayton     // \/  0x25BD
289744d93782SGreg Clayton 
289844d93782SGreg Clayton     bool window_is_active = window.IsActive();
2899b9c1b51eSKate Stone     for (auto &row : rows) {
290044d93782SGreg Clayton       const bool last_child = row.parent && &rows[rows.size() - 1] == &row;
290144d93782SGreg Clayton       // Save the row index in each Row structure
290244d93782SGreg Clayton       row.row_idx = m_num_rows;
290344d93782SGreg Clayton       if ((m_num_rows >= m_first_visible_row) &&
2904b9c1b51eSKate Stone           ((m_num_rows - m_first_visible_row) <
2905b9c1b51eSKate Stone            static_cast<size_t>(NumVisibleRows()))) {
290644d93782SGreg Clayton         row.x = m_min_x;
290744d93782SGreg Clayton         row.y = m_num_rows - m_first_visible_row + 1;
2908b9c1b51eSKate Stone         if (DisplayRowObject(window, row, options,
2909b9c1b51eSKate Stone                              window_is_active &&
2910b9c1b51eSKate Stone                                  m_num_rows == m_selected_row_idx,
2911b9c1b51eSKate Stone                              last_child)) {
291244d93782SGreg Clayton           ++m_num_rows;
2913b9c1b51eSKate Stone         } else {
291444d93782SGreg Clayton           row.x = 0;
291544d93782SGreg Clayton           row.y = 0;
291644d93782SGreg Clayton         }
2917b9c1b51eSKate Stone       } else {
291844d93782SGreg Clayton         row.x = 0;
291944d93782SGreg Clayton         row.y = 0;
292044d93782SGreg Clayton         ++m_num_rows;
292144d93782SGreg Clayton       }
292244d93782SGreg Clayton 
29238369b28dSGreg Clayton       auto &children = row.GetChildren();
29248369b28dSGreg Clayton       if (row.expanded && !children.empty()) {
29258369b28dSGreg Clayton         DisplayRows(window, children, options);
292644d93782SGreg Clayton       }
292744d93782SGreg Clayton     }
292844d93782SGreg Clayton   }
292944d93782SGreg Clayton 
29308369b28dSGreg Clayton   int CalculateTotalNumberRows(std::vector<Row> &rows) {
293144d93782SGreg Clayton     int row_count = 0;
29328369b28dSGreg Clayton     for (auto &row : rows) {
293344d93782SGreg Clayton       ++row_count;
293444d93782SGreg Clayton       if (row.expanded)
29358369b28dSGreg Clayton         row_count += CalculateTotalNumberRows(row.GetChildren());
293644d93782SGreg Clayton     }
293744d93782SGreg Clayton     return row_count;
293844d93782SGreg Clayton   }
2939315b6884SEugene Zelenko 
2940b9c1b51eSKate Stone   static Row *GetRowForRowIndexImpl(std::vector<Row> &rows, size_t &row_index) {
2941b9c1b51eSKate Stone     for (auto &row : rows) {
294244d93782SGreg Clayton       if (row_index == 0)
294344d93782SGreg Clayton         return &row;
2944b9c1b51eSKate Stone       else {
294544d93782SGreg Clayton         --row_index;
29468369b28dSGreg Clayton         auto &children = row.GetChildren();
29478369b28dSGreg Clayton         if (row.expanded && !children.empty()) {
29488369b28dSGreg Clayton           Row *result = GetRowForRowIndexImpl(children, row_index);
294944d93782SGreg Clayton           if (result)
295044d93782SGreg Clayton             return result;
295144d93782SGreg Clayton         }
295244d93782SGreg Clayton       }
295344d93782SGreg Clayton     }
2954c5dac77aSEugene Zelenko     return nullptr;
295544d93782SGreg Clayton   }
295644d93782SGreg Clayton 
2957b9c1b51eSKate Stone   Row *GetRowForRowIndex(size_t row_index) {
295844d93782SGreg Clayton     return GetRowForRowIndexImpl(m_rows, row_index);
295944d93782SGreg Clayton   }
296044d93782SGreg Clayton 
2961b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
296244d93782SGreg Clayton 
296344d93782SGreg Clayton   static DisplayOptions g_options;
296444d93782SGreg Clayton };
296544d93782SGreg Clayton 
2966b9c1b51eSKate Stone class FrameVariablesWindowDelegate : public ValueObjectListDelegate {
296744d93782SGreg Clayton public:
2968b9c1b51eSKate Stone   FrameVariablesWindowDelegate(Debugger &debugger)
2969b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger),
2970b9c1b51eSKate Stone         m_frame_block(nullptr) {}
297144d93782SGreg Clayton 
2972315b6884SEugene Zelenko   ~FrameVariablesWindowDelegate() override = default;
297344d93782SGreg Clayton 
2974b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
297544d93782SGreg Clayton     return "Frame variable window keyboard shortcuts:";
297644d93782SGreg Clayton   }
297744d93782SGreg Clayton 
2978b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
2979b9c1b51eSKate Stone     ExecutionContext exe_ctx(
2980b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
298144d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
2982c5dac77aSEugene Zelenko     Block *frame_block = nullptr;
2983c5dac77aSEugene Zelenko     StackFrame *frame = nullptr;
298444d93782SGreg Clayton 
2985b9c1b51eSKate Stone     if (process) {
298644d93782SGreg Clayton       StateType state = process->GetState();
2987b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
298844d93782SGreg Clayton         frame = exe_ctx.GetFramePtr();
298944d93782SGreg Clayton         if (frame)
299044d93782SGreg Clayton           frame_block = frame->GetFrameBlock();
2991b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
299244d93782SGreg Clayton         return true; // Don't do any updating when we are running
299344d93782SGreg Clayton       }
299444d93782SGreg Clayton     }
299544d93782SGreg Clayton 
299644d93782SGreg Clayton     ValueObjectList local_values;
2997b9c1b51eSKate Stone     if (frame_block) {
299844d93782SGreg Clayton       // Only update the variables if they have changed
2999b9c1b51eSKate Stone       if (m_frame_block != frame_block) {
300044d93782SGreg Clayton         m_frame_block = frame_block;
300144d93782SGreg Clayton 
300244d93782SGreg Clayton         VariableList *locals = frame->GetVariableList(true);
3003b9c1b51eSKate Stone         if (locals) {
300444d93782SGreg Clayton           const DynamicValueType use_dynamic = eDynamicDontRunTarget;
300544d93782SGreg Clayton           const size_t num_locals = locals->GetSize();
3006b9c1b51eSKate Stone           for (size_t i = 0; i < num_locals; ++i) {
3007b9c1b51eSKate Stone             ValueObjectSP value_sp = frame->GetValueObjectForFrameVariable(
3008b9c1b51eSKate Stone                 locals->GetVariableAtIndex(i), use_dynamic);
3009b9c1b51eSKate Stone             if (value_sp) {
3010eb72dc7dSGreg Clayton               ValueObjectSP synthetic_value_sp = value_sp->GetSyntheticValue();
3011eb72dc7dSGreg Clayton               if (synthetic_value_sp)
3012eb72dc7dSGreg Clayton                 local_values.Append(synthetic_value_sp);
3013eb72dc7dSGreg Clayton               else
3014eb72dc7dSGreg Clayton                 local_values.Append(value_sp);
3015eb72dc7dSGreg Clayton             }
3016eb72dc7dSGreg Clayton           }
301744d93782SGreg Clayton           // Update the values
301844d93782SGreg Clayton           SetValues(local_values);
301944d93782SGreg Clayton         }
302044d93782SGreg Clayton       }
3021b9c1b51eSKate Stone     } else {
3022c5dac77aSEugene Zelenko       m_frame_block = nullptr;
302344d93782SGreg Clayton       // Update the values with an empty list if there is no frame
302444d93782SGreg Clayton       SetValues(local_values);
302544d93782SGreg Clayton     }
302644d93782SGreg Clayton 
302744d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
302844d93782SGreg Clayton   }
302944d93782SGreg Clayton 
303044d93782SGreg Clayton protected:
303144d93782SGreg Clayton   Debugger &m_debugger;
303244d93782SGreg Clayton   Block *m_frame_block;
303344d93782SGreg Clayton };
303444d93782SGreg Clayton 
3035b9c1b51eSKate Stone class RegistersWindowDelegate : public ValueObjectListDelegate {
303644d93782SGreg Clayton public:
3037b9c1b51eSKate Stone   RegistersWindowDelegate(Debugger &debugger)
3038b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger) {}
303944d93782SGreg Clayton 
3040315b6884SEugene Zelenko   ~RegistersWindowDelegate() override = default;
304144d93782SGreg Clayton 
3042b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
304344d93782SGreg Clayton     return "Register window keyboard shortcuts:";
304444d93782SGreg Clayton   }
304544d93782SGreg Clayton 
3046b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3047b9c1b51eSKate Stone     ExecutionContext exe_ctx(
3048b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
304944d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
305044d93782SGreg Clayton 
305144d93782SGreg Clayton     ValueObjectList value_list;
3052b9c1b51eSKate Stone     if (frame) {
3053b9c1b51eSKate Stone       if (frame->GetStackID() != m_stack_id) {
305444d93782SGreg Clayton         m_stack_id = frame->GetStackID();
305544d93782SGreg Clayton         RegisterContextSP reg_ctx(frame->GetRegisterContext());
3056b9c1b51eSKate Stone         if (reg_ctx) {
305744d93782SGreg Clayton           const uint32_t num_sets = reg_ctx->GetRegisterSetCount();
3058b9c1b51eSKate Stone           for (uint32_t set_idx = 0; set_idx < num_sets; ++set_idx) {
3059b9c1b51eSKate Stone             value_list.Append(
3060b9c1b51eSKate Stone                 ValueObjectRegisterSet::Create(frame, reg_ctx, set_idx));
306144d93782SGreg Clayton           }
306244d93782SGreg Clayton         }
306344d93782SGreg Clayton         SetValues(value_list);
306444d93782SGreg Clayton       }
3065b9c1b51eSKate Stone     } else {
306644d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
306744d93782SGreg Clayton       if (process && process->IsAlive())
306844d93782SGreg Clayton         return true; // Don't do any updating if we are running
3069b9c1b51eSKate Stone       else {
307005097246SAdrian Prantl         // Update the values with an empty list if there is no process or the
307105097246SAdrian Prantl         // process isn't alive anymore
307244d93782SGreg Clayton         SetValues(value_list);
307344d93782SGreg Clayton       }
307444d93782SGreg Clayton     }
307544d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
307644d93782SGreg Clayton   }
307744d93782SGreg Clayton 
307844d93782SGreg Clayton protected:
307944d93782SGreg Clayton   Debugger &m_debugger;
308044d93782SGreg Clayton   StackID m_stack_id;
308144d93782SGreg Clayton };
308244d93782SGreg Clayton 
3083b9c1b51eSKate Stone static const char *CursesKeyToCString(int ch) {
308444d93782SGreg Clayton   static char g_desc[32];
3085b9c1b51eSKate Stone   if (ch >= KEY_F0 && ch < KEY_F0 + 64) {
308644d93782SGreg Clayton     snprintf(g_desc, sizeof(g_desc), "F%u", ch - KEY_F0);
308744d93782SGreg Clayton     return g_desc;
308844d93782SGreg Clayton   }
3089b9c1b51eSKate Stone   switch (ch) {
3090b9c1b51eSKate Stone   case KEY_DOWN:
3091b9c1b51eSKate Stone     return "down";
3092b9c1b51eSKate Stone   case KEY_UP:
3093b9c1b51eSKate Stone     return "up";
3094b9c1b51eSKate Stone   case KEY_LEFT:
3095b9c1b51eSKate Stone     return "left";
3096b9c1b51eSKate Stone   case KEY_RIGHT:
3097b9c1b51eSKate Stone     return "right";
3098b9c1b51eSKate Stone   case KEY_HOME:
3099b9c1b51eSKate Stone     return "home";
3100b9c1b51eSKate Stone   case KEY_BACKSPACE:
3101b9c1b51eSKate Stone     return "backspace";
3102b9c1b51eSKate Stone   case KEY_DL:
3103b9c1b51eSKate Stone     return "delete-line";
3104b9c1b51eSKate Stone   case KEY_IL:
3105b9c1b51eSKate Stone     return "insert-line";
3106b9c1b51eSKate Stone   case KEY_DC:
3107b9c1b51eSKate Stone     return "delete-char";
3108b9c1b51eSKate Stone   case KEY_IC:
3109b9c1b51eSKate Stone     return "insert-char";
3110b9c1b51eSKate Stone   case KEY_CLEAR:
3111b9c1b51eSKate Stone     return "clear";
3112b9c1b51eSKate Stone   case KEY_EOS:
3113b9c1b51eSKate Stone     return "clear-to-eos";
3114b9c1b51eSKate Stone   case KEY_EOL:
3115b9c1b51eSKate Stone     return "clear-to-eol";
3116b9c1b51eSKate Stone   case KEY_SF:
3117b9c1b51eSKate Stone     return "scroll-forward";
3118b9c1b51eSKate Stone   case KEY_SR:
3119b9c1b51eSKate Stone     return "scroll-backward";
3120b9c1b51eSKate Stone   case KEY_NPAGE:
3121b9c1b51eSKate Stone     return "page-down";
3122b9c1b51eSKate Stone   case KEY_PPAGE:
3123b9c1b51eSKate Stone     return "page-up";
3124b9c1b51eSKate Stone   case KEY_STAB:
3125b9c1b51eSKate Stone     return "set-tab";
3126b9c1b51eSKate Stone   case KEY_CTAB:
3127b9c1b51eSKate Stone     return "clear-tab";
3128b9c1b51eSKate Stone   case KEY_CATAB:
3129b9c1b51eSKate Stone     return "clear-all-tabs";
3130b9c1b51eSKate Stone   case KEY_ENTER:
3131b9c1b51eSKate Stone     return "enter";
3132b9c1b51eSKate Stone   case KEY_PRINT:
3133b9c1b51eSKate Stone     return "print";
3134b9c1b51eSKate Stone   case KEY_LL:
3135b9c1b51eSKate Stone     return "lower-left key";
3136b9c1b51eSKate Stone   case KEY_A1:
3137b9c1b51eSKate Stone     return "upper left of keypad";
3138b9c1b51eSKate Stone   case KEY_A3:
3139b9c1b51eSKate Stone     return "upper right of keypad";
3140b9c1b51eSKate Stone   case KEY_B2:
3141b9c1b51eSKate Stone     return "center of keypad";
3142b9c1b51eSKate Stone   case KEY_C1:
3143b9c1b51eSKate Stone     return "lower left of keypad";
3144b9c1b51eSKate Stone   case KEY_C3:
3145b9c1b51eSKate Stone     return "lower right of keypad";
3146b9c1b51eSKate Stone   case KEY_BTAB:
3147b9c1b51eSKate Stone     return "back-tab key";
3148b9c1b51eSKate Stone   case KEY_BEG:
3149b9c1b51eSKate Stone     return "begin key";
3150b9c1b51eSKate Stone   case KEY_CANCEL:
3151b9c1b51eSKate Stone     return "cancel key";
3152b9c1b51eSKate Stone   case KEY_CLOSE:
3153b9c1b51eSKate Stone     return "close key";
3154b9c1b51eSKate Stone   case KEY_COMMAND:
3155b9c1b51eSKate Stone     return "command key";
3156b9c1b51eSKate Stone   case KEY_COPY:
3157b9c1b51eSKate Stone     return "copy key";
3158b9c1b51eSKate Stone   case KEY_CREATE:
3159b9c1b51eSKate Stone     return "create key";
3160b9c1b51eSKate Stone   case KEY_END:
3161b9c1b51eSKate Stone     return "end key";
3162b9c1b51eSKate Stone   case KEY_EXIT:
3163b9c1b51eSKate Stone     return "exit key";
3164b9c1b51eSKate Stone   case KEY_FIND:
3165b9c1b51eSKate Stone     return "find key";
3166b9c1b51eSKate Stone   case KEY_HELP:
3167b9c1b51eSKate Stone     return "help key";
3168b9c1b51eSKate Stone   case KEY_MARK:
3169b9c1b51eSKate Stone     return "mark key";
3170b9c1b51eSKate Stone   case KEY_MESSAGE:
3171b9c1b51eSKate Stone     return "message key";
3172b9c1b51eSKate Stone   case KEY_MOVE:
3173b9c1b51eSKate Stone     return "move key";
3174b9c1b51eSKate Stone   case KEY_NEXT:
3175b9c1b51eSKate Stone     return "next key";
3176b9c1b51eSKate Stone   case KEY_OPEN:
3177b9c1b51eSKate Stone     return "open key";
3178b9c1b51eSKate Stone   case KEY_OPTIONS:
3179b9c1b51eSKate Stone     return "options key";
3180b9c1b51eSKate Stone   case KEY_PREVIOUS:
3181b9c1b51eSKate Stone     return "previous key";
3182b9c1b51eSKate Stone   case KEY_REDO:
3183b9c1b51eSKate Stone     return "redo key";
3184b9c1b51eSKate Stone   case KEY_REFERENCE:
3185b9c1b51eSKate Stone     return "reference key";
3186b9c1b51eSKate Stone   case KEY_REFRESH:
3187b9c1b51eSKate Stone     return "refresh key";
3188b9c1b51eSKate Stone   case KEY_REPLACE:
3189b9c1b51eSKate Stone     return "replace key";
3190b9c1b51eSKate Stone   case KEY_RESTART:
3191b9c1b51eSKate Stone     return "restart key";
3192b9c1b51eSKate Stone   case KEY_RESUME:
3193b9c1b51eSKate Stone     return "resume key";
3194b9c1b51eSKate Stone   case KEY_SAVE:
3195b9c1b51eSKate Stone     return "save key";
3196b9c1b51eSKate Stone   case KEY_SBEG:
3197b9c1b51eSKate Stone     return "shifted begin key";
3198b9c1b51eSKate Stone   case KEY_SCANCEL:
3199b9c1b51eSKate Stone     return "shifted cancel key";
3200b9c1b51eSKate Stone   case KEY_SCOMMAND:
3201b9c1b51eSKate Stone     return "shifted command key";
3202b9c1b51eSKate Stone   case KEY_SCOPY:
3203b9c1b51eSKate Stone     return "shifted copy key";
3204b9c1b51eSKate Stone   case KEY_SCREATE:
3205b9c1b51eSKate Stone     return "shifted create key";
3206b9c1b51eSKate Stone   case KEY_SDC:
3207b9c1b51eSKate Stone     return "shifted delete-character key";
3208b9c1b51eSKate Stone   case KEY_SDL:
3209b9c1b51eSKate Stone     return "shifted delete-line key";
3210b9c1b51eSKate Stone   case KEY_SELECT:
3211b9c1b51eSKate Stone     return "select key";
3212b9c1b51eSKate Stone   case KEY_SEND:
3213b9c1b51eSKate Stone     return "shifted end key";
3214b9c1b51eSKate Stone   case KEY_SEOL:
3215b9c1b51eSKate Stone     return "shifted clear-to-end-of-line key";
3216b9c1b51eSKate Stone   case KEY_SEXIT:
3217b9c1b51eSKate Stone     return "shifted exit key";
3218b9c1b51eSKate Stone   case KEY_SFIND:
3219b9c1b51eSKate Stone     return "shifted find key";
3220b9c1b51eSKate Stone   case KEY_SHELP:
3221b9c1b51eSKate Stone     return "shifted help key";
3222b9c1b51eSKate Stone   case KEY_SHOME:
3223b9c1b51eSKate Stone     return "shifted home key";
3224b9c1b51eSKate Stone   case KEY_SIC:
3225b9c1b51eSKate Stone     return "shifted insert-character key";
3226b9c1b51eSKate Stone   case KEY_SLEFT:
3227b9c1b51eSKate Stone     return "shifted left-arrow key";
3228b9c1b51eSKate Stone   case KEY_SMESSAGE:
3229b9c1b51eSKate Stone     return "shifted message key";
3230b9c1b51eSKate Stone   case KEY_SMOVE:
3231b9c1b51eSKate Stone     return "shifted move key";
3232b9c1b51eSKate Stone   case KEY_SNEXT:
3233b9c1b51eSKate Stone     return "shifted next key";
3234b9c1b51eSKate Stone   case KEY_SOPTIONS:
3235b9c1b51eSKate Stone     return "shifted options key";
3236b9c1b51eSKate Stone   case KEY_SPREVIOUS:
3237b9c1b51eSKate Stone     return "shifted previous key";
3238b9c1b51eSKate Stone   case KEY_SPRINT:
3239b9c1b51eSKate Stone     return "shifted print key";
3240b9c1b51eSKate Stone   case KEY_SREDO:
3241b9c1b51eSKate Stone     return "shifted redo key";
3242b9c1b51eSKate Stone   case KEY_SREPLACE:
3243b9c1b51eSKate Stone     return "shifted replace key";
3244b9c1b51eSKate Stone   case KEY_SRIGHT:
3245b9c1b51eSKate Stone     return "shifted right-arrow key";
3246b9c1b51eSKate Stone   case KEY_SRSUME:
3247b9c1b51eSKate Stone     return "shifted resume key";
3248b9c1b51eSKate Stone   case KEY_SSAVE:
3249b9c1b51eSKate Stone     return "shifted save key";
3250b9c1b51eSKate Stone   case KEY_SSUSPEND:
3251b9c1b51eSKate Stone     return "shifted suspend key";
3252b9c1b51eSKate Stone   case KEY_SUNDO:
3253b9c1b51eSKate Stone     return "shifted undo key";
3254b9c1b51eSKate Stone   case KEY_SUSPEND:
3255b9c1b51eSKate Stone     return "suspend key";
3256b9c1b51eSKate Stone   case KEY_UNDO:
3257b9c1b51eSKate Stone     return "undo key";
3258b9c1b51eSKate Stone   case KEY_MOUSE:
3259b9c1b51eSKate Stone     return "Mouse event has occurred";
3260b9c1b51eSKate Stone   case KEY_RESIZE:
3261b9c1b51eSKate Stone     return "Terminal resize event";
326227801f4fSBruce Mitchener #ifdef KEY_EVENT
3263b9c1b51eSKate Stone   case KEY_EVENT:
3264b9c1b51eSKate Stone     return "We were interrupted by an event";
326527801f4fSBruce Mitchener #endif
3266b9c1b51eSKate Stone   case KEY_RETURN:
3267b9c1b51eSKate Stone     return "return";
3268b9c1b51eSKate Stone   case ' ':
3269b9c1b51eSKate Stone     return "space";
3270b9c1b51eSKate Stone   case '\t':
3271b9c1b51eSKate Stone     return "tab";
3272b9c1b51eSKate Stone   case KEY_ESCAPE:
3273b9c1b51eSKate Stone     return "escape";
327444d93782SGreg Clayton   default:
327544d93782SGreg Clayton     if (isprint(ch))
327644d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "%c", ch);
327744d93782SGreg Clayton     else
327844d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "\\x%2.2x", ch);
327944d93782SGreg Clayton     return g_desc;
328044d93782SGreg Clayton   }
3281c5dac77aSEugene Zelenko   return nullptr;
328244d93782SGreg Clayton }
328344d93782SGreg Clayton 
3284b9c1b51eSKate Stone HelpDialogDelegate::HelpDialogDelegate(const char *text,
3285b9c1b51eSKate Stone                                        KeyHelp *key_help_array)
3286b9c1b51eSKate Stone     : m_text(), m_first_visible_line(0) {
3287b9c1b51eSKate Stone   if (text && text[0]) {
328844d93782SGreg Clayton     m_text.SplitIntoLines(text);
328944d93782SGreg Clayton     m_text.AppendString("");
329044d93782SGreg Clayton   }
3291b9c1b51eSKate Stone   if (key_help_array) {
3292b9c1b51eSKate Stone     for (KeyHelp *key = key_help_array; key->ch; ++key) {
329344d93782SGreg Clayton       StreamString key_description;
3294b9c1b51eSKate Stone       key_description.Printf("%10s - %s", CursesKeyToCString(key->ch),
3295b9c1b51eSKate Stone                              key->description);
3296c156427dSZachary Turner       m_text.AppendString(key_description.GetString());
329744d93782SGreg Clayton     }
329844d93782SGreg Clayton   }
329944d93782SGreg Clayton }
330044d93782SGreg Clayton 
3301315b6884SEugene Zelenko HelpDialogDelegate::~HelpDialogDelegate() = default;
330244d93782SGreg Clayton 
3303b9c1b51eSKate Stone bool HelpDialogDelegate::WindowDelegateDraw(Window &window, bool force) {
330444d93782SGreg Clayton   window.Erase();
330544d93782SGreg Clayton   const int window_height = window.GetHeight();
330644d93782SGreg Clayton   int x = 2;
330744d93782SGreg Clayton   int y = 1;
330844d93782SGreg Clayton   const int min_y = y;
330944d93782SGreg Clayton   const int max_y = window_height - 1 - y;
33103985c8c6SSaleem Abdulrasool   const size_t num_visible_lines = max_y - min_y + 1;
331144d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
331244d93782SGreg Clayton   const char *bottom_message;
331344d93782SGreg Clayton   if (num_lines <= num_visible_lines)
331444d93782SGreg Clayton     bottom_message = "Press any key to exit";
331544d93782SGreg Clayton   else
331644d93782SGreg Clayton     bottom_message = "Use arrows to scroll, any other key to exit";
331744d93782SGreg Clayton   window.DrawTitleBox(window.GetName(), bottom_message);
3318b9c1b51eSKate Stone   while (y <= max_y) {
331944d93782SGreg Clayton     window.MoveCursor(x, y);
3320b9c1b51eSKate Stone     window.PutCStringTruncated(
3321b9c1b51eSKate Stone         m_text.GetStringAtIndex(m_first_visible_line + y - min_y), 1);
332244d93782SGreg Clayton     ++y;
332344d93782SGreg Clayton   }
332444d93782SGreg Clayton   return true;
332544d93782SGreg Clayton }
332644d93782SGreg Clayton 
3327b9c1b51eSKate Stone HandleCharResult HelpDialogDelegate::WindowDelegateHandleChar(Window &window,
3328b9c1b51eSKate Stone                                                               int key) {
332944d93782SGreg Clayton   bool done = false;
333044d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
333144d93782SGreg Clayton   const size_t num_visible_lines = window.GetHeight() - 2;
333244d93782SGreg Clayton 
3333b9c1b51eSKate Stone   if (num_lines <= num_visible_lines) {
333444d93782SGreg Clayton     done = true;
333505097246SAdrian Prantl     // If we have all lines visible and don't need scrolling, then any key
333605097246SAdrian Prantl     // press will cause us to exit
3337b9c1b51eSKate Stone   } else {
3338b9c1b51eSKate Stone     switch (key) {
333944d93782SGreg Clayton     case KEY_UP:
334044d93782SGreg Clayton       if (m_first_visible_line > 0)
334144d93782SGreg Clayton         --m_first_visible_line;
334244d93782SGreg Clayton       break;
334344d93782SGreg Clayton 
334444d93782SGreg Clayton     case KEY_DOWN:
334544d93782SGreg Clayton       if (m_first_visible_line + num_visible_lines < num_lines)
334644d93782SGreg Clayton         ++m_first_visible_line;
334744d93782SGreg Clayton       break;
334844d93782SGreg Clayton 
334944d93782SGreg Clayton     case KEY_PPAGE:
335044d93782SGreg Clayton     case ',':
3351b9c1b51eSKate Stone       if (m_first_visible_line > 0) {
33523985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_visible_lines)
335344d93782SGreg Clayton           m_first_visible_line -= num_visible_lines;
335444d93782SGreg Clayton         else
335544d93782SGreg Clayton           m_first_visible_line = 0;
335644d93782SGreg Clayton       }
335744d93782SGreg Clayton       break;
3358315b6884SEugene Zelenko 
335944d93782SGreg Clayton     case KEY_NPAGE:
336044d93782SGreg Clayton     case '.':
3361b9c1b51eSKate Stone       if (m_first_visible_line + num_visible_lines < num_lines) {
336244d93782SGreg Clayton         m_first_visible_line += num_visible_lines;
33633985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > num_lines)
336444d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
336544d93782SGreg Clayton       }
336644d93782SGreg Clayton       break;
3367315b6884SEugene Zelenko 
336844d93782SGreg Clayton     default:
336944d93782SGreg Clayton       done = true;
337044d93782SGreg Clayton       break;
337144d93782SGreg Clayton     }
337244d93782SGreg Clayton   }
337344d93782SGreg Clayton   if (done)
337444d93782SGreg Clayton     window.GetParent()->RemoveSubWindow(&window);
337544d93782SGreg Clayton   return eKeyHandled;
337644d93782SGreg Clayton }
337744d93782SGreg Clayton 
3378b9c1b51eSKate Stone class ApplicationDelegate : public WindowDelegate, public MenuDelegate {
337944d93782SGreg Clayton public:
338044d93782SGreg Clayton   enum {
338144d93782SGreg Clayton     eMenuID_LLDB = 1,
338244d93782SGreg Clayton     eMenuID_LLDBAbout,
338344d93782SGreg Clayton     eMenuID_LLDBExit,
338444d93782SGreg Clayton 
338544d93782SGreg Clayton     eMenuID_Target,
338644d93782SGreg Clayton     eMenuID_TargetCreate,
338744d93782SGreg Clayton     eMenuID_TargetDelete,
338844d93782SGreg Clayton 
338944d93782SGreg Clayton     eMenuID_Process,
339044d93782SGreg Clayton     eMenuID_ProcessAttach,
339144d93782SGreg Clayton     eMenuID_ProcessDetach,
339244d93782SGreg Clayton     eMenuID_ProcessLaunch,
339344d93782SGreg Clayton     eMenuID_ProcessContinue,
339444d93782SGreg Clayton     eMenuID_ProcessHalt,
339544d93782SGreg Clayton     eMenuID_ProcessKill,
339644d93782SGreg Clayton 
339744d93782SGreg Clayton     eMenuID_Thread,
339844d93782SGreg Clayton     eMenuID_ThreadStepIn,
339944d93782SGreg Clayton     eMenuID_ThreadStepOver,
340044d93782SGreg Clayton     eMenuID_ThreadStepOut,
340144d93782SGreg Clayton 
340244d93782SGreg Clayton     eMenuID_View,
340344d93782SGreg Clayton     eMenuID_ViewBacktrace,
340444d93782SGreg Clayton     eMenuID_ViewRegisters,
340544d93782SGreg Clayton     eMenuID_ViewSource,
340644d93782SGreg Clayton     eMenuID_ViewVariables,
340744d93782SGreg Clayton 
340844d93782SGreg Clayton     eMenuID_Help,
340944d93782SGreg Clayton     eMenuID_HelpGUIHelp
341044d93782SGreg Clayton   };
341144d93782SGreg Clayton 
3412b9c1b51eSKate Stone   ApplicationDelegate(Application &app, Debugger &debugger)
3413b9c1b51eSKate Stone       : WindowDelegate(), MenuDelegate(), m_app(app), m_debugger(debugger) {}
341444d93782SGreg Clayton 
3415315b6884SEugene Zelenko   ~ApplicationDelegate() override = default;
3416bd5ae6b4SGreg Clayton 
3417b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
341844d93782SGreg Clayton     return false; // Drawing not handled, let standard window drawing happen
341944d93782SGreg Clayton   }
342044d93782SGreg Clayton 
3421b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override {
3422b9c1b51eSKate Stone     switch (key) {
34235fdb09bbSGreg Clayton     case '\t':
342444d93782SGreg Clayton       window.SelectNextWindowAsActive();
342544d93782SGreg Clayton       return eKeyHandled;
34265fdb09bbSGreg Clayton 
34275fdb09bbSGreg Clayton     case 'h':
34285fdb09bbSGreg Clayton       window.CreateHelpSubwindow();
34295fdb09bbSGreg Clayton       return eKeyHandled;
34305fdb09bbSGreg Clayton 
34315fdb09bbSGreg Clayton     case KEY_ESCAPE:
34325fdb09bbSGreg Clayton       return eQuitApplication;
34335fdb09bbSGreg Clayton 
34345fdb09bbSGreg Clayton     default:
34355fdb09bbSGreg Clayton       break;
343644d93782SGreg Clayton     }
343744d93782SGreg Clayton     return eKeyNotHandled;
343844d93782SGreg Clayton   }
343944d93782SGreg Clayton 
3440b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
34415fdb09bbSGreg Clayton     return "Welcome to the LLDB curses GUI.\n\n"
34425fdb09bbSGreg Clayton            "Press the TAB key to change the selected view.\n"
3443b9c1b51eSKate Stone            "Each view has its own keyboard shortcuts, press 'h' to open a "
3444b9c1b51eSKate Stone            "dialog to display them.\n\n"
34455fdb09bbSGreg Clayton            "Common key bindings for all views:";
34465fdb09bbSGreg Clayton   }
34475fdb09bbSGreg Clayton 
3448b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
34495fdb09bbSGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
34505fdb09bbSGreg Clayton         {'\t', "Select next view"},
34515fdb09bbSGreg Clayton         {'h', "Show help dialog with view specific key bindings"},
34525fdb09bbSGreg Clayton         {',', "Page up"},
34535fdb09bbSGreg Clayton         {'.', "Page down"},
34545fdb09bbSGreg Clayton         {KEY_UP, "Select previous"},
34555fdb09bbSGreg Clayton         {KEY_DOWN, "Select next"},
34565fdb09bbSGreg Clayton         {KEY_LEFT, "Unexpand or select parent"},
34575fdb09bbSGreg Clayton         {KEY_RIGHT, "Expand"},
34585fdb09bbSGreg Clayton         {KEY_PPAGE, "Page up"},
34595fdb09bbSGreg Clayton         {KEY_NPAGE, "Page down"},
3460b9c1b51eSKate Stone         {'\0', nullptr}};
34615fdb09bbSGreg Clayton     return g_source_view_key_help;
34625fdb09bbSGreg Clayton   }
34635fdb09bbSGreg Clayton 
3464b9c1b51eSKate Stone   MenuActionResult MenuDelegateAction(Menu &menu) override {
3465b9c1b51eSKate Stone     switch (menu.GetIdentifier()) {
3466b9c1b51eSKate Stone     case eMenuID_ThreadStepIn: {
3467b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3468b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3469b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
347044d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3471b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3472b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
34734b4b2478SJim Ingham           exe_ctx.GetThreadRef().StepIn(true);
347444d93782SGreg Clayton       }
347544d93782SGreg Clayton     }
347644d93782SGreg Clayton       return MenuActionResult::Handled;
347744d93782SGreg Clayton 
3478b9c1b51eSKate Stone     case eMenuID_ThreadStepOut: {
3479b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3480b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3481b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
348244d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3483b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3484b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
348544d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
348644d93782SGreg Clayton       }
348744d93782SGreg Clayton     }
348844d93782SGreg Clayton       return MenuActionResult::Handled;
348944d93782SGreg Clayton 
3490b9c1b51eSKate Stone     case eMenuID_ThreadStepOver: {
3491b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3492b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3493b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
349444d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3495b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3496b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
349744d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOver(true);
349844d93782SGreg Clayton       }
349944d93782SGreg Clayton     }
350044d93782SGreg Clayton       return MenuActionResult::Handled;
350144d93782SGreg Clayton 
3502b9c1b51eSKate Stone     case eMenuID_ProcessContinue: {
3503b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3504b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3505b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
350644d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3507b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3508b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
350944d93782SGreg Clayton           process->Resume();
351044d93782SGreg Clayton       }
351144d93782SGreg Clayton     }
351244d93782SGreg Clayton       return MenuActionResult::Handled;
351344d93782SGreg Clayton 
3514b9c1b51eSKate Stone     case eMenuID_ProcessKill: {
3515b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3516b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3517b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
351844d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
351944d93782SGreg Clayton         if (process && process->IsAlive())
3520ede3193bSJason Molenda           process->Destroy(false);
352144d93782SGreg Clayton       }
352244d93782SGreg Clayton     }
352344d93782SGreg Clayton       return MenuActionResult::Handled;
352444d93782SGreg Clayton 
3525b9c1b51eSKate Stone     case eMenuID_ProcessHalt: {
3526b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3527b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3528b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
352944d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
353044d93782SGreg Clayton         if (process && process->IsAlive())
353144d93782SGreg Clayton           process->Halt();
353244d93782SGreg Clayton       }
353344d93782SGreg Clayton     }
353444d93782SGreg Clayton       return MenuActionResult::Handled;
353544d93782SGreg Clayton 
3536b9c1b51eSKate Stone     case eMenuID_ProcessDetach: {
3537b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3538b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3539b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
354044d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
354144d93782SGreg Clayton         if (process && process->IsAlive())
354244d93782SGreg Clayton           process->Detach(false);
354344d93782SGreg Clayton       }
354444d93782SGreg Clayton     }
354544d93782SGreg Clayton       return MenuActionResult::Handled;
354644d93782SGreg Clayton 
3547b9c1b51eSKate Stone     case eMenuID_Process: {
3548b9c1b51eSKate Stone       // Populate the menu with all of the threads if the process is stopped
354905097246SAdrian Prantl       // when the Process menu gets selected and is about to display its
355005097246SAdrian Prantl       // submenu.
355144d93782SGreg Clayton       Menus &submenus = menu.GetSubmenus();
3552b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3553b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
355444d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
3555b9c1b51eSKate Stone       if (process && process->IsAlive() &&
3556b9c1b51eSKate Stone           StateIsStoppedState(process->GetState(), true)) {
355744d93782SGreg Clayton         if (submenus.size() == 7)
355844d93782SGreg Clayton           menu.AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
355944d93782SGreg Clayton         else if (submenus.size() > 8)
356044d93782SGreg Clayton           submenus.erase(submenus.begin() + 8, submenus.end());
356144d93782SGreg Clayton 
356244d93782SGreg Clayton         ThreadList &threads = process->GetThreadList();
3563bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
356444d93782SGreg Clayton         size_t num_threads = threads.GetSize();
3565b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
356644d93782SGreg Clayton           ThreadSP thread_sp = threads.GetThreadAtIndex(i);
356744d93782SGreg Clayton           char menu_char = '\0';
356844d93782SGreg Clayton           if (i < 9)
356944d93782SGreg Clayton             menu_char = '1' + i;
357044d93782SGreg Clayton           StreamString thread_menu_title;
357144d93782SGreg Clayton           thread_menu_title.Printf("Thread %u", thread_sp->GetIndexID());
357244d93782SGreg Clayton           const char *thread_name = thread_sp->GetName();
357344d93782SGreg Clayton           if (thread_name && thread_name[0])
357444d93782SGreg Clayton             thread_menu_title.Printf(" %s", thread_name);
3575b9c1b51eSKate Stone           else {
357644d93782SGreg Clayton             const char *queue_name = thread_sp->GetQueueName();
357744d93782SGreg Clayton             if (queue_name && queue_name[0])
357844d93782SGreg Clayton               thread_menu_title.Printf(" %s", queue_name);
357944d93782SGreg Clayton           }
3580b9c1b51eSKate Stone           menu.AddSubmenu(
3581c156427dSZachary Turner               MenuSP(new Menu(thread_menu_title.GetString().str().c_str(),
3582c156427dSZachary Turner                               nullptr, menu_char, thread_sp->GetID())));
358344d93782SGreg Clayton         }
3584b9c1b51eSKate Stone       } else if (submenus.size() > 7) {
358505097246SAdrian Prantl         // Remove the separator and any other thread submenu items that were
358605097246SAdrian Prantl         // previously added
358744d93782SGreg Clayton         submenus.erase(submenus.begin() + 7, submenus.end());
358844d93782SGreg Clayton       }
3589b9c1b51eSKate Stone       // Since we are adding and removing items we need to recalculate the name
3590b9c1b51eSKate Stone       // lengths
359144d93782SGreg Clayton       menu.RecalculateNameLengths();
359244d93782SGreg Clayton     }
359344d93782SGreg Clayton       return MenuActionResult::Handled;
359444d93782SGreg Clayton 
3595b9c1b51eSKate Stone     case eMenuID_ViewVariables: {
359644d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
359744d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
359844d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
359944d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
360044d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
360144d93782SGreg Clayton 
3602b9c1b51eSKate Stone       if (variables_window_sp) {
360344d93782SGreg Clayton         const Rect variables_bounds = variables_window_sp->GetBounds();
360444d93782SGreg Clayton 
360544d93782SGreg Clayton         main_window_sp->RemoveSubWindow(variables_window_sp.get());
360644d93782SGreg Clayton 
3607b9c1b51eSKate Stone         if (registers_window_sp) {
3608b9c1b51eSKate Stone           // We have a registers window, so give all the area back to the
3609b9c1b51eSKate Stone           // registers window
361044d93782SGreg Clayton           Rect registers_bounds = variables_bounds;
361144d93782SGreg Clayton           registers_bounds.size.width = source_bounds.size.width;
361244d93782SGreg Clayton           registers_window_sp->SetBounds(registers_bounds);
3613b9c1b51eSKate Stone         } else {
361405097246SAdrian Prantl           // We have no registers window showing so give the bottom area back
361505097246SAdrian Prantl           // to the source view
361644d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3617b9c1b51eSKate Stone                                    source_bounds.size.height +
3618b9c1b51eSKate Stone                                        variables_bounds.size.height);
361944d93782SGreg Clayton         }
3620b9c1b51eSKate Stone       } else {
362144d93782SGreg Clayton         Rect new_variables_rect;
3622b9c1b51eSKate Stone         if (registers_window_sp) {
362344d93782SGreg Clayton           // We have a registers window so split the area of the registers
362444d93782SGreg Clayton           // window into two columns where the left hand side will be the
362544d93782SGreg Clayton           // variables and the right hand side will be the registers
362644d93782SGreg Clayton           const Rect variables_bounds = registers_window_sp->GetBounds();
362744d93782SGreg Clayton           Rect new_registers_rect;
3628b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_variables_rect,
3629b9c1b51eSKate Stone                                                    new_registers_rect);
363044d93782SGreg Clayton           registers_window_sp->SetBounds(new_registers_rect);
3631b9c1b51eSKate Stone         } else {
363244d93782SGreg Clayton           // No variables window, grab the bottom part of the source window
363344d93782SGreg Clayton           Rect new_source_rect;
3634b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3635b9c1b51eSKate Stone                                                   new_variables_rect);
363644d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
363744d93782SGreg Clayton         }
3638b9c1b51eSKate Stone         WindowSP new_window_sp = main_window_sp->CreateSubWindow(
3639b9c1b51eSKate Stone             "Variables", new_variables_rect, false);
3640b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3641b9c1b51eSKate Stone             WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
364244d93782SGreg Clayton       }
364344d93782SGreg Clayton       touchwin(stdscr);
364444d93782SGreg Clayton     }
364544d93782SGreg Clayton       return MenuActionResult::Handled;
364644d93782SGreg Clayton 
3647b9c1b51eSKate Stone     case eMenuID_ViewRegisters: {
364844d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
364944d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
365044d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
365144d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
365244d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
365344d93782SGreg Clayton 
3654b9c1b51eSKate Stone       if (registers_window_sp) {
3655b9c1b51eSKate Stone         if (variables_window_sp) {
365644d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
365744d93782SGreg Clayton 
3658b9c1b51eSKate Stone           // We have a variables window, so give all the area back to the
3659b9c1b51eSKate Stone           // variables window
3660b9c1b51eSKate Stone           variables_window_sp->Resize(variables_bounds.size.width +
3661b9c1b51eSKate Stone                                           registers_window_sp->GetWidth(),
366244d93782SGreg Clayton                                       variables_bounds.size.height);
3663b9c1b51eSKate Stone         } else {
366405097246SAdrian Prantl           // We have no variables window showing so give the bottom area back
366505097246SAdrian Prantl           // to the source view
366644d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3667b9c1b51eSKate Stone                                    source_bounds.size.height +
3668b9c1b51eSKate Stone                                        registers_window_sp->GetHeight());
366944d93782SGreg Clayton         }
367044d93782SGreg Clayton         main_window_sp->RemoveSubWindow(registers_window_sp.get());
3671b9c1b51eSKate Stone       } else {
367244d93782SGreg Clayton         Rect new_regs_rect;
3673b9c1b51eSKate Stone         if (variables_window_sp) {
367405097246SAdrian Prantl           // We have a variables window, split it into two columns where the
367505097246SAdrian Prantl           // left hand side will be the variables and the right hand side will
367605097246SAdrian Prantl           // be the registers
367744d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
367844d93782SGreg Clayton           Rect new_vars_rect;
3679b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_vars_rect,
3680b9c1b51eSKate Stone                                                    new_regs_rect);
368144d93782SGreg Clayton           variables_window_sp->SetBounds(new_vars_rect);
3682b9c1b51eSKate Stone         } else {
368344d93782SGreg Clayton           // No registers window, grab the bottom part of the source window
368444d93782SGreg Clayton           Rect new_source_rect;
3685b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3686b9c1b51eSKate Stone                                                   new_regs_rect);
368744d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
368844d93782SGreg Clayton         }
3689b9c1b51eSKate Stone         WindowSP new_window_sp =
3690b9c1b51eSKate Stone             main_window_sp->CreateSubWindow("Registers", new_regs_rect, false);
3691b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3692b9c1b51eSKate Stone             WindowDelegateSP(new RegistersWindowDelegate(m_debugger)));
369344d93782SGreg Clayton       }
369444d93782SGreg Clayton       touchwin(stdscr);
369544d93782SGreg Clayton     }
369644d93782SGreg Clayton       return MenuActionResult::Handled;
369744d93782SGreg Clayton 
369844d93782SGreg Clayton     case eMenuID_HelpGUIHelp:
36995fdb09bbSGreg Clayton       m_app.GetMainWindow()->CreateHelpSubwindow();
370044d93782SGreg Clayton       return MenuActionResult::Handled;
370144d93782SGreg Clayton 
370244d93782SGreg Clayton     default:
370344d93782SGreg Clayton       break;
370444d93782SGreg Clayton     }
370544d93782SGreg Clayton 
370644d93782SGreg Clayton     return MenuActionResult::NotHandled;
370744d93782SGreg Clayton   }
3708b9c1b51eSKate Stone 
370944d93782SGreg Clayton protected:
371044d93782SGreg Clayton   Application &m_app;
371144d93782SGreg Clayton   Debugger &m_debugger;
371244d93782SGreg Clayton };
371344d93782SGreg Clayton 
3714b9c1b51eSKate Stone class StatusBarWindowDelegate : public WindowDelegate {
371544d93782SGreg Clayton public:
3716b9c1b51eSKate Stone   StatusBarWindowDelegate(Debugger &debugger) : m_debugger(debugger) {
3717b9c1b51eSKate Stone     FormatEntity::Parse("Thread: ${thread.id%tid}", m_format);
371844d93782SGreg Clayton   }
371944d93782SGreg Clayton 
3720315b6884SEugene Zelenko   ~StatusBarWindowDelegate() override = default;
3721bd5ae6b4SGreg Clayton 
3722b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3723b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3724b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
372544d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
372644d93782SGreg Clayton     Thread *thread = exe_ctx.GetThreadPtr();
372744d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
372844d93782SGreg Clayton     window.Erase();
372944d93782SGreg Clayton     window.SetBackground(2);
373044d93782SGreg Clayton     window.MoveCursor(0, 0);
3731b9c1b51eSKate Stone     if (process) {
373244d93782SGreg Clayton       const StateType state = process->GetState();
3733b9c1b51eSKate Stone       window.Printf("Process: %5" PRIu64 " %10s", process->GetID(),
3734b9c1b51eSKate Stone                     StateAsCString(state));
373544d93782SGreg Clayton 
3736b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
37375b031ebcSEd Maste         StreamString strm;
3738b9c1b51eSKate Stone         if (thread && FormatEntity::Format(m_format, strm, nullptr, &exe_ctx,
3739b9c1b51eSKate Stone                                            nullptr, nullptr, false, false)) {
374044d93782SGreg Clayton           window.MoveCursor(40, 0);
3741c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), 1);
37425b031ebcSEd Maste         }
374344d93782SGreg Clayton 
374444d93782SGreg Clayton         window.MoveCursor(60, 0);
374544d93782SGreg Clayton         if (frame)
3746b9c1b51eSKate Stone           window.Printf("Frame: %3u  PC = 0x%16.16" PRIx64,
3747b9c1b51eSKate Stone                         frame->GetFrameIndex(),
3748b9c1b51eSKate Stone                         frame->GetFrameCodeAddress().GetOpcodeLoadAddress(
3749b9c1b51eSKate Stone                             exe_ctx.GetTargetPtr()));
3750b9c1b51eSKate Stone       } else if (state == eStateExited) {
375144d93782SGreg Clayton         const char *exit_desc = process->GetExitDescription();
375244d93782SGreg Clayton         const int exit_status = process->GetExitStatus();
375344d93782SGreg Clayton         if (exit_desc && exit_desc[0])
375444d93782SGreg Clayton           window.Printf(" with status = %i (%s)", exit_status, exit_desc);
375544d93782SGreg Clayton         else
375644d93782SGreg Clayton           window.Printf(" with status = %i", exit_status);
375744d93782SGreg Clayton       }
375844d93782SGreg Clayton     }
375944d93782SGreg Clayton     return true;
376044d93782SGreg Clayton   }
376144d93782SGreg Clayton 
376244d93782SGreg Clayton protected:
376344d93782SGreg Clayton   Debugger &m_debugger;
3764554f68d3SGreg Clayton   FormatEntity::Entry m_format;
376544d93782SGreg Clayton };
376644d93782SGreg Clayton 
3767b9c1b51eSKate Stone class SourceFileWindowDelegate : public WindowDelegate {
376844d93782SGreg Clayton public:
3769b9c1b51eSKate Stone   SourceFileWindowDelegate(Debugger &debugger)
3770b9c1b51eSKate Stone       : WindowDelegate(), m_debugger(debugger), m_sc(), m_file_sp(),
3771b9c1b51eSKate Stone         m_disassembly_scope(nullptr), m_disassembly_sp(), m_disassembly_range(),
3772b9c1b51eSKate Stone         m_title(), m_line_width(4), m_selected_line(0), m_pc_line(0),
3773b9c1b51eSKate Stone         m_stop_id(0), m_frame_idx(UINT32_MAX), m_first_visible_line(0),
3774b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
377544d93782SGreg Clayton 
3776315b6884SEugene Zelenko   ~SourceFileWindowDelegate() override = default;
377744d93782SGreg Clayton 
3778b9c1b51eSKate Stone   void Update(const SymbolContext &sc) { m_sc = sc; }
377944d93782SGreg Clayton 
3780b9c1b51eSKate Stone   uint32_t NumVisibleLines() const { return m_max_y - m_min_y; }
378144d93782SGreg Clayton 
3782b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
378344d93782SGreg Clayton     return "Source/Disassembly window keyboard shortcuts:";
378444d93782SGreg Clayton   }
378544d93782SGreg Clayton 
3786b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
378744d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
378844d93782SGreg Clayton         {KEY_RETURN, "Run to selected line with one shot breakpoint"},
378944d93782SGreg Clayton         {KEY_UP, "Select previous source line"},
379044d93782SGreg Clayton         {KEY_DOWN, "Select next source line"},
379144d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
379244d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
379344d93782SGreg Clayton         {'b', "Set breakpoint on selected source/disassembly line"},
379444d93782SGreg Clayton         {'c', "Continue process"},
379544d93782SGreg Clayton         {'d', "Detach and resume process"},
379644d93782SGreg Clayton         {'D', "Detach with process suspended"},
379744d93782SGreg Clayton         {'h', "Show help dialog"},
379844d93782SGreg Clayton         {'k', "Kill process"},
379944d93782SGreg Clayton         {'n', "Step over (source line)"},
380044d93782SGreg Clayton         {'N', "Step over (single instruction)"},
380144d93782SGreg Clayton         {'o', "Step out"},
380244d93782SGreg Clayton         {'s', "Step in (source line)"},
380344d93782SGreg Clayton         {'S', "Step in (single instruction)"},
380444d93782SGreg Clayton         {',', "Page up"},
380544d93782SGreg Clayton         {'.', "Page down"},
3806b9c1b51eSKate Stone         {'\0', nullptr}};
380744d93782SGreg Clayton     return g_source_view_key_help;
380844d93782SGreg Clayton   }
380944d93782SGreg Clayton 
3810b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3811b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3812b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
381344d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
3814c5dac77aSEugene Zelenko     Thread *thread = nullptr;
381544d93782SGreg Clayton 
381644d93782SGreg Clayton     bool update_location = false;
3817b9c1b51eSKate Stone     if (process) {
381844d93782SGreg Clayton       StateType state = process->GetState();
3819b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
382044d93782SGreg Clayton         // We are stopped, so it is ok to
382144d93782SGreg Clayton         update_location = true;
382244d93782SGreg Clayton       }
382344d93782SGreg Clayton     }
382444d93782SGreg Clayton 
382544d93782SGreg Clayton     m_min_x = 1;
3826ec990867SGreg Clayton     m_min_y = 2;
382744d93782SGreg Clayton     m_max_x = window.GetMaxX() - 1;
382844d93782SGreg Clayton     m_max_y = window.GetMaxY() - 1;
382944d93782SGreg Clayton 
383044d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
383144d93782SGreg Clayton     StackFrameSP frame_sp;
383244d93782SGreg Clayton     bool set_selected_line_to_pc = false;
383344d93782SGreg Clayton 
3834b9c1b51eSKate Stone     if (update_location) {
383544d93782SGreg Clayton       const bool process_alive = process ? process->IsAlive() : false;
383644d93782SGreg Clayton       bool thread_changed = false;
3837b9c1b51eSKate Stone       if (process_alive) {
383844d93782SGreg Clayton         thread = exe_ctx.GetThreadPtr();
3839b9c1b51eSKate Stone         if (thread) {
384044d93782SGreg Clayton           frame_sp = thread->GetSelectedFrame();
384144d93782SGreg Clayton           auto tid = thread->GetID();
384244d93782SGreg Clayton           thread_changed = tid != m_tid;
384344d93782SGreg Clayton           m_tid = tid;
3844b9c1b51eSKate Stone         } else {
3845b9c1b51eSKate Stone           if (m_tid != LLDB_INVALID_THREAD_ID) {
384644d93782SGreg Clayton             thread_changed = true;
384744d93782SGreg Clayton             m_tid = LLDB_INVALID_THREAD_ID;
384844d93782SGreg Clayton           }
384944d93782SGreg Clayton         }
385044d93782SGreg Clayton       }
385144d93782SGreg Clayton       const uint32_t stop_id = process ? process->GetStopID() : 0;
385244d93782SGreg Clayton       const bool stop_id_changed = stop_id != m_stop_id;
385344d93782SGreg Clayton       bool frame_changed = false;
385444d93782SGreg Clayton       m_stop_id = stop_id;
3855ec990867SGreg Clayton       m_title.Clear();
3856b9c1b51eSKate Stone       if (frame_sp) {
385744d93782SGreg Clayton         m_sc = frame_sp->GetSymbolContext(eSymbolContextEverything);
3858b9c1b51eSKate Stone         if (m_sc.module_sp) {
3859b9c1b51eSKate Stone           m_title.Printf(
3860b9c1b51eSKate Stone               "%s", m_sc.module_sp->GetFileSpec().GetFilename().GetCString());
3861ec990867SGreg Clayton           ConstString func_name = m_sc.GetFunctionName();
3862ec990867SGreg Clayton           if (func_name)
3863ec990867SGreg Clayton             m_title.Printf("`%s", func_name.GetCString());
3864ec990867SGreg Clayton         }
386544d93782SGreg Clayton         const uint32_t frame_idx = frame_sp->GetFrameIndex();
386644d93782SGreg Clayton         frame_changed = frame_idx != m_frame_idx;
386744d93782SGreg Clayton         m_frame_idx = frame_idx;
3868b9c1b51eSKate Stone       } else {
386944d93782SGreg Clayton         m_sc.Clear(true);
387044d93782SGreg Clayton         frame_changed = m_frame_idx != UINT32_MAX;
387144d93782SGreg Clayton         m_frame_idx = UINT32_MAX;
387244d93782SGreg Clayton       }
387344d93782SGreg Clayton 
3874b9c1b51eSKate Stone       const bool context_changed =
3875b9c1b51eSKate Stone           thread_changed || frame_changed || stop_id_changed;
387644d93782SGreg Clayton 
3877b9c1b51eSKate Stone       if (process_alive) {
3878b9c1b51eSKate Stone         if (m_sc.line_entry.IsValid()) {
387944d93782SGreg Clayton           m_pc_line = m_sc.line_entry.line;
388044d93782SGreg Clayton           if (m_pc_line != UINT32_MAX)
388144d93782SGreg Clayton             --m_pc_line; // Convert to zero based line number...
388244d93782SGreg Clayton           // Update the selected line if the stop ID changed...
388344d93782SGreg Clayton           if (context_changed)
388444d93782SGreg Clayton             m_selected_line = m_pc_line;
388544d93782SGreg Clayton 
3886b9c1b51eSKate Stone           if (m_file_sp && m_file_sp->FileSpecMatches(m_sc.line_entry.file)) {
388705097246SAdrian Prantl             // Same file, nothing to do, we should either have the lines or not
388805097246SAdrian Prantl             // (source file missing)
3889b9c1b51eSKate Stone             if (m_selected_line >= static_cast<size_t>(m_first_visible_line)) {
389044d93782SGreg Clayton               if (m_selected_line >= m_first_visible_line + num_visible_lines)
389144d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
3892b9c1b51eSKate Stone             } else {
389344d93782SGreg Clayton               if (m_selected_line > 10)
389444d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
389544d93782SGreg Clayton               else
389644d93782SGreg Clayton                 m_first_visible_line = 0;
389744d93782SGreg Clayton             }
3898b9c1b51eSKate Stone           } else {
389944d93782SGreg Clayton             // File changed, set selected line to the line with the PC
390044d93782SGreg Clayton             m_selected_line = m_pc_line;
3901b9c1b51eSKate Stone             m_file_sp =
3902b9c1b51eSKate Stone                 m_debugger.GetSourceManager().GetFile(m_sc.line_entry.file);
3903b9c1b51eSKate Stone             if (m_file_sp) {
390444d93782SGreg Clayton               const size_t num_lines = m_file_sp->GetNumLines();
3905eebf32faSPavel Labath               m_line_width = 1;
390644d93782SGreg Clayton               for (size_t n = num_lines; n >= 10; n = n / 10)
390744d93782SGreg Clayton                 ++m_line_width;
390844d93782SGreg Clayton 
3909b9c1b51eSKate Stone               if (num_lines < num_visible_lines ||
3910b9c1b51eSKate Stone                   m_selected_line < num_visible_lines)
391144d93782SGreg Clayton                 m_first_visible_line = 0;
391244d93782SGreg Clayton               else
391344d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
391444d93782SGreg Clayton             }
391544d93782SGreg Clayton           }
3916b9c1b51eSKate Stone         } else {
391744d93782SGreg Clayton           m_file_sp.reset();
391844d93782SGreg Clayton         }
391944d93782SGreg Clayton 
3920b9c1b51eSKate Stone         if (!m_file_sp || m_file_sp->GetNumLines() == 0) {
392144d93782SGreg Clayton           // Show disassembly
392244d93782SGreg Clayton           bool prefer_file_cache = false;
3923b9c1b51eSKate Stone           if (m_sc.function) {
3924b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.function) {
392544d93782SGreg Clayton               m_disassembly_scope = m_sc.function;
3926b9c1b51eSKate Stone               m_disassembly_sp = m_sc.function->GetInstructions(
3927b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3928b9c1b51eSKate Stone               if (m_disassembly_sp) {
392944d93782SGreg Clayton                 set_selected_line_to_pc = true;
393044d93782SGreg Clayton                 m_disassembly_range = m_sc.function->GetAddressRange();
3931b9c1b51eSKate Stone               } else {
393244d93782SGreg Clayton                 m_disassembly_range.Clear();
393344d93782SGreg Clayton               }
3934b9c1b51eSKate Stone             } else {
393544d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
393644d93782SGreg Clayton             }
3937b9c1b51eSKate Stone           } else if (m_sc.symbol) {
3938b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.symbol) {
393944d93782SGreg Clayton               m_disassembly_scope = m_sc.symbol;
3940b9c1b51eSKate Stone               m_disassembly_sp = m_sc.symbol->GetInstructions(
3941b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3942b9c1b51eSKate Stone               if (m_disassembly_sp) {
394344d93782SGreg Clayton                 set_selected_line_to_pc = true;
3944b9c1b51eSKate Stone                 m_disassembly_range.GetBaseAddress() =
3945b9c1b51eSKate Stone                     m_sc.symbol->GetAddress();
394644d93782SGreg Clayton                 m_disassembly_range.SetByteSize(m_sc.symbol->GetByteSize());
3947b9c1b51eSKate Stone               } else {
394844d93782SGreg Clayton                 m_disassembly_range.Clear();
394944d93782SGreg Clayton               }
3950b9c1b51eSKate Stone             } else {
395144d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
395244d93782SGreg Clayton             }
395344d93782SGreg Clayton           }
395444d93782SGreg Clayton         }
3955b9c1b51eSKate Stone       } else {
395644d93782SGreg Clayton         m_pc_line = UINT32_MAX;
395744d93782SGreg Clayton       }
395844d93782SGreg Clayton     }
395944d93782SGreg Clayton 
3960ec990867SGreg Clayton     const int window_width = window.GetWidth();
396144d93782SGreg Clayton     window.Erase();
396244d93782SGreg Clayton     window.DrawTitleBox("Sources");
3963b9c1b51eSKate Stone     if (!m_title.GetString().empty()) {
3964ec990867SGreg Clayton       window.AttributeOn(A_REVERSE);
3965ec990867SGreg Clayton       window.MoveCursor(1, 1);
3966ec990867SGreg Clayton       window.PutChar(' ');
3967c156427dSZachary Turner       window.PutCStringTruncated(m_title.GetString().str().c_str(), 1);
3968ec990867SGreg Clayton       int x = window.GetCursorX();
3969b9c1b51eSKate Stone       if (x < window_width - 1) {
3970ec990867SGreg Clayton         window.Printf("%*s", window_width - x - 1, "");
3971ec990867SGreg Clayton       }
3972ec990867SGreg Clayton       window.AttributeOff(A_REVERSE);
3973ec990867SGreg Clayton     }
397444d93782SGreg Clayton 
397544d93782SGreg Clayton     Target *target = exe_ctx.GetTargetPtr();
397644d93782SGreg Clayton     const size_t num_source_lines = GetNumSourceLines();
3977b9c1b51eSKate Stone     if (num_source_lines > 0) {
397844d93782SGreg Clayton       // Display source
397944d93782SGreg Clayton       BreakpointLines bp_lines;
3980b9c1b51eSKate Stone       if (target) {
398144d93782SGreg Clayton         BreakpointList &bp_list = target->GetBreakpointList();
398244d93782SGreg Clayton         const size_t num_bps = bp_list.GetSize();
3983b9c1b51eSKate Stone         for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
398444d93782SGreg Clayton           BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
398544d93782SGreg Clayton           const size_t num_bps_locs = bp_sp->GetNumLocations();
3986b9c1b51eSKate Stone           for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
3987b9c1b51eSKate Stone             BreakpointLocationSP bp_loc_sp =
3988b9c1b51eSKate Stone                 bp_sp->GetLocationAtIndex(bp_loc_idx);
398944d93782SGreg Clayton             LineEntry bp_loc_line_entry;
3990b9c1b51eSKate Stone             if (bp_loc_sp->GetAddress().CalculateSymbolContextLineEntry(
3991b9c1b51eSKate Stone                     bp_loc_line_entry)) {
3992b9c1b51eSKate Stone               if (m_file_sp->GetFileSpec() == bp_loc_line_entry.file) {
399344d93782SGreg Clayton                 bp_lines.insert(bp_loc_line_entry.line);
399444d93782SGreg Clayton               }
399544d93782SGreg Clayton             }
399644d93782SGreg Clayton           }
399744d93782SGreg Clayton         }
399844d93782SGreg Clayton       }
399944d93782SGreg Clayton 
400044d93782SGreg Clayton       const attr_t selected_highlight_attr = A_REVERSE;
400144d93782SGreg Clayton       const attr_t pc_highlight_attr = COLOR_PAIR(1);
400244d93782SGreg Clayton 
4003b9c1b51eSKate Stone       for (size_t i = 0; i < num_visible_lines; ++i) {
400444d93782SGreg Clayton         const uint32_t curr_line = m_first_visible_line + i;
4005b9c1b51eSKate Stone         if (curr_line < num_source_lines) {
4006ec990867SGreg Clayton           const int line_y = m_min_y + i;
400744d93782SGreg Clayton           window.MoveCursor(1, line_y);
400844d93782SGreg Clayton           const bool is_pc_line = curr_line == m_pc_line;
400944d93782SGreg Clayton           const bool line_is_selected = m_selected_line == curr_line;
401044d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
401144d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
401244d93782SGreg Clayton           attr_t highlight_attr = 0;
401344d93782SGreg Clayton           attr_t bp_attr = 0;
401444d93782SGreg Clayton           if (is_pc_line)
401544d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
401644d93782SGreg Clayton           else if (line_is_selected)
401744d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
401844d93782SGreg Clayton 
401944d93782SGreg Clayton           if (bp_lines.find(curr_line + 1) != bp_lines.end())
402044d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
402144d93782SGreg Clayton 
402244d93782SGreg Clayton           if (bp_attr)
402344d93782SGreg Clayton             window.AttributeOn(bp_attr);
402444d93782SGreg Clayton 
4025eebf32faSPavel Labath           window.Printf(" %*u ", m_line_width, curr_line + 1);
402644d93782SGreg Clayton 
402744d93782SGreg Clayton           if (bp_attr)
402844d93782SGreg Clayton             window.AttributeOff(bp_attr);
402944d93782SGreg Clayton 
403044d93782SGreg Clayton           window.PutChar(ACS_VLINE);
403144d93782SGreg Clayton           // Mark the line with the PC with a diamond
403244d93782SGreg Clayton           if (is_pc_line)
403344d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
403444d93782SGreg Clayton           else
403544d93782SGreg Clayton             window.PutChar(' ');
403644d93782SGreg Clayton 
403744d93782SGreg Clayton           if (highlight_attr)
403844d93782SGreg Clayton             window.AttributeOn(highlight_attr);
4039b9c1b51eSKate Stone           const uint32_t line_len =
4040b9c1b51eSKate Stone               m_file_sp->GetLineLength(curr_line + 1, false);
404144d93782SGreg Clayton           if (line_len > 0)
404244d93782SGreg Clayton             window.PutCString(m_file_sp->PeekLineData(curr_line + 1), line_len);
404344d93782SGreg Clayton 
4044b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4045b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
404644d93782SGreg Clayton             StopInfoSP stop_info_sp;
404744d93782SGreg Clayton             if (thread)
404844d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4049b9c1b51eSKate Stone             if (stop_info_sp) {
405044d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4051b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
405244d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4053ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
405444d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4055b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4056b9c1b51eSKate Stone                 // line_y);
4057b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4058b9c1b51eSKate Stone                               stop_description);
405944d93782SGreg Clayton               }
4060b9c1b51eSKate Stone             } else {
4061ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
406244d93782SGreg Clayton             }
406344d93782SGreg Clayton           }
406444d93782SGreg Clayton           if (highlight_attr)
406544d93782SGreg Clayton             window.AttributeOff(highlight_attr);
4066b9c1b51eSKate Stone         } else {
406744d93782SGreg Clayton           break;
406844d93782SGreg Clayton         }
406944d93782SGreg Clayton       }
4070b9c1b51eSKate Stone     } else {
407144d93782SGreg Clayton       size_t num_disassembly_lines = GetNumDisassemblyLines();
4072b9c1b51eSKate Stone       if (num_disassembly_lines > 0) {
407344d93782SGreg Clayton         // Display disassembly
407444d93782SGreg Clayton         BreakpointAddrs bp_file_addrs;
407544d93782SGreg Clayton         Target *target = exe_ctx.GetTargetPtr();
4076b9c1b51eSKate Stone         if (target) {
407744d93782SGreg Clayton           BreakpointList &bp_list = target->GetBreakpointList();
407844d93782SGreg Clayton           const size_t num_bps = bp_list.GetSize();
4079b9c1b51eSKate Stone           for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
408044d93782SGreg Clayton             BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
408144d93782SGreg Clayton             const size_t num_bps_locs = bp_sp->GetNumLocations();
4082b9c1b51eSKate Stone             for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs;
4083b9c1b51eSKate Stone                  ++bp_loc_idx) {
4084b9c1b51eSKate Stone               BreakpointLocationSP bp_loc_sp =
4085b9c1b51eSKate Stone                   bp_sp->GetLocationAtIndex(bp_loc_idx);
408644d93782SGreg Clayton               LineEntry bp_loc_line_entry;
4087b9c1b51eSKate Stone               const lldb::addr_t file_addr =
4088b9c1b51eSKate Stone                   bp_loc_sp->GetAddress().GetFileAddress();
4089b9c1b51eSKate Stone               if (file_addr != LLDB_INVALID_ADDRESS) {
409044d93782SGreg Clayton                 if (m_disassembly_range.ContainsFileAddress(file_addr))
409144d93782SGreg Clayton                   bp_file_addrs.insert(file_addr);
409244d93782SGreg Clayton               }
409344d93782SGreg Clayton             }
409444d93782SGreg Clayton           }
409544d93782SGreg Clayton         }
409644d93782SGreg Clayton 
409744d93782SGreg Clayton         const attr_t selected_highlight_attr = A_REVERSE;
409844d93782SGreg Clayton         const attr_t pc_highlight_attr = COLOR_PAIR(1);
409944d93782SGreg Clayton 
410044d93782SGreg Clayton         StreamString strm;
410144d93782SGreg Clayton 
410244d93782SGreg Clayton         InstructionList &insts = m_disassembly_sp->GetInstructionList();
410344d93782SGreg Clayton         Address pc_address;
410444d93782SGreg Clayton 
410544d93782SGreg Clayton         if (frame_sp)
410644d93782SGreg Clayton           pc_address = frame_sp->GetFrameCodeAddress();
4107b9c1b51eSKate Stone         const uint32_t pc_idx =
4108b9c1b51eSKate Stone             pc_address.IsValid()
4109b9c1b51eSKate Stone                 ? insts.GetIndexOfInstructionAtAddress(pc_address)
4110b9c1b51eSKate Stone                 : UINT32_MAX;
4111b9c1b51eSKate Stone         if (set_selected_line_to_pc) {
411244d93782SGreg Clayton           m_selected_line = pc_idx;
411344d93782SGreg Clayton         }
411444d93782SGreg Clayton 
411544d93782SGreg Clayton         const uint32_t non_visible_pc_offset = (num_visible_lines / 5);
41163985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_disassembly_lines)
411744d93782SGreg Clayton           m_first_visible_line = 0;
411844d93782SGreg Clayton 
4119b9c1b51eSKate Stone         if (pc_idx < num_disassembly_lines) {
41203985c8c6SSaleem Abdulrasool           if (pc_idx < static_cast<uint32_t>(m_first_visible_line) ||
412144d93782SGreg Clayton               pc_idx >= m_first_visible_line + num_visible_lines)
412244d93782SGreg Clayton             m_first_visible_line = pc_idx - non_visible_pc_offset;
412344d93782SGreg Clayton         }
412444d93782SGreg Clayton 
4125b9c1b51eSKate Stone         for (size_t i = 0; i < num_visible_lines; ++i) {
412644d93782SGreg Clayton           const uint32_t inst_idx = m_first_visible_line + i;
412744d93782SGreg Clayton           Instruction *inst = insts.GetInstructionAtIndex(inst_idx).get();
412844d93782SGreg Clayton           if (!inst)
412944d93782SGreg Clayton             break;
413044d93782SGreg Clayton 
4131ec990867SGreg Clayton           const int line_y = m_min_y + i;
4132ec990867SGreg Clayton           window.MoveCursor(1, line_y);
413344d93782SGreg Clayton           const bool is_pc_line = frame_sp && inst_idx == pc_idx;
413444d93782SGreg Clayton           const bool line_is_selected = m_selected_line == inst_idx;
413544d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
413644d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
413744d93782SGreg Clayton           attr_t highlight_attr = 0;
413844d93782SGreg Clayton           attr_t bp_attr = 0;
413944d93782SGreg Clayton           if (is_pc_line)
414044d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
414144d93782SGreg Clayton           else if (line_is_selected)
414244d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
414344d93782SGreg Clayton 
4144b9c1b51eSKate Stone           if (bp_file_addrs.find(inst->GetAddress().GetFileAddress()) !=
4145b9c1b51eSKate Stone               bp_file_addrs.end())
414644d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
414744d93782SGreg Clayton 
414844d93782SGreg Clayton           if (bp_attr)
414944d93782SGreg Clayton             window.AttributeOn(bp_attr);
415044d93782SGreg Clayton 
4151324a1036SSaleem Abdulrasool           window.Printf(" 0x%16.16llx ",
4152b9c1b51eSKate Stone                         static_cast<unsigned long long>(
4153b9c1b51eSKate Stone                             inst->GetAddress().GetLoadAddress(target)));
415444d93782SGreg Clayton 
415544d93782SGreg Clayton           if (bp_attr)
415644d93782SGreg Clayton             window.AttributeOff(bp_attr);
415744d93782SGreg Clayton 
415844d93782SGreg Clayton           window.PutChar(ACS_VLINE);
415944d93782SGreg Clayton           // Mark the line with the PC with a diamond
416044d93782SGreg Clayton           if (is_pc_line)
416144d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
416244d93782SGreg Clayton           else
416344d93782SGreg Clayton             window.PutChar(' ');
416444d93782SGreg Clayton 
416544d93782SGreg Clayton           if (highlight_attr)
416644d93782SGreg Clayton             window.AttributeOn(highlight_attr);
416744d93782SGreg Clayton 
416844d93782SGreg Clayton           const char *mnemonic = inst->GetMnemonic(&exe_ctx);
416944d93782SGreg Clayton           const char *operands = inst->GetOperands(&exe_ctx);
417044d93782SGreg Clayton           const char *comment = inst->GetComment(&exe_ctx);
417144d93782SGreg Clayton 
4172c5dac77aSEugene Zelenko           if (mnemonic != nullptr && mnemonic[0] == '\0')
4173c5dac77aSEugene Zelenko             mnemonic = nullptr;
4174c5dac77aSEugene Zelenko           if (operands != nullptr && operands[0] == '\0')
4175c5dac77aSEugene Zelenko             operands = nullptr;
4176c5dac77aSEugene Zelenko           if (comment != nullptr && comment[0] == '\0')
4177c5dac77aSEugene Zelenko             comment = nullptr;
417844d93782SGreg Clayton 
417944d93782SGreg Clayton           strm.Clear();
418044d93782SGreg Clayton 
4181c5dac77aSEugene Zelenko           if (mnemonic != nullptr && operands != nullptr && comment != nullptr)
418244d93782SGreg Clayton             strm.Printf("%-8s %-25s ; %s", mnemonic, operands, comment);
4183c5dac77aSEugene Zelenko           else if (mnemonic != nullptr && operands != nullptr)
418444d93782SGreg Clayton             strm.Printf("%-8s %s", mnemonic, operands);
4185c5dac77aSEugene Zelenko           else if (mnemonic != nullptr)
418644d93782SGreg Clayton             strm.Printf("%s", mnemonic);
418744d93782SGreg Clayton 
418844d93782SGreg Clayton           int right_pad = 1;
4189c156427dSZachary Turner           window.PutCStringTruncated(strm.GetData(), right_pad);
419044d93782SGreg Clayton 
4191b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4192b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
419344d93782SGreg Clayton             StopInfoSP stop_info_sp;
419444d93782SGreg Clayton             if (thread)
419544d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4196b9c1b51eSKate Stone             if (stop_info_sp) {
419744d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4198b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
419944d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4200ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
420144d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4202b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4203b9c1b51eSKate Stone                 // line_y);
4204b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4205b9c1b51eSKate Stone                               stop_description);
420644d93782SGreg Clayton               }
4207b9c1b51eSKate Stone             } else {
4208ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
420944d93782SGreg Clayton             }
421044d93782SGreg Clayton           }
421144d93782SGreg Clayton           if (highlight_attr)
421244d93782SGreg Clayton             window.AttributeOff(highlight_attr);
421344d93782SGreg Clayton         }
421444d93782SGreg Clayton       }
421544d93782SGreg Clayton     }
421644d93782SGreg Clayton     return true; // Drawing handled
421744d93782SGreg Clayton   }
421844d93782SGreg Clayton 
4219b9c1b51eSKate Stone   size_t GetNumLines() {
422044d93782SGreg Clayton     size_t num_lines = GetNumSourceLines();
422144d93782SGreg Clayton     if (num_lines == 0)
422244d93782SGreg Clayton       num_lines = GetNumDisassemblyLines();
422344d93782SGreg Clayton     return num_lines;
422444d93782SGreg Clayton   }
422544d93782SGreg Clayton 
4226b9c1b51eSKate Stone   size_t GetNumSourceLines() const {
422744d93782SGreg Clayton     if (m_file_sp)
422844d93782SGreg Clayton       return m_file_sp->GetNumLines();
422944d93782SGreg Clayton     return 0;
423044d93782SGreg Clayton   }
4231315b6884SEugene Zelenko 
4232b9c1b51eSKate Stone   size_t GetNumDisassemblyLines() const {
423344d93782SGreg Clayton     if (m_disassembly_sp)
423444d93782SGreg Clayton       return m_disassembly_sp->GetInstructionList().GetSize();
423544d93782SGreg Clayton     return 0;
423644d93782SGreg Clayton   }
423744d93782SGreg Clayton 
4238b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
423944d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
424044d93782SGreg Clayton     const size_t num_lines = GetNumLines();
424144d93782SGreg Clayton 
4242b9c1b51eSKate Stone     switch (c) {
424344d93782SGreg Clayton     case ',':
424444d93782SGreg Clayton     case KEY_PPAGE:
424544d93782SGreg Clayton       // Page up key
42463985c8c6SSaleem Abdulrasool       if (static_cast<uint32_t>(m_first_visible_line) > num_visible_lines)
424744d93782SGreg Clayton         m_first_visible_line -= num_visible_lines;
424844d93782SGreg Clayton       else
424944d93782SGreg Clayton         m_first_visible_line = 0;
425044d93782SGreg Clayton       m_selected_line = m_first_visible_line;
425144d93782SGreg Clayton       return eKeyHandled;
425244d93782SGreg Clayton 
425344d93782SGreg Clayton     case '.':
425444d93782SGreg Clayton     case KEY_NPAGE:
425544d93782SGreg Clayton       // Page down key
425644d93782SGreg Clayton       {
425744d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < num_lines)
425844d93782SGreg Clayton           m_first_visible_line += num_visible_lines;
425944d93782SGreg Clayton         else if (num_lines < num_visible_lines)
426044d93782SGreg Clayton           m_first_visible_line = 0;
426144d93782SGreg Clayton         else
426244d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
426344d93782SGreg Clayton         m_selected_line = m_first_visible_line;
426444d93782SGreg Clayton       }
426544d93782SGreg Clayton       return eKeyHandled;
426644d93782SGreg Clayton 
426744d93782SGreg Clayton     case KEY_UP:
4268b9c1b51eSKate Stone       if (m_selected_line > 0) {
426944d93782SGreg Clayton         m_selected_line--;
42703985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > m_selected_line)
427144d93782SGreg Clayton           m_first_visible_line = m_selected_line;
427244d93782SGreg Clayton       }
427344d93782SGreg Clayton       return eKeyHandled;
427444d93782SGreg Clayton 
427544d93782SGreg Clayton     case KEY_DOWN:
4276b9c1b51eSKate Stone       if (m_selected_line + 1 < num_lines) {
427744d93782SGreg Clayton         m_selected_line++;
427844d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < m_selected_line)
427944d93782SGreg Clayton           m_first_visible_line++;
428044d93782SGreg Clayton       }
428144d93782SGreg Clayton       return eKeyHandled;
428244d93782SGreg Clayton 
428344d93782SGreg Clayton     case '\r':
428444d93782SGreg Clayton     case '\n':
428544d93782SGreg Clayton     case KEY_ENTER:
428644d93782SGreg Clayton       // Set a breakpoint and run to the line using a one shot breakpoint
4287b9c1b51eSKate Stone       if (GetNumSourceLines() > 0) {
4288b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4289b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4290b9c1b51eSKate Stone         if (exe_ctx.HasProcessScope() && exe_ctx.GetProcessRef().IsAlive()) {
4291b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4292b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
429344d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4294b9c1b51eSKate Stone               m_selected_line +
4295b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4296431b1584SAdrian Prantl               0,     // Unspecified column.
42972411167fSJim Ingham               0,     // No offset
429844d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
429944d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
430044d93782SGreg Clayton               false,               // internal
4301055ad9beSIlia K               false,               // request_hardware
4302055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
430344d93782SGreg Clayton           // Make breakpoint one shot
430444d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
430544d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
430644d93782SGreg Clayton         }
4307b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4308b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4309b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4310b9c1b51eSKate Stone                                       .get();
4311b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4312b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4313b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
431444d93782SGreg Clayton           Address addr = inst->GetAddress();
4315b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4316b9c1b51eSKate Stone               addr,   // lldb_private::Address
431744d93782SGreg Clayton               false,  // internal
431844d93782SGreg Clayton               false); // request_hardware
431944d93782SGreg Clayton           // Make breakpoint one shot
432044d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
432144d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
432244d93782SGreg Clayton         }
432344d93782SGreg Clayton       }
432444d93782SGreg Clayton       return eKeyHandled;
432544d93782SGreg Clayton 
432644d93782SGreg Clayton     case 'b': // 'b' == toggle breakpoint on currently selected line
4327b9c1b51eSKate Stone       if (m_selected_line < GetNumSourceLines()) {
4328b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4329b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4330b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
4331b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4332b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
433344d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4334b9c1b51eSKate Stone               m_selected_line +
4335b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4336431b1584SAdrian Prantl               0,     // No column specified.
43372411167fSJim Ingham               0,     // No offset
433844d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
433944d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
434044d93782SGreg Clayton               false,               // internal
4341055ad9beSIlia K               false,               // request_hardware
4342055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
434344d93782SGreg Clayton         }
4344b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4345b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4346b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4347b9c1b51eSKate Stone                                       .get();
4348b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4349b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4350b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
435144d93782SGreg Clayton           Address addr = inst->GetAddress();
4352b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4353b9c1b51eSKate Stone               addr,   // lldb_private::Address
435444d93782SGreg Clayton               false,  // internal
435544d93782SGreg Clayton               false); // request_hardware
435644d93782SGreg Clayton         }
435744d93782SGreg Clayton       }
435844d93782SGreg Clayton       return eKeyHandled;
435944d93782SGreg Clayton 
436044d93782SGreg Clayton     case 'd': // 'd' == detach and let run
436144d93782SGreg Clayton     case 'D': // 'D' == detach and keep stopped
436244d93782SGreg Clayton     {
4363b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4364b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
436544d93782SGreg Clayton       if (exe_ctx.HasProcessScope())
436644d93782SGreg Clayton         exe_ctx.GetProcessRef().Detach(c == 'D');
436744d93782SGreg Clayton     }
436844d93782SGreg Clayton       return eKeyHandled;
436944d93782SGreg Clayton 
437044d93782SGreg Clayton     case 'k':
437144d93782SGreg Clayton       // 'k' == kill
437244d93782SGreg Clayton       {
4373b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4374b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
437544d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
4376ede3193bSJason Molenda           exe_ctx.GetProcessRef().Destroy(false);
437744d93782SGreg Clayton       }
437844d93782SGreg Clayton       return eKeyHandled;
437944d93782SGreg Clayton 
438044d93782SGreg Clayton     case 'c':
438144d93782SGreg Clayton       // 'c' == continue
438244d93782SGreg Clayton       {
4383b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4384b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
438544d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
438644d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
438744d93782SGreg Clayton       }
438844d93782SGreg Clayton       return eKeyHandled;
438944d93782SGreg Clayton 
439044d93782SGreg Clayton     case 'o':
439144d93782SGreg Clayton       // 'o' == step out
439244d93782SGreg Clayton       {
4393b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4394b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4395b9c1b51eSKate Stone         if (exe_ctx.HasThreadScope() &&
4396b9c1b51eSKate Stone             StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
439744d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
439844d93782SGreg Clayton         }
439944d93782SGreg Clayton       }
440044d93782SGreg Clayton       return eKeyHandled;
4401315b6884SEugene Zelenko 
440244d93782SGreg Clayton     case 'n': // 'n' == step over
440344d93782SGreg Clayton     case 'N': // 'N' == step over instruction
440444d93782SGreg Clayton     {
4405b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4406b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
4407b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope() &&
4408b9c1b51eSKate Stone           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
440944d93782SGreg Clayton         bool source_step = (c == 'n');
441044d93782SGreg Clayton         exe_ctx.GetThreadRef().StepOver(source_step);
441144d93782SGreg Clayton       }
441244d93782SGreg Clayton     }
441344d93782SGreg Clayton       return eKeyHandled;
4414315b6884SEugene Zelenko 
441544d93782SGreg Clayton     case 's': // 's' == step into
441644d93782SGreg Clayton     case 'S': // 'S' == step into instruction
441744d93782SGreg Clayton     {
4418b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4419b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
4420b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope() &&
4421b9c1b51eSKate Stone           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
442244d93782SGreg Clayton         bool source_step = (c == 's');
44234b4b2478SJim Ingham         exe_ctx.GetThreadRef().StepIn(source_step);
442444d93782SGreg Clayton       }
442544d93782SGreg Clayton     }
442644d93782SGreg Clayton       return eKeyHandled;
442744d93782SGreg Clayton 
442844d93782SGreg Clayton     case 'h':
442944d93782SGreg Clayton       window.CreateHelpSubwindow();
443044d93782SGreg Clayton       return eKeyHandled;
443144d93782SGreg Clayton 
443244d93782SGreg Clayton     default:
443344d93782SGreg Clayton       break;
443444d93782SGreg Clayton     }
443544d93782SGreg Clayton     return eKeyNotHandled;
443644d93782SGreg Clayton   }
443744d93782SGreg Clayton 
443844d93782SGreg Clayton protected:
443944d93782SGreg Clayton   typedef std::set<uint32_t> BreakpointLines;
444044d93782SGreg Clayton   typedef std::set<lldb::addr_t> BreakpointAddrs;
444144d93782SGreg Clayton 
444244d93782SGreg Clayton   Debugger &m_debugger;
444344d93782SGreg Clayton   SymbolContext m_sc;
444444d93782SGreg Clayton   SourceManager::FileSP m_file_sp;
444544d93782SGreg Clayton   SymbolContextScope *m_disassembly_scope;
444644d93782SGreg Clayton   lldb::DisassemblerSP m_disassembly_sp;
444744d93782SGreg Clayton   AddressRange m_disassembly_range;
4448ec990867SGreg Clayton   StreamString m_title;
444944d93782SGreg Clayton   lldb::user_id_t m_tid;
445044d93782SGreg Clayton   int m_line_width;
445144d93782SGreg Clayton   uint32_t m_selected_line; // The selected line
445244d93782SGreg Clayton   uint32_t m_pc_line;       // The line with the PC
445344d93782SGreg Clayton   uint32_t m_stop_id;
445444d93782SGreg Clayton   uint32_t m_frame_idx;
445544d93782SGreg Clayton   int m_first_visible_line;
445644d93782SGreg Clayton   int m_min_x;
445744d93782SGreg Clayton   int m_min_y;
445844d93782SGreg Clayton   int m_max_x;
445944d93782SGreg Clayton   int m_max_y;
446044d93782SGreg Clayton };
446144d93782SGreg Clayton 
446244d93782SGreg Clayton DisplayOptions ValueObjectListDelegate::g_options = {true};
446344d93782SGreg Clayton 
4464b9c1b51eSKate Stone IOHandlerCursesGUI::IOHandlerCursesGUI(Debugger &debugger)
4465b9c1b51eSKate Stone     : IOHandler(debugger, IOHandler::Type::Curses) {}
446644d93782SGreg Clayton 
4467b9c1b51eSKate Stone void IOHandlerCursesGUI::Activate() {
446844d93782SGreg Clayton   IOHandler::Activate();
4469b9c1b51eSKate Stone   if (!m_app_ap) {
447044d93782SGreg Clayton     m_app_ap.reset(new Application(GetInputFILE(), GetOutputFILE()));
447144d93782SGreg Clayton 
447244d93782SGreg Clayton     // This is both a window and a menu delegate
4473b9c1b51eSKate Stone     std::shared_ptr<ApplicationDelegate> app_delegate_sp(
4474b9c1b51eSKate Stone         new ApplicationDelegate(*m_app_ap, m_debugger));
447544d93782SGreg Clayton 
4476b9c1b51eSKate Stone     MenuDelegateSP app_menu_delegate_sp =
4477b9c1b51eSKate Stone         std::static_pointer_cast<MenuDelegate>(app_delegate_sp);
4478b9c1b51eSKate Stone     MenuSP lldb_menu_sp(
4479b9c1b51eSKate Stone         new Menu("LLDB", "F1", KEY_F(1), ApplicationDelegate::eMenuID_LLDB));
4480b9c1b51eSKate Stone     MenuSP exit_menuitem_sp(
4481b9c1b51eSKate Stone         new Menu("Exit", nullptr, 'x', ApplicationDelegate::eMenuID_LLDBExit));
448244d93782SGreg Clayton     exit_menuitem_sp->SetCannedResult(MenuActionResult::Quit);
4483b9c1b51eSKate Stone     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(
4484b9c1b51eSKate Stone         "About LLDB", nullptr, 'a', ApplicationDelegate::eMenuID_LLDBAbout)));
448544d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
448644d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(exit_menuitem_sp);
448744d93782SGreg Clayton 
4488b9c1b51eSKate Stone     MenuSP target_menu_sp(new Menu("Target", "F2", KEY_F(2),
4489b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Target));
4490b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4491b9c1b51eSKate Stone         "Create", nullptr, 'c', ApplicationDelegate::eMenuID_TargetCreate)));
4492b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4493b9c1b51eSKate Stone         "Delete", nullptr, 'd', ApplicationDelegate::eMenuID_TargetDelete)));
449444d93782SGreg Clayton 
4495b9c1b51eSKate Stone     MenuSP process_menu_sp(new Menu("Process", "F3", KEY_F(3),
4496b9c1b51eSKate Stone                                     ApplicationDelegate::eMenuID_Process));
4497b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4498b9c1b51eSKate Stone         "Attach", nullptr, 'a', ApplicationDelegate::eMenuID_ProcessAttach)));
4499b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4500b9c1b51eSKate Stone         "Detach", nullptr, 'd', ApplicationDelegate::eMenuID_ProcessDetach)));
4501b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4502b9c1b51eSKate Stone         "Launch", nullptr, 'l', ApplicationDelegate::eMenuID_ProcessLaunch)));
450344d93782SGreg Clayton     process_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
4504b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(
4505b9c1b51eSKate Stone         MenuSP(new Menu("Continue", nullptr, 'c',
4506b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ProcessContinue)));
4507b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4508b9c1b51eSKate Stone         "Halt", nullptr, 'h', ApplicationDelegate::eMenuID_ProcessHalt)));
4509b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4510b9c1b51eSKate Stone         "Kill", nullptr, 'k', ApplicationDelegate::eMenuID_ProcessKill)));
451144d93782SGreg Clayton 
4512b9c1b51eSKate Stone     MenuSP thread_menu_sp(new Menu("Thread", "F4", KEY_F(4),
4513b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Thread));
4514b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4515b9c1b51eSKate Stone         "Step In", nullptr, 'i', ApplicationDelegate::eMenuID_ThreadStepIn)));
4516b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(
4517b9c1b51eSKate Stone         MenuSP(new Menu("Step Over", nullptr, 'v',
4518b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ThreadStepOver)));
4519b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4520b9c1b51eSKate Stone         "Step Out", nullptr, 'o', ApplicationDelegate::eMenuID_ThreadStepOut)));
452144d93782SGreg Clayton 
4522b9c1b51eSKate Stone     MenuSP view_menu_sp(
4523b9c1b51eSKate Stone         new Menu("View", "F5", KEY_F(5), ApplicationDelegate::eMenuID_View));
4524b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4525b9c1b51eSKate Stone         MenuSP(new Menu("Backtrace", nullptr, 'b',
4526b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewBacktrace)));
4527b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4528b9c1b51eSKate Stone         MenuSP(new Menu("Registers", nullptr, 'r',
4529b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewRegisters)));
4530b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(MenuSP(new Menu(
4531b9c1b51eSKate Stone         "Source", nullptr, 's', ApplicationDelegate::eMenuID_ViewSource)));
4532b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4533b9c1b51eSKate Stone         MenuSP(new Menu("Variables", nullptr, 'v',
4534b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewVariables)));
453544d93782SGreg Clayton 
4536b9c1b51eSKate Stone     MenuSP help_menu_sp(
4537b9c1b51eSKate Stone         new Menu("Help", "F6", KEY_F(6), ApplicationDelegate::eMenuID_Help));
4538b9c1b51eSKate Stone     help_menu_sp->AddSubmenu(MenuSP(new Menu(
4539b9c1b51eSKate Stone         "GUI Help", nullptr, 'g', ApplicationDelegate::eMenuID_HelpGUIHelp)));
454044d93782SGreg Clayton 
454144d93782SGreg Clayton     m_app_ap->Initialize();
454244d93782SGreg Clayton     WindowSP &main_window_sp = m_app_ap->GetMainWindow();
454344d93782SGreg Clayton 
454444d93782SGreg Clayton     MenuSP menubar_sp(new Menu(Menu::Type::Bar));
454544d93782SGreg Clayton     menubar_sp->AddSubmenu(lldb_menu_sp);
454644d93782SGreg Clayton     menubar_sp->AddSubmenu(target_menu_sp);
454744d93782SGreg Clayton     menubar_sp->AddSubmenu(process_menu_sp);
454844d93782SGreg Clayton     menubar_sp->AddSubmenu(thread_menu_sp);
454944d93782SGreg Clayton     menubar_sp->AddSubmenu(view_menu_sp);
455044d93782SGreg Clayton     menubar_sp->AddSubmenu(help_menu_sp);
455144d93782SGreg Clayton     menubar_sp->SetDelegate(app_menu_delegate_sp);
455244d93782SGreg Clayton 
455344d93782SGreg Clayton     Rect content_bounds = main_window_sp->GetFrame();
455444d93782SGreg Clayton     Rect menubar_bounds = content_bounds.MakeMenuBar();
455544d93782SGreg Clayton     Rect status_bounds = content_bounds.MakeStatusBar();
455644d93782SGreg Clayton     Rect source_bounds;
455744d93782SGreg Clayton     Rect variables_bounds;
455844d93782SGreg Clayton     Rect threads_bounds;
455944d93782SGreg Clayton     Rect source_variables_bounds;
4560b9c1b51eSKate Stone     content_bounds.VerticalSplitPercentage(0.80, source_variables_bounds,
4561b9c1b51eSKate Stone                                            threads_bounds);
4562b9c1b51eSKate Stone     source_variables_bounds.HorizontalSplitPercentage(0.70, source_bounds,
4563b9c1b51eSKate Stone                                                       variables_bounds);
456444d93782SGreg Clayton 
4565b9c1b51eSKate Stone     WindowSP menubar_window_sp =
4566b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Menubar", menubar_bounds, false);
456705097246SAdrian Prantl     // Let the menubar get keys if the active window doesn't handle the keys
456805097246SAdrian Prantl     // that are typed so it can respond to menubar key presses.
4569b9c1b51eSKate Stone     menubar_window_sp->SetCanBeActive(
4570b9c1b51eSKate Stone         false); // Don't let the menubar become the active window
457144d93782SGreg Clayton     menubar_window_sp->SetDelegate(menubar_sp);
457244d93782SGreg Clayton 
4573b9c1b51eSKate Stone     WindowSP source_window_sp(
4574b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Source", source_bounds, true));
4575b9c1b51eSKate Stone     WindowSP variables_window_sp(
4576b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Variables", variables_bounds, false));
4577b9c1b51eSKate Stone     WindowSP threads_window_sp(
4578b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Threads", threads_bounds, false));
4579b9c1b51eSKate Stone     WindowSP status_window_sp(
45806bb7e21fSPavel Labath         main_window_sp->CreateSubWindow("Status", status_bounds, false));
4581b9c1b51eSKate Stone     status_window_sp->SetCanBeActive(
4582b9c1b51eSKate Stone         false); // Don't let the status bar become the active window
4583b9c1b51eSKate Stone     main_window_sp->SetDelegate(
4584b9c1b51eSKate Stone         std::static_pointer_cast<WindowDelegate>(app_delegate_sp));
4585b9c1b51eSKate Stone     source_window_sp->SetDelegate(
4586b9c1b51eSKate Stone         WindowDelegateSP(new SourceFileWindowDelegate(m_debugger)));
4587b9c1b51eSKate Stone     variables_window_sp->SetDelegate(
4588b9c1b51eSKate Stone         WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
4589ec990867SGreg Clayton     TreeDelegateSP thread_delegate_sp(new ThreadsTreeDelegate(m_debugger));
4590b9c1b51eSKate Stone     threads_window_sp->SetDelegate(WindowDelegateSP(
4591b9c1b51eSKate Stone         new TreeWindowDelegate(m_debugger, thread_delegate_sp)));
4592b9c1b51eSKate Stone     status_window_sp->SetDelegate(
4593b9c1b51eSKate Stone         WindowDelegateSP(new StatusBarWindowDelegate(m_debugger)));
459444d93782SGreg Clayton 
45955fdb09bbSGreg Clayton     // Show the main help window once the first time the curses GUI is launched
45965fdb09bbSGreg Clayton     static bool g_showed_help = false;
4597b9c1b51eSKate Stone     if (!g_showed_help) {
45985fdb09bbSGreg Clayton       g_showed_help = true;
45995fdb09bbSGreg Clayton       main_window_sp->CreateHelpSubwindow();
46005fdb09bbSGreg Clayton     }
46015fdb09bbSGreg Clayton 
460244d93782SGreg Clayton     init_pair(1, COLOR_WHITE, COLOR_BLUE);
460344d93782SGreg Clayton     init_pair(2, COLOR_BLACK, COLOR_WHITE);
460444d93782SGreg Clayton     init_pair(3, COLOR_MAGENTA, COLOR_WHITE);
460544d93782SGreg Clayton     init_pair(4, COLOR_MAGENTA, COLOR_BLACK);
460644d93782SGreg Clayton     init_pair(5, COLOR_RED, COLOR_BLACK);
460744d93782SGreg Clayton   }
460844d93782SGreg Clayton }
460944d93782SGreg Clayton 
4610b9c1b51eSKate Stone void IOHandlerCursesGUI::Deactivate() { m_app_ap->Terminate(); }
461144d93782SGreg Clayton 
4612b9c1b51eSKate Stone void IOHandlerCursesGUI::Run() {
461344d93782SGreg Clayton   m_app_ap->Run(m_debugger);
461444d93782SGreg Clayton   SetIsDone(true);
461544d93782SGreg Clayton }
461644d93782SGreg Clayton 
4617315b6884SEugene Zelenko IOHandlerCursesGUI::~IOHandlerCursesGUI() = default;
461844d93782SGreg Clayton 
4619b9c1b51eSKate Stone void IOHandlerCursesGUI::Cancel() {}
462044d93782SGreg Clayton 
4621b9c1b51eSKate Stone bool IOHandlerCursesGUI::Interrupt() { return false; }
462244d93782SGreg Clayton 
4623b9c1b51eSKate Stone void IOHandlerCursesGUI::GotEOF() {}
462444d93782SGreg Clayton 
4625315b6884SEugene Zelenko #endif // LLDB_DISABLE_CURSES
4626