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 
55fab31220STed Woodward #ifdef _MSC_VER
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;
7344d93782SGreg Clayton 
74b9c1b51eSKate Stone IOHandler::IOHandler(Debugger &debugger, IOHandler::Type type)
75b9c1b51eSKate Stone     : IOHandler(debugger, type,
7644d93782SGreg Clayton                 StreamFileSP(), // Adopt STDIN from top input reader
7744d93782SGreg Clayton                 StreamFileSP(), // Adopt STDOUT from top input reader
78340b0309SGreg Clayton                 StreamFileSP(), // Adopt STDERR from top input reader
79d77c2e09SJonas Devlieghere                 0,              // Flags
80d77c2e09SJonas Devlieghere                 nullptr         // Shadow file recorder
81d77c2e09SJonas Devlieghere       ) {}
8244d93782SGreg Clayton 
83b9c1b51eSKate Stone IOHandler::IOHandler(Debugger &debugger, IOHandler::Type type,
8444d93782SGreg Clayton                      const lldb::StreamFileSP &input_sp,
8544d93782SGreg Clayton                      const lldb::StreamFileSP &output_sp,
86d77c2e09SJonas Devlieghere                      const lldb::StreamFileSP &error_sp, uint32_t flags,
87d77c2e09SJonas Devlieghere                      repro::DataRecorder *data_recorder)
88b9c1b51eSKate Stone     : m_debugger(debugger), m_input_sp(input_sp), m_output_sp(output_sp),
89d77c2e09SJonas Devlieghere       m_error_sp(error_sp), m_data_recorder(data_recorder), m_popped(false),
90d77c2e09SJonas Devlieghere       m_flags(flags), m_type(type), m_user_data(nullptr), m_done(false),
91d77c2e09SJonas Devlieghere       m_active(false) {
9244d93782SGreg Clayton   // If any files are not specified, then adopt them from the top input reader.
9344d93782SGreg Clayton   if (!m_input_sp || !m_output_sp || !m_error_sp)
94b9c1b51eSKate Stone     debugger.AdoptTopIOHandlerFilesIfInvalid(m_input_sp, m_output_sp,
9544d93782SGreg Clayton                                              m_error_sp);
9644d93782SGreg Clayton }
9744d93782SGreg Clayton 
98315b6884SEugene Zelenko IOHandler::~IOHandler() = default;
9944d93782SGreg Clayton 
100b9c1b51eSKate Stone int IOHandler::GetInputFD() {
101c5dac77aSEugene Zelenko   return (m_input_sp ? m_input_sp->GetFile().GetDescriptor() : -1);
10244d93782SGreg Clayton }
10344d93782SGreg Clayton 
104b9c1b51eSKate Stone int IOHandler::GetOutputFD() {
105c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetDescriptor() : -1);
10644d93782SGreg Clayton }
10744d93782SGreg Clayton 
108b9c1b51eSKate Stone int IOHandler::GetErrorFD() {
109c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetDescriptor() : -1);
11044d93782SGreg Clayton }
11144d93782SGreg Clayton 
112b9c1b51eSKate Stone FILE *IOHandler::GetInputFILE() {
113c5dac77aSEugene Zelenko   return (m_input_sp ? m_input_sp->GetFile().GetStream() : nullptr);
11444d93782SGreg Clayton }
11544d93782SGreg Clayton 
116b9c1b51eSKate Stone FILE *IOHandler::GetOutputFILE() {
117c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetStream() : nullptr);
11844d93782SGreg Clayton }
11944d93782SGreg Clayton 
120b9c1b51eSKate Stone FILE *IOHandler::GetErrorFILE() {
121c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetStream() : nullptr);
12244d93782SGreg Clayton }
12344d93782SGreg Clayton 
124b9c1b51eSKate Stone StreamFileSP &IOHandler::GetInputStreamFile() { return m_input_sp; }
12544d93782SGreg Clayton 
126b9c1b51eSKate Stone StreamFileSP &IOHandler::GetOutputStreamFile() { return m_output_sp; }
12744d93782SGreg Clayton 
128b9c1b51eSKate Stone StreamFileSP &IOHandler::GetErrorStreamFile() { return m_error_sp; }
12944d93782SGreg Clayton 
130b9c1b51eSKate Stone bool IOHandler::GetIsInteractive() {
131340b0309SGreg Clayton   return GetInputStreamFile()->GetFile().GetIsInteractive();
132340b0309SGreg Clayton }
133340b0309SGreg Clayton 
134b9c1b51eSKate Stone bool IOHandler::GetIsRealTerminal() {
135340b0309SGreg Clayton   return GetInputStreamFile()->GetFile().GetIsRealTerminal();
136340b0309SGreg Clayton }
13744d93782SGreg Clayton 
138b9c1b51eSKate Stone void IOHandler::SetPopped(bool b) { m_popped.SetValue(b, eBroadcastOnChange); }
139e30f11d9SKate Stone 
140b9c1b51eSKate Stone void IOHandler::WaitForPop() { m_popped.WaitForValueEqualTo(true); }
141e30f11d9SKate Stone 
142b9c1b51eSKate Stone void IOHandlerStack::PrintAsync(Stream *stream, const char *s, size_t len) {
143b9c1b51eSKate Stone   if (stream) {
14416ff8604SSaleem Abdulrasool     std::lock_guard<std::recursive_mutex> guard(m_mutex);
1454446487dSPavel Labath     if (m_top)
1464446487dSPavel Labath       m_top->PrintAsync(stream, s, len);
1474446487dSPavel Labath   }
1484446487dSPavel Labath }
1494446487dSPavel Labath 
1507a120c8bSZachary Turner IOHandlerConfirm::IOHandlerConfirm(Debugger &debugger, llvm::StringRef prompt,
151b9c1b51eSKate Stone                                    bool default_response)
152b9c1b51eSKate Stone     : IOHandlerEditline(
153b9c1b51eSKate Stone           debugger, IOHandler::Type::Confirm,
154c5dac77aSEugene Zelenko           nullptr, // nullptr editline_name means no history loaded/saved
155514d8cd8SZachary Turner           llvm::StringRef(), // No prompt
156514d8cd8SZachary Turner           llvm::StringRef(), // No continuation prompt
15744d93782SGreg Clayton           false,             // Multi-line
158e30f11d9SKate Stone           false, // Don't colorize the prompt (i.e. the confirm message.)
159d77c2e09SJonas Devlieghere           0, *this, nullptr),
160b9c1b51eSKate Stone       m_default_response(default_response), m_user_response(default_response) {
16144d93782SGreg Clayton   StreamString prompt_stream;
16244d93782SGreg Clayton   prompt_stream.PutCString(prompt);
16344d93782SGreg Clayton   if (m_default_response)
16444d93782SGreg Clayton     prompt_stream.Printf(": [Y/n] ");
16544d93782SGreg Clayton   else
16644d93782SGreg Clayton     prompt_stream.Printf(": [y/N] ");
16744d93782SGreg Clayton 
168514d8cd8SZachary Turner   SetPrompt(prompt_stream.GetString());
16944d93782SGreg Clayton }
17044d93782SGreg Clayton 
171315b6884SEugene Zelenko IOHandlerConfirm::~IOHandlerConfirm() = default;
17244d93782SGreg Clayton 
1737f88829cSRaphael Isemann int IOHandlerConfirm::IOHandlerComplete(
1747f88829cSRaphael Isemann     IOHandler &io_handler, const char *current_line, const char *cursor,
1757f88829cSRaphael Isemann     const char *last_char, int skip_first_n_matches, int max_matches,
1767f88829cSRaphael Isemann     StringList &matches, StringList &descriptions) {
177b9c1b51eSKate Stone   if (current_line == cursor) {
178b9c1b51eSKate Stone     if (m_default_response) {
17944d93782SGreg Clayton       matches.AppendString("y");
180b9c1b51eSKate Stone     } else {
18144d93782SGreg Clayton       matches.AppendString("n");
18244d93782SGreg Clayton     }
18344d93782SGreg Clayton   }
18444d93782SGreg Clayton   return matches.GetSize();
18544d93782SGreg Clayton }
18644d93782SGreg Clayton 
187b9c1b51eSKate Stone void IOHandlerConfirm::IOHandlerInputComplete(IOHandler &io_handler,
188b9c1b51eSKate Stone                                               std::string &line) {
189b9c1b51eSKate Stone   if (line.empty()) {
19044d93782SGreg Clayton     // User just hit enter, set the response to the default
19144d93782SGreg Clayton     m_user_response = m_default_response;
19244d93782SGreg Clayton     io_handler.SetIsDone(true);
19344d93782SGreg Clayton     return;
19444d93782SGreg Clayton   }
19544d93782SGreg Clayton 
196b9c1b51eSKate Stone   if (line.size() == 1) {
197b9c1b51eSKate Stone     switch (line[0]) {
19844d93782SGreg Clayton     case 'y':
19944d93782SGreg Clayton     case 'Y':
20044d93782SGreg Clayton       m_user_response = true;
20144d93782SGreg Clayton       io_handler.SetIsDone(true);
20244d93782SGreg Clayton       return;
20344d93782SGreg Clayton     case 'n':
20444d93782SGreg Clayton     case 'N':
20544d93782SGreg Clayton       m_user_response = false;
20644d93782SGreg Clayton       io_handler.SetIsDone(true);
20744d93782SGreg Clayton       return;
20844d93782SGreg Clayton     default:
20944d93782SGreg Clayton       break;
21044d93782SGreg Clayton     }
21144d93782SGreg Clayton   }
21244d93782SGreg Clayton 
213b9c1b51eSKate Stone   if (line == "yes" || line == "YES" || line == "Yes") {
21444d93782SGreg Clayton     m_user_response = true;
21544d93782SGreg Clayton     io_handler.SetIsDone(true);
216b9c1b51eSKate Stone   } else if (line == "no" || line == "NO" || line == "No") {
21744d93782SGreg Clayton     m_user_response = false;
21844d93782SGreg Clayton     io_handler.SetIsDone(true);
21944d93782SGreg Clayton   }
22044d93782SGreg Clayton }
22144d93782SGreg Clayton 
2227f88829cSRaphael Isemann int IOHandlerDelegate::IOHandlerComplete(
2237f88829cSRaphael Isemann     IOHandler &io_handler, const char *current_line, const char *cursor,
2247f88829cSRaphael Isemann     const char *last_char, int skip_first_n_matches, int max_matches,
2257f88829cSRaphael Isemann     StringList &matches, StringList &descriptions) {
226b9c1b51eSKate Stone   switch (m_completion) {
22744d93782SGreg Clayton   case Completion::None:
22844d93782SGreg Clayton     break;
22944d93782SGreg Clayton 
23044d93782SGreg Clayton   case Completion::LLDBCommand:
231b9c1b51eSKate Stone     return io_handler.GetDebugger().GetCommandInterpreter().HandleCompletion(
232b9c1b51eSKate Stone         current_line, cursor, last_char, skip_first_n_matches, max_matches,
2337f88829cSRaphael Isemann         matches, descriptions);
234b9c1b51eSKate Stone   case Completion::Expression: {
2357f88829cSRaphael Isemann     CompletionResult result;
236*e010f6baSRaphael Isemann     CompletionRequest request(current_line, cursor - current_line,
2377f88829cSRaphael Isemann                               skip_first_n_matches, max_matches, result);
238b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
239b9c1b51eSKate Stone         io_handler.GetDebugger().GetCommandInterpreter(),
240a2e76c0bSRaphael Isemann         CommandCompletions::eVariablePathCompletion, request, nullptr);
2417f88829cSRaphael Isemann     result.GetMatches(matches);
2427f88829cSRaphael Isemann     result.GetDescriptions(descriptions);
24344d93782SGreg Clayton 
2441a6d7ab5SRaphael Isemann     size_t num_matches = request.GetNumberOfMatches();
245b9c1b51eSKate Stone     if (num_matches > 0) {
24644d93782SGreg Clayton       std::string common_prefix;
2471a6d7ab5SRaphael Isemann       matches.LongestCommonPrefix(common_prefix);
248a2e76c0bSRaphael Isemann       const size_t partial_name_len = request.GetCursorArgumentPrefix().size();
24944d93782SGreg Clayton 
25005097246SAdrian Prantl       // If we matched a unique single command, add a space... Only do this if
25105097246SAdrian Prantl       // the completer told us this was a complete word, however...
252a2e76c0bSRaphael Isemann       if (num_matches == 1 && request.GetWordComplete()) {
25344d93782SGreg Clayton         common_prefix.push_back(' ');
25444d93782SGreg Clayton       }
25544d93782SGreg Clayton       common_prefix.erase(0, partial_name_len);
25644d93782SGreg Clayton       matches.InsertStringAtIndex(0, std::move(common_prefix));
25744d93782SGreg Clayton     }
25844d93782SGreg Clayton     return num_matches;
259b9c1b51eSKate Stone   } break;
26044d93782SGreg Clayton   }
26144d93782SGreg Clayton 
26244d93782SGreg Clayton   return 0;
26344d93782SGreg Clayton }
26444d93782SGreg Clayton 
265b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
266b9c1b51eSKate Stone     Debugger &debugger, IOHandler::Type type,
26744d93782SGreg Clayton     const char *editline_name, // Used for saving history files
268514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
269514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
270d77c2e09SJonas Devlieghere     IOHandlerDelegate &delegate, repro::DataRecorder *data_recorder)
271b9c1b51eSKate Stone     : IOHandlerEditline(debugger, type,
27244d93782SGreg Clayton                         StreamFileSP(), // Inherit input from top input reader
27344d93782SGreg Clayton                         StreamFileSP(), // Inherit output from top input reader
27444d93782SGreg Clayton                         StreamFileSP(), // Inherit error from top input reader
275340b0309SGreg Clayton                         0,              // Flags
27644d93782SGreg Clayton                         editline_name,  // Used for saving history files
277b9c1b51eSKate Stone                         prompt, continuation_prompt, multi_line, color_prompts,
278d77c2e09SJonas Devlieghere                         line_number_start, delegate, data_recorder) {}
27944d93782SGreg Clayton 
280b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
281b9c1b51eSKate Stone     Debugger &debugger, IOHandler::Type type,
282b9c1b51eSKate Stone     const lldb::StreamFileSP &input_sp, const lldb::StreamFileSP &output_sp,
283b9c1b51eSKate Stone     const lldb::StreamFileSP &error_sp, uint32_t flags,
28444d93782SGreg Clayton     const char *editline_name, // Used for saving history files
285514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
286514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
287d77c2e09SJonas Devlieghere     IOHandlerDelegate &delegate, repro::DataRecorder *data_recorder)
288d77c2e09SJonas Devlieghere     : IOHandler(debugger, type, input_sp, output_sp, error_sp, flags,
289d77c2e09SJonas Devlieghere                 data_recorder),
290cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
291d5b44036SJonas Devlieghere       m_editline_up(),
292cacde7dfSTodd Fiala #endif
293b9c1b51eSKate Stone       m_delegate(delegate), m_prompt(), m_continuation_prompt(),
294b9c1b51eSKate Stone       m_current_lines_ptr(nullptr), m_base_line_number(line_number_start),
295b9c1b51eSKate Stone       m_curr_line_idx(UINT32_MAX), m_multi_line(multi_line),
296b9c1b51eSKate Stone       m_color_prompts(color_prompts), m_interrupt_exits(true),
297b9c1b51eSKate Stone       m_editing(false) {
29844d93782SGreg Clayton   SetPrompt(prompt);
29944d93782SGreg Clayton 
300cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
301914b8d98SDeepak Panickal   bool use_editline = false;
302340b0309SGreg Clayton 
303340b0309SGreg Clayton   use_editline = m_input_sp->GetFile().GetIsRealTerminal();
30444d93782SGreg Clayton 
305b9c1b51eSKate Stone   if (use_editline) {
306d5b44036SJonas Devlieghere     m_editline_up.reset(new Editline(editline_name, GetInputFILE(),
307b9c1b51eSKate Stone                                      GetOutputFILE(), GetErrorFILE(),
308e30f11d9SKate Stone                                      m_color_prompts));
309d5b44036SJonas Devlieghere     m_editline_up->SetIsInputCompleteCallback(IsInputCompleteCallback, this);
310d5b44036SJonas Devlieghere     m_editline_up->SetAutoCompleteCallback(AutoCompleteCallback, this);
311e30f11d9SKate Stone     // See if the delegate supports fixing indentation
312e30f11d9SKate Stone     const char *indent_chars = delegate.IOHandlerGetFixIndentationCharacters();
313b9c1b51eSKate Stone     if (indent_chars) {
314b9c1b51eSKate Stone       // The delegate does support indentation, hook it up so when any
31505097246SAdrian Prantl       // indentation character is typed, the delegate gets a chance to fix it
316d5b44036SJonas Devlieghere       m_editline_up->SetFixIndentationCallback(FixIndentationCallback, this,
317b9c1b51eSKate Stone                                                indent_chars);
318e30f11d9SKate Stone     }
31944d93782SGreg Clayton   }
320cacde7dfSTodd Fiala #endif
321e30f11d9SKate Stone   SetBaseLineNumber(m_base_line_number);
322514d8cd8SZachary Turner   SetPrompt(prompt);
323e30f11d9SKate Stone   SetContinuationPrompt(continuation_prompt);
32444d93782SGreg Clayton }
32544d93782SGreg Clayton 
326b9c1b51eSKate Stone IOHandlerEditline::~IOHandlerEditline() {
327cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
328d5b44036SJonas Devlieghere   m_editline_up.reset();
329cacde7dfSTodd Fiala #endif
33044d93782SGreg Clayton }
33144d93782SGreg Clayton 
332b9c1b51eSKate Stone void IOHandlerEditline::Activate() {
333e30f11d9SKate Stone   IOHandler::Activate();
3340affb582SDave Lee   m_delegate.IOHandlerActivated(*this, GetIsInteractive());
335e30f11d9SKate Stone }
336e30f11d9SKate Stone 
337b9c1b51eSKate Stone void IOHandlerEditline::Deactivate() {
338e30f11d9SKate Stone   IOHandler::Deactivate();
339e30f11d9SKate Stone   m_delegate.IOHandlerDeactivated(*this);
340e30f11d9SKate Stone }
341e30f11d9SKate Stone 
342b9c1b51eSKate Stone bool IOHandlerEditline::GetLine(std::string &line, bool &interrupted) {
343cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
344d5b44036SJonas Devlieghere   if (m_editline_up) {
345d77c2e09SJonas Devlieghere     bool b = m_editline_up->GetLine(line, interrupted);
346d77c2e09SJonas Devlieghere     if (m_data_recorder)
347d77c2e09SJonas Devlieghere       m_data_recorder->Record(line, true);
348d77c2e09SJonas Devlieghere     return b;
349b9c1b51eSKate Stone   } else {
350cacde7dfSTodd Fiala #endif
35144d93782SGreg Clayton     line.clear();
35244d93782SGreg Clayton 
35344d93782SGreg Clayton     FILE *in = GetInputFILE();
354b9c1b51eSKate Stone     if (in) {
355b9c1b51eSKate Stone       if (GetIsInteractive()) {
356c5dac77aSEugene Zelenko         const char *prompt = nullptr;
357e30f11d9SKate Stone 
358e30f11d9SKate Stone         if (m_multi_line && m_curr_line_idx > 0)
359e30f11d9SKate Stone           prompt = GetContinuationPrompt();
360e30f11d9SKate Stone 
361c5dac77aSEugene Zelenko         if (prompt == nullptr)
362e30f11d9SKate Stone           prompt = GetPrompt();
363e30f11d9SKate Stone 
364b9c1b51eSKate Stone         if (prompt && prompt[0]) {
36544d93782SGreg Clayton           FILE *out = GetOutputFILE();
366b9c1b51eSKate Stone           if (out) {
36744d93782SGreg Clayton             ::fprintf(out, "%s", prompt);
36844d93782SGreg Clayton             ::fflush(out);
36944d93782SGreg Clayton           }
37044d93782SGreg Clayton         }
37144d93782SGreg Clayton       }
37244d93782SGreg Clayton       char buffer[256];
37344d93782SGreg Clayton       bool done = false;
3740f86e6e7SGreg Clayton       bool got_line = false;
375e034a04eSGreg Clayton       m_editing = true;
376b9c1b51eSKate Stone       while (!done) {
377e7167908SNathan Lanza #ifdef _WIN32
378e7167908SNathan Lanza         // ReadFile on Windows is supposed to set ERROR_OPERATION_ABORTED
379e7167908SNathan Lanza         // according to the docs on MSDN. However, this has evidently been a
380e7167908SNathan Lanza         // known bug since Windows 8. Therefore, we can't detect if a signal
381e7167908SNathan Lanza         // interrupted in the fgets. So pressing ctrl-c causes the repl to end
382e7167908SNathan Lanza         // and the process to exit. A temporary workaround is just to attempt to
383e7167908SNathan Lanza         // fgets twice until this bug is fixed.
384e7167908SNathan Lanza         if (fgets(buffer, sizeof(buffer), in) == nullptr &&
385e7167908SNathan Lanza             fgets(buffer, sizeof(buffer), in) == nullptr) {
386cb305205SNathan Lanza           // this is the equivalent of EINTR for Windows
387cb305205SNathan Lanza           if (GetLastError() == ERROR_OPERATION_ABORTED)
388cb305205SNathan Lanza             continue;
389e7167908SNathan Lanza #else
390b9c1b51eSKate Stone         if (fgets(buffer, sizeof(buffer), in) == nullptr) {
391e7167908SNathan Lanza #endif
392c7797accSGreg Clayton           const int saved_errno = errno;
393c9cf5798SGreg Clayton           if (feof(in))
39444d93782SGreg Clayton             done = true;
395b9c1b51eSKate Stone           else if (ferror(in)) {
396c7797accSGreg Clayton             if (saved_errno != EINTR)
397c7797accSGreg Clayton               done = true;
398c7797accSGreg Clayton           }
399b9c1b51eSKate Stone         } else {
4000f86e6e7SGreg Clayton           got_line = true;
40144d93782SGreg Clayton           size_t buffer_len = strlen(buffer);
40244d93782SGreg Clayton           assert(buffer[buffer_len] == '\0');
40344d93782SGreg Clayton           char last_char = buffer[buffer_len - 1];
404b9c1b51eSKate Stone           if (last_char == '\r' || last_char == '\n') {
40544d93782SGreg Clayton             done = true;
40644d93782SGreg Clayton             // Strip trailing newlines
407b9c1b51eSKate Stone             while (last_char == '\r' || last_char == '\n') {
40844d93782SGreg Clayton               --buffer_len;
40944d93782SGreg Clayton               if (buffer_len == 0)
41044d93782SGreg Clayton                 break;
41144d93782SGreg Clayton               last_char = buffer[buffer_len - 1];
41244d93782SGreg Clayton             }
41344d93782SGreg Clayton           }
41444d93782SGreg Clayton           line.append(buffer, buffer_len);
41544d93782SGreg Clayton         }
41644d93782SGreg Clayton       }
417e034a04eSGreg Clayton       m_editing = false;
418d77c2e09SJonas Devlieghere       if (m_data_recorder && got_line)
419d77c2e09SJonas Devlieghere         m_data_recorder->Record(line, true);
42005097246SAdrian Prantl       // We might have gotten a newline on a line by itself make sure to return
42105097246SAdrian Prantl       // true in this case.
4220f86e6e7SGreg Clayton       return got_line;
423b9c1b51eSKate Stone     } else {
42444d93782SGreg Clayton       // No more input file, we are done...
42544d93782SGreg Clayton       SetIsDone(true);
42644d93782SGreg Clayton     }
427340b0309SGreg Clayton     return false;
428cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
42944d93782SGreg Clayton   }
430cacde7dfSTodd Fiala #endif
43144d93782SGreg Clayton }
43244d93782SGreg Clayton 
433cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
434b9c1b51eSKate Stone bool IOHandlerEditline::IsInputCompleteCallback(Editline *editline,
43544d93782SGreg Clayton                                                 StringList &lines,
436b9c1b51eSKate Stone                                                 void *baton) {
43744d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
438b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerIsInputComplete(*editline_reader,
439b9c1b51eSKate Stone                                                               lines);
440e30f11d9SKate Stone }
441e30f11d9SKate Stone 
442b9c1b51eSKate Stone int IOHandlerEditline::FixIndentationCallback(Editline *editline,
443e30f11d9SKate Stone                                               const StringList &lines,
444e30f11d9SKate Stone                                               int cursor_position,
445b9c1b51eSKate Stone                                               void *baton) {
446e30f11d9SKate Stone   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
447b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerFixIndentation(
448b9c1b51eSKate Stone       *editline_reader, lines, cursor_position);
44944d93782SGreg Clayton }
45044d93782SGreg Clayton 
4517f88829cSRaphael Isemann int IOHandlerEditline::AutoCompleteCallback(
4527f88829cSRaphael Isemann     const char *current_line, const char *cursor, const char *last_char,
4537f88829cSRaphael Isemann     int skip_first_n_matches, int max_matches, StringList &matches,
4547f88829cSRaphael Isemann     StringList &descriptions, void *baton) {
45544d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
45644d93782SGreg Clayton   if (editline_reader)
457b9c1b51eSKate Stone     return editline_reader->m_delegate.IOHandlerComplete(
458b9c1b51eSKate Stone         *editline_reader, current_line, cursor, last_char, skip_first_n_matches,
4597f88829cSRaphael Isemann         max_matches, matches, descriptions);
46044d93782SGreg Clayton   return 0;
46144d93782SGreg Clayton }
462cacde7dfSTodd Fiala #endif
46344d93782SGreg Clayton 
464b9c1b51eSKate Stone const char *IOHandlerEditline::GetPrompt() {
465cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
466d5b44036SJonas Devlieghere   if (m_editline_up) {
467d5b44036SJonas Devlieghere     return m_editline_up->GetPrompt();
468b9c1b51eSKate Stone   } else {
469cacde7dfSTodd Fiala #endif
470cacde7dfSTodd Fiala     if (m_prompt.empty())
471c5dac77aSEugene Zelenko       return nullptr;
472cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
473cacde7dfSTodd Fiala   }
474cacde7dfSTodd Fiala #endif
47544d93782SGreg Clayton   return m_prompt.c_str();
47644d93782SGreg Clayton }
47744d93782SGreg Clayton 
478514d8cd8SZachary Turner bool IOHandlerEditline::SetPrompt(llvm::StringRef prompt) {
479514d8cd8SZachary Turner   m_prompt = prompt;
480514d8cd8SZachary Turner 
481cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
482d5b44036SJonas Devlieghere   if (m_editline_up)
483d5b44036SJonas Devlieghere     m_editline_up->SetPrompt(m_prompt.empty() ? nullptr : m_prompt.c_str());
484cacde7dfSTodd Fiala #endif
48544d93782SGreg Clayton   return true;
48644d93782SGreg Clayton }
48744d93782SGreg Clayton 
488b9c1b51eSKate Stone const char *IOHandlerEditline::GetContinuationPrompt() {
489b9c1b51eSKate Stone   return (m_continuation_prompt.empty() ? nullptr
490b9c1b51eSKate Stone                                         : m_continuation_prompt.c_str());
491e30f11d9SKate Stone }
492e30f11d9SKate Stone 
493514d8cd8SZachary Turner void IOHandlerEditline::SetContinuationPrompt(llvm::StringRef prompt) {
494514d8cd8SZachary Turner   m_continuation_prompt = prompt;
495e30f11d9SKate Stone 
496d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
497d5b44036SJonas Devlieghere   if (m_editline_up)
498d5b44036SJonas Devlieghere     m_editline_up->SetContinuationPrompt(m_continuation_prompt.empty()
499b9c1b51eSKate Stone                                              ? nullptr
500b9c1b51eSKate Stone                                              : m_continuation_prompt.c_str());
501d553d00cSZachary Turner #endif
502e30f11d9SKate Stone }
503e30f11d9SKate Stone 
504b9c1b51eSKate Stone void IOHandlerEditline::SetBaseLineNumber(uint32_t line) {
505f6913cd7SGreg Clayton   m_base_line_number = line;
506f6913cd7SGreg Clayton }
507e30f11d9SKate Stone 
508b9c1b51eSKate Stone uint32_t IOHandlerEditline::GetCurrentLineIndex() const {
509d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
510d5b44036SJonas Devlieghere   if (m_editline_up)
511d5b44036SJonas Devlieghere     return m_editline_up->GetCurrentLine();
512e30f11d9SKate Stone #endif
513e30f11d9SKate Stone   return m_curr_line_idx;
514e30f11d9SKate Stone }
515e30f11d9SKate Stone 
516b9c1b51eSKate Stone bool IOHandlerEditline::GetLines(StringList &lines, bool &interrupted) {
517e30f11d9SKate Stone   m_current_lines_ptr = &lines;
518e30f11d9SKate Stone 
51944d93782SGreg Clayton   bool success = false;
520cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
521d5b44036SJonas Devlieghere   if (m_editline_up) {
522d5b44036SJonas Devlieghere     return m_editline_up->GetLines(m_base_line_number, lines, interrupted);
523b9c1b51eSKate Stone   } else {
524cacde7dfSTodd Fiala #endif
525e30f11d9SKate Stone     bool done = false;
52697206d57SZachary Turner     Status error;
52744d93782SGreg Clayton 
528b9c1b51eSKate Stone     while (!done) {
529f6913cd7SGreg Clayton       // Show line numbers if we are asked to
53044d93782SGreg Clayton       std::string line;
531b9c1b51eSKate Stone       if (m_base_line_number > 0 && GetIsInteractive()) {
532f6913cd7SGreg Clayton         FILE *out = GetOutputFILE();
533f6913cd7SGreg Clayton         if (out)
534b9c1b51eSKate Stone           ::fprintf(out, "%u%s", m_base_line_number + (uint32_t)lines.GetSize(),
535b9c1b51eSKate Stone                     GetPrompt() == nullptr ? " " : "");
536f6913cd7SGreg Clayton       }
537f6913cd7SGreg Clayton 
538e30f11d9SKate Stone       m_curr_line_idx = lines.GetSize();
539e30f11d9SKate Stone 
540f0066ad0SGreg Clayton       bool interrupted = false;
541b9c1b51eSKate Stone       if (GetLine(line, interrupted) && !interrupted) {
54244d93782SGreg Clayton         lines.AppendString(line);
543e30f11d9SKate Stone         done = m_delegate.IOHandlerIsInputComplete(*this, lines);
544b9c1b51eSKate Stone       } else {
545e30f11d9SKate Stone         done = true;
54644d93782SGreg Clayton       }
54744d93782SGreg Clayton     }
54844d93782SGreg Clayton     success = lines.GetSize() > 0;
549cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
55044d93782SGreg Clayton   }
551cacde7dfSTodd Fiala #endif
55244d93782SGreg Clayton   return success;
55344d93782SGreg Clayton }
55444d93782SGreg Clayton 
55505097246SAdrian Prantl // Each IOHandler gets to run until it is done. It should read data from the
55605097246SAdrian Prantl // "in" and place output into "out" and "err and return when done.
557b9c1b51eSKate Stone void IOHandlerEditline::Run() {
55844d93782SGreg Clayton   std::string line;
559b9c1b51eSKate Stone   while (IsActive()) {
560f0066ad0SGreg Clayton     bool interrupted = false;
561b9c1b51eSKate Stone     if (m_multi_line) {
56244d93782SGreg Clayton       StringList lines;
563b9c1b51eSKate Stone       if (GetLines(lines, interrupted)) {
564b9c1b51eSKate Stone         if (interrupted) {
565e30f11d9SKate Stone           m_done = m_interrupt_exits;
566e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
567e30f11d9SKate Stone 
568b9c1b51eSKate Stone         } else {
56944d93782SGreg Clayton           line = lines.CopyList();
57044d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
57144d93782SGreg Clayton         }
572b9c1b51eSKate Stone       } else {
57344d93782SGreg Clayton         m_done = true;
57444d93782SGreg Clayton       }
575b9c1b51eSKate Stone     } else {
576b9c1b51eSKate Stone       if (GetLine(line, interrupted)) {
577e30f11d9SKate Stone         if (interrupted)
578e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
579e30f11d9SKate Stone         else
58044d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
581b9c1b51eSKate Stone       } else {
58244d93782SGreg Clayton         m_done = true;
58344d93782SGreg Clayton       }
58444d93782SGreg Clayton     }
58544d93782SGreg Clayton   }
58644d93782SGreg Clayton }
58744d93782SGreg Clayton 
588b9c1b51eSKate Stone void IOHandlerEditline::Cancel() {
589cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
590d5b44036SJonas Devlieghere   if (m_editline_up)
591d5b44036SJonas Devlieghere     m_editline_up->Cancel();
592cacde7dfSTodd Fiala #endif
593e68f5d6bSGreg Clayton }
594e68f5d6bSGreg Clayton 
595b9c1b51eSKate Stone bool IOHandlerEditline::Interrupt() {
596f0066ad0SGreg Clayton   // Let the delgate handle it first
597f0066ad0SGreg Clayton   if (m_delegate.IOHandlerInterrupt(*this))
598f0066ad0SGreg Clayton     return true;
599f0066ad0SGreg Clayton 
600cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
601d5b44036SJonas Devlieghere   if (m_editline_up)
602d5b44036SJonas Devlieghere     return m_editline_up->Interrupt();
603cacde7dfSTodd Fiala #endif
604f0066ad0SGreg Clayton   return false;
60544d93782SGreg Clayton }
60644d93782SGreg Clayton 
607b9c1b51eSKate Stone void IOHandlerEditline::GotEOF() {
608cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
609d5b44036SJonas Devlieghere   if (m_editline_up)
610d5b44036SJonas Devlieghere     m_editline_up->Interrupt();
611cacde7dfSTodd Fiala #endif
61244d93782SGreg Clayton }
61344d93782SGreg Clayton 
614b9c1b51eSKate Stone void IOHandlerEditline::PrintAsync(Stream *stream, const char *s, size_t len) {
6154446487dSPavel Labath #ifndef LLDB_DISABLE_LIBEDIT
616d5b44036SJonas Devlieghere   if (m_editline_up)
617d5b44036SJonas Devlieghere     m_editline_up->PrintAsync(stream, s, len);
6184446487dSPavel Labath   else
6194446487dSPavel Labath #endif
620fab31220STed Woodward   {
621fab31220STed Woodward #ifdef _MSC_VER
622341e4789SDawn Perchik     const char *prompt = GetPrompt();
623b9c1b51eSKate Stone     if (prompt) {
624fab31220STed Woodward       // Back up over previous prompt using Windows API
625fab31220STed Woodward       CONSOLE_SCREEN_BUFFER_INFO screen_buffer_info;
626fab31220STed Woodward       HANDLE console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
627fab31220STed Woodward       GetConsoleScreenBufferInfo(console_handle, &screen_buffer_info);
628fab31220STed Woodward       COORD coord = screen_buffer_info.dwCursorPosition;
629fab31220STed Woodward       coord.X -= strlen(prompt);
630fab31220STed Woodward       if (coord.X < 0)
631fab31220STed Woodward         coord.X = 0;
632fab31220STed Woodward       SetConsoleCursorPosition(console_handle, coord);
633fab31220STed Woodward     }
634fab31220STed Woodward #endif
6354446487dSPavel Labath     IOHandler::PrintAsync(stream, s, len);
636341e4789SDawn Perchik #ifdef _MSC_VER
637fab31220STed Woodward     if (prompt)
638b9c1b51eSKate Stone       IOHandler::PrintAsync(GetOutputStreamFile().get(), prompt,
639b9c1b51eSKate Stone                             strlen(prompt));
640341e4789SDawn Perchik #endif
641fab31220STed Woodward   }
6424446487dSPavel Labath }
6434446487dSPavel Labath 
64405097246SAdrian Prantl // we may want curses to be disabled for some builds for instance, windows
645914b8d98SDeepak Panickal #ifndef LLDB_DISABLE_CURSES
646914b8d98SDeepak Panickal 
64744d93782SGreg Clayton #define KEY_RETURN 10
64844d93782SGreg Clayton #define KEY_ESCAPE 27
64944d93782SGreg Clayton 
650b9c1b51eSKate Stone namespace curses {
65144d93782SGreg Clayton class Menu;
65244d93782SGreg Clayton class MenuDelegate;
65344d93782SGreg Clayton class Window;
65444d93782SGreg Clayton class WindowDelegate;
65544d93782SGreg Clayton typedef std::shared_ptr<Menu> MenuSP;
65644d93782SGreg Clayton typedef std::shared_ptr<MenuDelegate> MenuDelegateSP;
65744d93782SGreg Clayton typedef std::shared_ptr<Window> WindowSP;
65844d93782SGreg Clayton typedef std::shared_ptr<WindowDelegate> WindowDelegateSP;
65944d93782SGreg Clayton typedef std::vector<MenuSP> Menus;
66044d93782SGreg Clayton typedef std::vector<WindowSP> Windows;
66144d93782SGreg Clayton typedef std::vector<WindowDelegateSP> WindowDelegates;
66244d93782SGreg Clayton 
66344d93782SGreg Clayton #if 0
66444d93782SGreg Clayton type summary add -s "x=${var.x}, y=${var.y}" curses::Point
66544d93782SGreg Clayton type summary add -s "w=${var.width}, h=${var.height}" curses::Size
66644d93782SGreg Clayton type summary add -s "${var.origin%S} ${var.size%S}" curses::Rect
66744d93782SGreg Clayton #endif
668315b6884SEugene Zelenko 
669b9c1b51eSKate Stone struct Point {
67044d93782SGreg Clayton   int x;
67144d93782SGreg Clayton   int y;
67244d93782SGreg Clayton 
673b9c1b51eSKate Stone   Point(int _x = 0, int _y = 0) : x(_x), y(_y) {}
67444d93782SGreg Clayton 
675b9c1b51eSKate Stone   void Clear() {
67644d93782SGreg Clayton     x = 0;
67744d93782SGreg Clayton     y = 0;
67844d93782SGreg Clayton   }
67944d93782SGreg Clayton 
680b9c1b51eSKate Stone   Point &operator+=(const Point &rhs) {
68144d93782SGreg Clayton     x += rhs.x;
68244d93782SGreg Clayton     y += rhs.y;
68344d93782SGreg Clayton     return *this;
68444d93782SGreg Clayton   }
68544d93782SGreg Clayton 
686b9c1b51eSKate Stone   void Dump() { printf("(x=%i, y=%i)\n", x, y); }
68744d93782SGreg Clayton };
68844d93782SGreg Clayton 
689b9c1b51eSKate Stone bool operator==(const Point &lhs, const Point &rhs) {
69044d93782SGreg Clayton   return lhs.x == rhs.x && lhs.y == rhs.y;
69144d93782SGreg Clayton }
692315b6884SEugene Zelenko 
693b9c1b51eSKate Stone bool operator!=(const Point &lhs, const Point &rhs) {
69444d93782SGreg Clayton   return lhs.x != rhs.x || lhs.y != rhs.y;
69544d93782SGreg Clayton }
69644d93782SGreg Clayton 
697b9c1b51eSKate Stone struct Size {
69844d93782SGreg Clayton   int width;
69944d93782SGreg Clayton   int height;
700b9c1b51eSKate Stone   Size(int w = 0, int h = 0) : width(w), height(h) {}
70144d93782SGreg Clayton 
702b9c1b51eSKate Stone   void Clear() {
70344d93782SGreg Clayton     width = 0;
70444d93782SGreg Clayton     height = 0;
70544d93782SGreg Clayton   }
70644d93782SGreg Clayton 
707b9c1b51eSKate Stone   void Dump() { printf("(w=%i, h=%i)\n", width, height); }
70844d93782SGreg Clayton };
70944d93782SGreg Clayton 
710b9c1b51eSKate Stone bool operator==(const Size &lhs, const Size &rhs) {
71144d93782SGreg Clayton   return lhs.width == rhs.width && lhs.height == rhs.height;
71244d93782SGreg Clayton }
713315b6884SEugene Zelenko 
714b9c1b51eSKate Stone bool operator!=(const Size &lhs, const Size &rhs) {
71544d93782SGreg Clayton   return lhs.width != rhs.width || lhs.height != rhs.height;
71644d93782SGreg Clayton }
71744d93782SGreg Clayton 
718b9c1b51eSKate Stone struct Rect {
71944d93782SGreg Clayton   Point origin;
72044d93782SGreg Clayton   Size size;
72144d93782SGreg Clayton 
722b9c1b51eSKate Stone   Rect() : origin(), size() {}
72344d93782SGreg Clayton 
724b9c1b51eSKate Stone   Rect(const Point &p, const Size &s) : origin(p), size(s) {}
72544d93782SGreg Clayton 
726b9c1b51eSKate Stone   void Clear() {
72744d93782SGreg Clayton     origin.Clear();
72844d93782SGreg Clayton     size.Clear();
72944d93782SGreg Clayton   }
73044d93782SGreg Clayton 
731b9c1b51eSKate Stone   void Dump() {
732b9c1b51eSKate Stone     printf("(x=%i, y=%i), w=%i, h=%i)\n", origin.x, origin.y, size.width,
733b9c1b51eSKate Stone            size.height);
73444d93782SGreg Clayton   }
73544d93782SGreg Clayton 
736b9c1b51eSKate Stone   void Inset(int w, int h) {
73744d93782SGreg Clayton     if (size.width > w * 2)
73844d93782SGreg Clayton       size.width -= w * 2;
73944d93782SGreg Clayton     origin.x += w;
74044d93782SGreg Clayton 
74144d93782SGreg Clayton     if (size.height > h * 2)
74244d93782SGreg Clayton       size.height -= h * 2;
74344d93782SGreg Clayton     origin.y += h;
74444d93782SGreg Clayton   }
745315b6884SEugene Zelenko 
74605097246SAdrian Prantl   // Return a status bar rectangle which is the last line of this rectangle.
74705097246SAdrian Prantl   // This rectangle will be modified to not include the status bar area.
748b9c1b51eSKate Stone   Rect MakeStatusBar() {
74944d93782SGreg Clayton     Rect status_bar;
750b9c1b51eSKate Stone     if (size.height > 1) {
75144d93782SGreg Clayton       status_bar.origin.x = origin.x;
75244d93782SGreg Clayton       status_bar.origin.y = size.height;
75344d93782SGreg Clayton       status_bar.size.width = size.width;
75444d93782SGreg Clayton       status_bar.size.height = 1;
75544d93782SGreg Clayton       --size.height;
75644d93782SGreg Clayton     }
75744d93782SGreg Clayton     return status_bar;
75844d93782SGreg Clayton   }
75944d93782SGreg Clayton 
76005097246SAdrian Prantl   // Return a menubar rectangle which is the first line of this rectangle. This
76105097246SAdrian Prantl   // rectangle will be modified to not include the menubar area.
762b9c1b51eSKate Stone   Rect MakeMenuBar() {
76344d93782SGreg Clayton     Rect menubar;
764b9c1b51eSKate Stone     if (size.height > 1) {
76544d93782SGreg Clayton       menubar.origin.x = origin.x;
76644d93782SGreg Clayton       menubar.origin.y = origin.y;
76744d93782SGreg Clayton       menubar.size.width = size.width;
76844d93782SGreg Clayton       menubar.size.height = 1;
76944d93782SGreg Clayton       ++origin.y;
77044d93782SGreg Clayton       --size.height;
77144d93782SGreg Clayton     }
77244d93782SGreg Clayton     return menubar;
77344d93782SGreg Clayton   }
77444d93782SGreg Clayton 
775b9c1b51eSKate Stone   void HorizontalSplitPercentage(float top_percentage, Rect &top,
776b9c1b51eSKate Stone                                  Rect &bottom) const {
77744d93782SGreg Clayton     float top_height = top_percentage * size.height;
77844d93782SGreg Clayton     HorizontalSplit(top_height, top, bottom);
77944d93782SGreg Clayton   }
78044d93782SGreg Clayton 
781b9c1b51eSKate Stone   void HorizontalSplit(int top_height, Rect &top, Rect &bottom) const {
78244d93782SGreg Clayton     top = *this;
783b9c1b51eSKate Stone     if (top_height < size.height) {
78444d93782SGreg Clayton       top.size.height = top_height;
78544d93782SGreg Clayton       bottom.origin.x = origin.x;
78644d93782SGreg Clayton       bottom.origin.y = origin.y + top.size.height;
78744d93782SGreg Clayton       bottom.size.width = size.width;
78844d93782SGreg Clayton       bottom.size.height = size.height - top.size.height;
789b9c1b51eSKate Stone     } else {
79044d93782SGreg Clayton       bottom.Clear();
79144d93782SGreg Clayton     }
79244d93782SGreg Clayton   }
79344d93782SGreg Clayton 
794b9c1b51eSKate Stone   void VerticalSplitPercentage(float left_percentage, Rect &left,
795b9c1b51eSKate Stone                                Rect &right) const {
79644d93782SGreg Clayton     float left_width = left_percentage * size.width;
79744d93782SGreg Clayton     VerticalSplit(left_width, left, right);
79844d93782SGreg Clayton   }
79944d93782SGreg Clayton 
800b9c1b51eSKate Stone   void VerticalSplit(int left_width, Rect &left, Rect &right) const {
80144d93782SGreg Clayton     left = *this;
802b9c1b51eSKate Stone     if (left_width < size.width) {
80344d93782SGreg Clayton       left.size.width = left_width;
80444d93782SGreg Clayton       right.origin.x = origin.x + left.size.width;
80544d93782SGreg Clayton       right.origin.y = origin.y;
80644d93782SGreg Clayton       right.size.width = size.width - left.size.width;
80744d93782SGreg Clayton       right.size.height = size.height;
808b9c1b51eSKate Stone     } else {
80944d93782SGreg Clayton       right.Clear();
81044d93782SGreg Clayton     }
81144d93782SGreg Clayton   }
81244d93782SGreg Clayton };
81344d93782SGreg Clayton 
814b9c1b51eSKate Stone bool operator==(const Rect &lhs, const Rect &rhs) {
81544d93782SGreg Clayton   return lhs.origin == rhs.origin && lhs.size == rhs.size;
81644d93782SGreg Clayton }
817315b6884SEugene Zelenko 
818b9c1b51eSKate Stone bool operator!=(const Rect &lhs, const Rect &rhs) {
81944d93782SGreg Clayton   return lhs.origin != rhs.origin || lhs.size != rhs.size;
82044d93782SGreg Clayton }
82144d93782SGreg Clayton 
822b9c1b51eSKate Stone enum HandleCharResult {
82344d93782SGreg Clayton   eKeyNotHandled = 0,
82444d93782SGreg Clayton   eKeyHandled = 1,
82544d93782SGreg Clayton   eQuitApplication = 2
82644d93782SGreg Clayton };
82744d93782SGreg Clayton 
828b9c1b51eSKate Stone enum class MenuActionResult {
82944d93782SGreg Clayton   Handled,
83044d93782SGreg Clayton   NotHandled,
83144d93782SGreg Clayton   Quit // Exit all menus and quit
83244d93782SGreg Clayton };
83344d93782SGreg Clayton 
834b9c1b51eSKate Stone struct KeyHelp {
83544d93782SGreg Clayton   int ch;
83644d93782SGreg Clayton   const char *description;
83744d93782SGreg Clayton };
83844d93782SGreg Clayton 
839b9c1b51eSKate Stone class WindowDelegate {
84044d93782SGreg Clayton public:
841b9c1b51eSKate Stone   virtual ~WindowDelegate() = default;
84244d93782SGreg Clayton 
843b9c1b51eSKate Stone   virtual bool WindowDelegateDraw(Window &window, bool force) {
84444d93782SGreg Clayton     return false; // Drawing not handled
84544d93782SGreg Clayton   }
84644d93782SGreg Clayton 
847b9c1b51eSKate Stone   virtual HandleCharResult WindowDelegateHandleChar(Window &window, int key) {
84844d93782SGreg Clayton     return eKeyNotHandled;
84944d93782SGreg Clayton   }
85044d93782SGreg Clayton 
851b9c1b51eSKate Stone   virtual const char *WindowDelegateGetHelpText() { return nullptr; }
85244d93782SGreg Clayton 
853b9c1b51eSKate Stone   virtual KeyHelp *WindowDelegateGetKeyHelp() { return nullptr; }
85444d93782SGreg Clayton };
85544d93782SGreg Clayton 
856b9c1b51eSKate Stone class HelpDialogDelegate : public WindowDelegate {
85744d93782SGreg Clayton public:
85844d93782SGreg Clayton   HelpDialogDelegate(const char *text, KeyHelp *key_help_array);
85944d93782SGreg Clayton 
860bd5ae6b4SGreg Clayton   ~HelpDialogDelegate() override;
86144d93782SGreg Clayton 
862b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
86344d93782SGreg Clayton 
864b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
86544d93782SGreg Clayton 
866b9c1b51eSKate Stone   size_t GetNumLines() const { return m_text.GetSize(); }
86744d93782SGreg Clayton 
868b9c1b51eSKate Stone   size_t GetMaxLineLength() const { return m_text.GetMaxStringLength(); }
86944d93782SGreg Clayton 
87044d93782SGreg Clayton protected:
87144d93782SGreg Clayton   StringList m_text;
87244d93782SGreg Clayton   int m_first_visible_line;
87344d93782SGreg Clayton };
87444d93782SGreg Clayton 
875b9c1b51eSKate Stone class Window {
87644d93782SGreg Clayton public:
877b9c1b51eSKate Stone   Window(const char *name)
878b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
879b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
880b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(false),
881b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {}
88244d93782SGreg Clayton 
883b9c1b51eSKate Stone   Window(const char *name, WINDOW *w, bool del = true)
884b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
885b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
886b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(del),
887b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
88844d93782SGreg Clayton     if (w)
88944d93782SGreg Clayton       Reset(w);
89044d93782SGreg Clayton   }
89144d93782SGreg Clayton 
892b9c1b51eSKate Stone   Window(const char *name, const Rect &bounds)
893b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_parent(nullptr), m_subwindows(),
894b9c1b51eSKate Stone         m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
895b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(true),
896b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
897b9c1b51eSKate Stone     Reset(::newwin(bounds.size.height, bounds.size.width, bounds.origin.y,
898b9c1b51eSKate Stone                    bounds.origin.y));
89944d93782SGreg Clayton   }
90044d93782SGreg Clayton 
901b9c1b51eSKate Stone   virtual ~Window() {
90244d93782SGreg Clayton     RemoveSubWindows();
90344d93782SGreg Clayton     Reset();
90444d93782SGreg Clayton   }
90544d93782SGreg Clayton 
906b9c1b51eSKate Stone   void Reset(WINDOW *w = nullptr, bool del = true) {
90744d93782SGreg Clayton     if (m_window == w)
90844d93782SGreg Clayton       return;
90944d93782SGreg Clayton 
910b9c1b51eSKate Stone     if (m_panel) {
91144d93782SGreg Clayton       ::del_panel(m_panel);
912c5dac77aSEugene Zelenko       m_panel = nullptr;
91344d93782SGreg Clayton     }
914b9c1b51eSKate Stone     if (m_window && m_delete) {
91544d93782SGreg Clayton       ::delwin(m_window);
916c5dac77aSEugene Zelenko       m_window = nullptr;
91744d93782SGreg Clayton       m_delete = false;
91844d93782SGreg Clayton     }
919b9c1b51eSKate Stone     if (w) {
92044d93782SGreg Clayton       m_window = w;
92144d93782SGreg Clayton       m_panel = ::new_panel(m_window);
92244d93782SGreg Clayton       m_delete = del;
92344d93782SGreg Clayton     }
92444d93782SGreg Clayton   }
92544d93782SGreg Clayton 
92644d93782SGreg Clayton   void AttributeOn(attr_t attr) { ::wattron(m_window, attr); }
92744d93782SGreg Clayton   void AttributeOff(attr_t attr) { ::wattroff(m_window, attr); }
928b9c1b51eSKate Stone   void Box(chtype v_char = ACS_VLINE, chtype h_char = ACS_HLINE) {
929b9c1b51eSKate Stone     ::box(m_window, v_char, h_char);
930b9c1b51eSKate Stone   }
93144d93782SGreg Clayton   void Clear() { ::wclear(m_window); }
93244d93782SGreg Clayton   void Erase() { ::werase(m_window); }
933b9c1b51eSKate Stone   Rect GetBounds() {
934b9c1b51eSKate Stone     return Rect(GetParentOrigin(), GetSize());
935b9c1b51eSKate Stone   } // Get the rectangle in our parent window
93644d93782SGreg Clayton   int GetChar() { return ::wgetch(m_window); }
93744d93782SGreg Clayton   int GetCursorX() { return getcurx(m_window); }
93844d93782SGreg Clayton   int GetCursorY() { return getcury(m_window); }
939b9c1b51eSKate Stone   Rect GetFrame() {
940b9c1b51eSKate Stone     return Rect(Point(), GetSize());
941b9c1b51eSKate Stone   } // Get our rectangle in our own coordinate system
94244d93782SGreg Clayton   Point GetParentOrigin() { return Point(GetParentX(), GetParentY()); }
94344d93782SGreg Clayton   Size GetSize() { return Size(GetWidth(), GetHeight()); }
94444d93782SGreg Clayton   int GetParentX() { return getparx(m_window); }
94544d93782SGreg Clayton   int GetParentY() { return getpary(m_window); }
94644d93782SGreg Clayton   int GetMaxX() { return getmaxx(m_window); }
94744d93782SGreg Clayton   int GetMaxY() { return getmaxy(m_window); }
94844d93782SGreg Clayton   int GetWidth() { return GetMaxX(); }
94944d93782SGreg Clayton   int GetHeight() { return GetMaxY(); }
95044d93782SGreg Clayton   void MoveCursor(int x, int y) { ::wmove(m_window, y, x); }
95144d93782SGreg Clayton   void MoveWindow(int x, int y) { MoveWindow(Point(x, y)); }
95244d93782SGreg Clayton   void Resize(int w, int h) { ::wresize(m_window, h, w); }
953b9c1b51eSKate Stone   void Resize(const Size &size) {
954b9c1b51eSKate Stone     ::wresize(m_window, size.height, size.width);
955b9c1b51eSKate Stone   }
95644d93782SGreg Clayton   void PutChar(int ch) { ::waddch(m_window, ch); }
95744d93782SGreg Clayton   void PutCString(const char *s, int len = -1) { ::waddnstr(m_window, s, len); }
95844d93782SGreg Clayton   void Refresh() { ::wrefresh(m_window); }
959b9c1b51eSKate Stone   void DeferredRefresh() {
96044d93782SGreg Clayton     // We are using panels, so we don't need to call this...
96144d93782SGreg Clayton     //::wnoutrefresh(m_window);
96244d93782SGreg Clayton   }
963b9c1b51eSKate Stone   void SetBackground(int color_pair_idx) {
964b9c1b51eSKate Stone     ::wbkgd(m_window, COLOR_PAIR(color_pair_idx));
965b9c1b51eSKate Stone   }
96644d93782SGreg Clayton   void UnderlineOn() { AttributeOn(A_UNDERLINE); }
96744d93782SGreg Clayton   void UnderlineOff() { AttributeOff(A_UNDERLINE); }
96844d93782SGreg Clayton 
969b9c1b51eSKate Stone   void PutCStringTruncated(const char *s, int right_pad) {
97044d93782SGreg Clayton     int bytes_left = GetWidth() - GetCursorX();
971b9c1b51eSKate Stone     if (bytes_left > right_pad) {
97244d93782SGreg Clayton       bytes_left -= right_pad;
97344d93782SGreg Clayton       ::waddnstr(m_window, s, bytes_left);
97444d93782SGreg Clayton     }
97544d93782SGreg Clayton   }
97644d93782SGreg Clayton 
977b9c1b51eSKate Stone   void MoveWindow(const Point &origin) {
97844d93782SGreg Clayton     const bool moving_window = origin != GetParentOrigin();
979b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
98044d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
98144d93782SGreg Clayton       Size size = GetSize();
982b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, size.height, size.width, origin.y,
983b9c1b51eSKate Stone                      origin.x),
984b9c1b51eSKate Stone             true);
985b9c1b51eSKate Stone     } else {
98644d93782SGreg Clayton       ::mvwin(m_window, origin.y, origin.x);
98744d93782SGreg Clayton     }
98844d93782SGreg Clayton   }
98944d93782SGreg Clayton 
990b9c1b51eSKate Stone   void SetBounds(const Rect &bounds) {
99144d93782SGreg Clayton     const bool moving_window = bounds.origin != GetParentOrigin();
992b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
99344d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
994b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, bounds.size.height, bounds.size.width,
995b9c1b51eSKate Stone                      bounds.origin.y, bounds.origin.x),
996b9c1b51eSKate Stone             true);
997b9c1b51eSKate Stone     } else {
99844d93782SGreg Clayton       if (moving_window)
99944d93782SGreg Clayton         MoveWindow(bounds.origin);
100044d93782SGreg Clayton       Resize(bounds.size);
100144d93782SGreg Clayton     }
100244d93782SGreg Clayton   }
100344d93782SGreg Clayton 
1004b9c1b51eSKate Stone   void Printf(const char *format, ...) __attribute__((format(printf, 2, 3))) {
100544d93782SGreg Clayton     va_list args;
100644d93782SGreg Clayton     va_start(args, format);
100744d93782SGreg Clayton     vwprintw(m_window, format, args);
100844d93782SGreg Clayton     va_end(args);
100944d93782SGreg Clayton   }
101044d93782SGreg Clayton 
1011b9c1b51eSKate Stone   void Touch() {
101244d93782SGreg Clayton     ::touchwin(m_window);
101344d93782SGreg Clayton     if (m_parent)
101444d93782SGreg Clayton       m_parent->Touch();
101544d93782SGreg Clayton   }
101644d93782SGreg Clayton 
1017b9c1b51eSKate Stone   WindowSP CreateSubWindow(const char *name, const Rect &bounds,
1018b9c1b51eSKate Stone                            bool make_active) {
1019c6091d2bSJonas Devlieghere     auto get_window = [this, &bounds]() {
1020c6091d2bSJonas Devlieghere       return m_window
1021c6091d2bSJonas Devlieghere                  ? ::subwin(m_window, bounds.size.height, bounds.size.width,
1022c6091d2bSJonas Devlieghere                             bounds.origin.y, bounds.origin.x)
1023c6091d2bSJonas Devlieghere                  : ::newwin(bounds.size.height, bounds.size.width,
1024c6091d2bSJonas Devlieghere                             bounds.origin.y, bounds.origin.x);
1025c6091d2bSJonas Devlieghere     };
1026c6091d2bSJonas Devlieghere     WindowSP subwindow_sp = std::make_shared<Window>(name, get_window(), true);
1027c6091d2bSJonas Devlieghere     subwindow_sp->m_is_subwin = subwindow_sp.operator bool();
102844d93782SGreg Clayton     subwindow_sp->m_parent = this;
1029b9c1b51eSKate Stone     if (make_active) {
103044d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
103144d93782SGreg Clayton       m_curr_active_window_idx = m_subwindows.size();
103244d93782SGreg Clayton     }
103344d93782SGreg Clayton     m_subwindows.push_back(subwindow_sp);
103444d93782SGreg Clayton     ::top_panel(subwindow_sp->m_panel);
103544d93782SGreg Clayton     m_needs_update = true;
103644d93782SGreg Clayton     return subwindow_sp;
103744d93782SGreg Clayton   }
103844d93782SGreg Clayton 
1039b9c1b51eSKate Stone   bool RemoveSubWindow(Window *window) {
104044d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
104144d93782SGreg Clayton     size_t i = 0;
1042b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
1043b9c1b51eSKate Stone       if ((*pos).get() == window) {
104444d93782SGreg Clayton         if (m_prev_active_window_idx == i)
104544d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1046b9c1b51eSKate Stone         else if (m_prev_active_window_idx != UINT32_MAX &&
1047b9c1b51eSKate Stone                  m_prev_active_window_idx > i)
104844d93782SGreg Clayton           --m_prev_active_window_idx;
104944d93782SGreg Clayton 
105044d93782SGreg Clayton         if (m_curr_active_window_idx == i)
105144d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
1052b9c1b51eSKate Stone         else if (m_curr_active_window_idx != UINT32_MAX &&
1053b9c1b51eSKate Stone                  m_curr_active_window_idx > i)
105444d93782SGreg Clayton           --m_curr_active_window_idx;
105544d93782SGreg Clayton         window->Erase();
105644d93782SGreg Clayton         m_subwindows.erase(pos);
105744d93782SGreg Clayton         m_needs_update = true;
105844d93782SGreg Clayton         if (m_parent)
105944d93782SGreg Clayton           m_parent->Touch();
106044d93782SGreg Clayton         else
106144d93782SGreg Clayton           ::touchwin(stdscr);
106244d93782SGreg Clayton         return true;
106344d93782SGreg Clayton       }
106444d93782SGreg Clayton     }
106544d93782SGreg Clayton     return false;
106644d93782SGreg Clayton   }
106744d93782SGreg Clayton 
1068b9c1b51eSKate Stone   WindowSP FindSubWindow(const char *name) {
106944d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
107044d93782SGreg Clayton     size_t i = 0;
1071b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
10728d20cfdfSJonas Devlieghere       if ((*pos)->m_name == name)
107344d93782SGreg Clayton         return *pos;
107444d93782SGreg Clayton     }
107544d93782SGreg Clayton     return WindowSP();
107644d93782SGreg Clayton   }
107744d93782SGreg Clayton 
1078b9c1b51eSKate Stone   void RemoveSubWindows() {
107944d93782SGreg Clayton     m_curr_active_window_idx = UINT32_MAX;
108044d93782SGreg Clayton     m_prev_active_window_idx = UINT32_MAX;
108144d93782SGreg Clayton     for (Windows::iterator pos = m_subwindows.begin();
1082b9c1b51eSKate Stone          pos != m_subwindows.end(); pos = m_subwindows.erase(pos)) {
108344d93782SGreg Clayton       (*pos)->Erase();
108444d93782SGreg Clayton     }
108544d93782SGreg Clayton     if (m_parent)
108644d93782SGreg Clayton       m_parent->Touch();
108744d93782SGreg Clayton     else
108844d93782SGreg Clayton       ::touchwin(stdscr);
108944d93782SGreg Clayton   }
109044d93782SGreg Clayton 
1091b9c1b51eSKate Stone   WINDOW *get() { return m_window; }
109244d93782SGreg Clayton 
1093b9c1b51eSKate Stone   operator WINDOW *() { return m_window; }
109444d93782SGreg Clayton 
109544d93782SGreg Clayton   // Window drawing utilities
1096b9c1b51eSKate Stone   void DrawTitleBox(const char *title, const char *bottom_message = nullptr) {
109744d93782SGreg Clayton     attr_t attr = 0;
109844d93782SGreg Clayton     if (IsActive())
109944d93782SGreg Clayton       attr = A_BOLD | COLOR_PAIR(2);
110044d93782SGreg Clayton     else
110144d93782SGreg Clayton       attr = 0;
110244d93782SGreg Clayton     if (attr)
110344d93782SGreg Clayton       AttributeOn(attr);
110444d93782SGreg Clayton 
110544d93782SGreg Clayton     Box();
110644d93782SGreg Clayton     MoveCursor(3, 0);
110744d93782SGreg Clayton 
1108b9c1b51eSKate Stone     if (title && title[0]) {
110944d93782SGreg Clayton       PutChar('<');
111044d93782SGreg Clayton       PutCString(title);
111144d93782SGreg Clayton       PutChar('>');
111244d93782SGreg Clayton     }
111344d93782SGreg Clayton 
1114b9c1b51eSKate Stone     if (bottom_message && bottom_message[0]) {
111544d93782SGreg Clayton       int bottom_message_length = strlen(bottom_message);
111644d93782SGreg Clayton       int x = GetWidth() - 3 - (bottom_message_length + 2);
111744d93782SGreg Clayton 
1118b9c1b51eSKate Stone       if (x > 0) {
111944d93782SGreg Clayton         MoveCursor(x, GetHeight() - 1);
112044d93782SGreg Clayton         PutChar('[');
112144d93782SGreg Clayton         PutCString(bottom_message);
112244d93782SGreg Clayton         PutChar(']');
1123b9c1b51eSKate Stone       } else {
112444d93782SGreg Clayton         MoveCursor(1, GetHeight() - 1);
112544d93782SGreg Clayton         PutChar('[');
112644d93782SGreg Clayton         PutCStringTruncated(bottom_message, 1);
112744d93782SGreg Clayton       }
112844d93782SGreg Clayton     }
112944d93782SGreg Clayton     if (attr)
113044d93782SGreg Clayton       AttributeOff(attr);
113144d93782SGreg Clayton   }
113244d93782SGreg Clayton 
1133b9c1b51eSKate Stone   virtual void Draw(bool force) {
113444d93782SGreg Clayton     if (m_delegate_sp && m_delegate_sp->WindowDelegateDraw(*this, force))
113544d93782SGreg Clayton       return;
113644d93782SGreg Clayton 
113744d93782SGreg Clayton     for (auto &subwindow_sp : m_subwindows)
113844d93782SGreg Clayton       subwindow_sp->Draw(force);
113944d93782SGreg Clayton   }
114044d93782SGreg Clayton 
1141b9c1b51eSKate Stone   bool CreateHelpSubwindow() {
1142b9c1b51eSKate Stone     if (m_delegate_sp) {
114344d93782SGreg Clayton       const char *text = m_delegate_sp->WindowDelegateGetHelpText();
114444d93782SGreg Clayton       KeyHelp *key_help = m_delegate_sp->WindowDelegateGetKeyHelp();
1145b9c1b51eSKate Stone       if ((text && text[0]) || key_help) {
1146d5b44036SJonas Devlieghere         std::unique_ptr<HelpDialogDelegate> help_delegate_up(
1147b9c1b51eSKate Stone             new HelpDialogDelegate(text, key_help));
1148d5b44036SJonas Devlieghere         const size_t num_lines = help_delegate_up->GetNumLines();
1149d5b44036SJonas Devlieghere         const size_t max_length = help_delegate_up->GetMaxLineLength();
115044d93782SGreg Clayton         Rect bounds = GetBounds();
115144d93782SGreg Clayton         bounds.Inset(1, 1);
1152b9c1b51eSKate Stone         if (max_length + 4 < static_cast<size_t>(bounds.size.width)) {
115344d93782SGreg Clayton           bounds.origin.x += (bounds.size.width - max_length + 4) / 2;
115444d93782SGreg Clayton           bounds.size.width = max_length + 4;
1155b9c1b51eSKate Stone         } else {
1156b9c1b51eSKate Stone           if (bounds.size.width > 100) {
115744d93782SGreg Clayton             const int inset_w = bounds.size.width / 4;
115844d93782SGreg Clayton             bounds.origin.x += inset_w;
115944d93782SGreg Clayton             bounds.size.width -= 2 * inset_w;
116044d93782SGreg Clayton           }
116144d93782SGreg Clayton         }
116244d93782SGreg Clayton 
1163b9c1b51eSKate Stone         if (num_lines + 2 < static_cast<size_t>(bounds.size.height)) {
116444d93782SGreg Clayton           bounds.origin.y += (bounds.size.height - num_lines + 2) / 2;
116544d93782SGreg Clayton           bounds.size.height = num_lines + 2;
1166b9c1b51eSKate Stone         } else {
1167b9c1b51eSKate Stone           if (bounds.size.height > 100) {
116844d93782SGreg Clayton             const int inset_h = bounds.size.height / 4;
116944d93782SGreg Clayton             bounds.origin.y += inset_h;
117044d93782SGreg Clayton             bounds.size.height -= 2 * inset_h;
117144d93782SGreg Clayton           }
117244d93782SGreg Clayton         }
11735fdb09bbSGreg Clayton         WindowSP help_window_sp;
11745fdb09bbSGreg Clayton         Window *parent_window = GetParent();
11755fdb09bbSGreg Clayton         if (parent_window)
11765fdb09bbSGreg Clayton           help_window_sp = parent_window->CreateSubWindow("Help", bounds, true);
11775fdb09bbSGreg Clayton         else
11785fdb09bbSGreg Clayton           help_window_sp = CreateSubWindow("Help", bounds, true);
1179b9c1b51eSKate Stone         help_window_sp->SetDelegate(
1180d5b44036SJonas Devlieghere             WindowDelegateSP(help_delegate_up.release()));
118144d93782SGreg Clayton         return true;
118244d93782SGreg Clayton       }
118344d93782SGreg Clayton     }
118444d93782SGreg Clayton     return false;
118544d93782SGreg Clayton   }
118644d93782SGreg Clayton 
1187b9c1b51eSKate Stone   virtual HandleCharResult HandleChar(int key) {
118844d93782SGreg Clayton     // Always check the active window first
118944d93782SGreg Clayton     HandleCharResult result = eKeyNotHandled;
119044d93782SGreg Clayton     WindowSP active_window_sp = GetActiveWindow();
1191b9c1b51eSKate Stone     if (active_window_sp) {
119244d93782SGreg Clayton       result = active_window_sp->HandleChar(key);
119344d93782SGreg Clayton       if (result != eKeyNotHandled)
119444d93782SGreg Clayton         return result;
119544d93782SGreg Clayton     }
119644d93782SGreg Clayton 
1197b9c1b51eSKate Stone     if (m_delegate_sp) {
119844d93782SGreg Clayton       result = m_delegate_sp->WindowDelegateHandleChar(*this, key);
119944d93782SGreg Clayton       if (result != eKeyNotHandled)
120044d93782SGreg Clayton         return result;
120144d93782SGreg Clayton     }
120244d93782SGreg Clayton 
120305097246SAdrian Prantl     // Then check for any windows that want any keys that weren't handled. This
120405097246SAdrian Prantl     // is typically only for a menubar. Make a copy of the subwindows in case
120505097246SAdrian Prantl     // any HandleChar() functions muck with the subwindows. If we don't do
120605097246SAdrian Prantl     // this, we can crash when iterating over the subwindows.
120744d93782SGreg Clayton     Windows subwindows(m_subwindows);
1208b9c1b51eSKate Stone     for (auto subwindow_sp : subwindows) {
1209b9c1b51eSKate Stone       if (!subwindow_sp->m_can_activate) {
121044d93782SGreg Clayton         HandleCharResult result = subwindow_sp->HandleChar(key);
121144d93782SGreg Clayton         if (result != eKeyNotHandled)
121244d93782SGreg Clayton           return result;
121344d93782SGreg Clayton       }
121444d93782SGreg Clayton     }
121544d93782SGreg Clayton 
121644d93782SGreg Clayton     return eKeyNotHandled;
121744d93782SGreg Clayton   }
121844d93782SGreg Clayton 
1219b9c1b51eSKate Stone   bool SetActiveWindow(Window *window) {
122044d93782SGreg Clayton     const size_t num_subwindows = m_subwindows.size();
1221b9c1b51eSKate Stone     for (size_t i = 0; i < num_subwindows; ++i) {
1222b9c1b51eSKate Stone       if (m_subwindows[i].get() == window) {
122344d93782SGreg Clayton         m_prev_active_window_idx = m_curr_active_window_idx;
122444d93782SGreg Clayton         ::top_panel(window->m_panel);
122544d93782SGreg Clayton         m_curr_active_window_idx = i;
122644d93782SGreg Clayton         return true;
122744d93782SGreg Clayton       }
122844d93782SGreg Clayton     }
122944d93782SGreg Clayton     return false;
123044d93782SGreg Clayton   }
123144d93782SGreg Clayton 
1232b9c1b51eSKate Stone   WindowSP GetActiveWindow() {
1233b9c1b51eSKate Stone     if (!m_subwindows.empty()) {
1234b9c1b51eSKate Stone       if (m_curr_active_window_idx >= m_subwindows.size()) {
1235b9c1b51eSKate Stone         if (m_prev_active_window_idx < m_subwindows.size()) {
123644d93782SGreg Clayton           m_curr_active_window_idx = m_prev_active_window_idx;
123744d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1238b9c1b51eSKate Stone         } else if (IsActive()) {
123944d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
124044d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
124144d93782SGreg Clayton 
124244d93782SGreg Clayton           // Find first window that wants to be active if this window is active
124344d93782SGreg Clayton           const size_t num_subwindows = m_subwindows.size();
1244b9c1b51eSKate Stone           for (size_t i = 0; i < num_subwindows; ++i) {
1245b9c1b51eSKate Stone             if (m_subwindows[i]->GetCanBeActive()) {
124644d93782SGreg Clayton               m_curr_active_window_idx = i;
124744d93782SGreg Clayton               break;
124844d93782SGreg Clayton             }
124944d93782SGreg Clayton           }
125044d93782SGreg Clayton         }
125144d93782SGreg Clayton       }
125244d93782SGreg Clayton 
125344d93782SGreg Clayton       if (m_curr_active_window_idx < m_subwindows.size())
125444d93782SGreg Clayton         return m_subwindows[m_curr_active_window_idx];
125544d93782SGreg Clayton     }
125644d93782SGreg Clayton     return WindowSP();
125744d93782SGreg Clayton   }
125844d93782SGreg Clayton 
1259b9c1b51eSKate Stone   bool GetCanBeActive() const { return m_can_activate; }
126044d93782SGreg Clayton 
1261b9c1b51eSKate Stone   void SetCanBeActive(bool b) { m_can_activate = b; }
126244d93782SGreg Clayton 
1263b9c1b51eSKate Stone   const WindowDelegateSP &GetDelegate() const { return m_delegate_sp; }
126444d93782SGreg Clayton 
1265b9c1b51eSKate Stone   void SetDelegate(const WindowDelegateSP &delegate_sp) {
126644d93782SGreg Clayton     m_delegate_sp = delegate_sp;
126744d93782SGreg Clayton   }
126844d93782SGreg Clayton 
1269b9c1b51eSKate Stone   Window *GetParent() const { return m_parent; }
127044d93782SGreg Clayton 
1271b9c1b51eSKate Stone   bool IsActive() const {
127244d93782SGreg Clayton     if (m_parent)
127344d93782SGreg Clayton       return m_parent->GetActiveWindow().get() == this;
127444d93782SGreg Clayton     else
127544d93782SGreg Clayton       return true; // Top level window is always active
127644d93782SGreg Clayton   }
127744d93782SGreg Clayton 
1278b9c1b51eSKate Stone   void SelectNextWindowAsActive() {
127944d93782SGreg Clayton     // Move active focus to next window
128044d93782SGreg Clayton     const size_t num_subwindows = m_subwindows.size();
1281b9c1b51eSKate Stone     if (m_curr_active_window_idx == UINT32_MAX) {
128244d93782SGreg Clayton       uint32_t idx = 0;
1283b9c1b51eSKate Stone       for (auto subwindow_sp : m_subwindows) {
1284b9c1b51eSKate Stone         if (subwindow_sp->GetCanBeActive()) {
128544d93782SGreg Clayton           m_curr_active_window_idx = idx;
128644d93782SGreg Clayton           break;
128744d93782SGreg Clayton         }
128844d93782SGreg Clayton         ++idx;
128944d93782SGreg Clayton       }
1290b9c1b51eSKate Stone     } else if (m_curr_active_window_idx + 1 < num_subwindows) {
129144d93782SGreg Clayton       bool handled = false;
129244d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1293b9c1b51eSKate Stone       for (size_t idx = m_curr_active_window_idx + 1; idx < num_subwindows;
1294b9c1b51eSKate Stone            ++idx) {
1295b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
129644d93782SGreg Clayton           m_curr_active_window_idx = idx;
129744d93782SGreg Clayton           handled = true;
129844d93782SGreg Clayton           break;
129944d93782SGreg Clayton         }
130044d93782SGreg Clayton       }
1301b9c1b51eSKate Stone       if (!handled) {
1302b9c1b51eSKate Stone         for (size_t idx = 0; idx <= m_prev_active_window_idx; ++idx) {
1303b9c1b51eSKate Stone           if (m_subwindows[idx]->GetCanBeActive()) {
130444d93782SGreg Clayton             m_curr_active_window_idx = idx;
130544d93782SGreg Clayton             break;
130644d93782SGreg Clayton           }
130744d93782SGreg Clayton         }
130844d93782SGreg Clayton       }
1309b9c1b51eSKate Stone     } else {
131044d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1311b9c1b51eSKate Stone       for (size_t idx = 0; idx < num_subwindows; ++idx) {
1312b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
131344d93782SGreg Clayton           m_curr_active_window_idx = idx;
131444d93782SGreg Clayton           break;
131544d93782SGreg Clayton         }
131644d93782SGreg Clayton       }
131744d93782SGreg Clayton     }
131844d93782SGreg Clayton   }
131944d93782SGreg Clayton 
1320b9c1b51eSKate Stone   const char *GetName() const { return m_name.c_str(); }
1321315b6884SEugene Zelenko 
132244d93782SGreg Clayton protected:
132344d93782SGreg Clayton   std::string m_name;
132444d93782SGreg Clayton   WINDOW *m_window;
132544d93782SGreg Clayton   PANEL *m_panel;
132644d93782SGreg Clayton   Window *m_parent;
132744d93782SGreg Clayton   Windows m_subwindows;
132844d93782SGreg Clayton   WindowDelegateSP m_delegate_sp;
132944d93782SGreg Clayton   uint32_t m_curr_active_window_idx;
133044d93782SGreg Clayton   uint32_t m_prev_active_window_idx;
133144d93782SGreg Clayton   bool m_delete;
133244d93782SGreg Clayton   bool m_needs_update;
133344d93782SGreg Clayton   bool m_can_activate;
133444d93782SGreg Clayton   bool m_is_subwin;
133544d93782SGreg Clayton 
133644d93782SGreg Clayton private:
133744d93782SGreg Clayton   DISALLOW_COPY_AND_ASSIGN(Window);
133844d93782SGreg Clayton };
133944d93782SGreg Clayton 
1340b9c1b51eSKate Stone class MenuDelegate {
134144d93782SGreg Clayton public:
1342315b6884SEugene Zelenko   virtual ~MenuDelegate() = default;
134344d93782SGreg Clayton 
1344b9c1b51eSKate Stone   virtual MenuActionResult MenuDelegateAction(Menu &menu) = 0;
134544d93782SGreg Clayton };
134644d93782SGreg Clayton 
1347b9c1b51eSKate Stone class Menu : public WindowDelegate {
134844d93782SGreg Clayton public:
1349b9c1b51eSKate Stone   enum class Type { Invalid, Bar, Item, Separator };
135044d93782SGreg Clayton 
135144d93782SGreg Clayton   // Menubar or separator constructor
135244d93782SGreg Clayton   Menu(Type type);
135344d93782SGreg Clayton 
135444d93782SGreg Clayton   // Menuitem constructor
1355b9c1b51eSKate Stone   Menu(const char *name, const char *key_name, int key_value,
135644d93782SGreg Clayton        uint64_t identifier);
135744d93782SGreg Clayton 
1358315b6884SEugene Zelenko   ~Menu() override = default;
135944d93782SGreg Clayton 
1360b9c1b51eSKate Stone   const MenuDelegateSP &GetDelegate() const { return m_delegate_sp; }
136144d93782SGreg Clayton 
1362b9c1b51eSKate Stone   void SetDelegate(const MenuDelegateSP &delegate_sp) {
136344d93782SGreg Clayton     m_delegate_sp = delegate_sp;
136444d93782SGreg Clayton   }
136544d93782SGreg Clayton 
1366b9c1b51eSKate Stone   void RecalculateNameLengths();
136744d93782SGreg Clayton 
1368b9c1b51eSKate Stone   void AddSubmenu(const MenuSP &menu_sp);
136944d93782SGreg Clayton 
1370b9c1b51eSKate Stone   int DrawAndRunMenu(Window &window);
137144d93782SGreg Clayton 
1372b9c1b51eSKate Stone   void DrawMenuTitle(Window &window, bool highlight);
137344d93782SGreg Clayton 
1374b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
137544d93782SGreg Clayton 
1376b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
137744d93782SGreg Clayton 
1378b9c1b51eSKate Stone   MenuActionResult ActionPrivate(Menu &menu) {
137944d93782SGreg Clayton     MenuActionResult result = MenuActionResult::NotHandled;
1380b9c1b51eSKate Stone     if (m_delegate_sp) {
138144d93782SGreg Clayton       result = m_delegate_sp->MenuDelegateAction(menu);
138244d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
138344d93782SGreg Clayton         return result;
1384b9c1b51eSKate Stone     } else if (m_parent) {
138544d93782SGreg Clayton       result = m_parent->ActionPrivate(menu);
138644d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
138744d93782SGreg Clayton         return result;
138844d93782SGreg Clayton     }
138944d93782SGreg Clayton     return m_canned_result;
139044d93782SGreg Clayton   }
139144d93782SGreg Clayton 
1392b9c1b51eSKate Stone   MenuActionResult Action() {
139305097246SAdrian Prantl     // Call the recursive action so it can try to handle it with the menu
139405097246SAdrian Prantl     // delegate, and if not, try our parent menu
139544d93782SGreg Clayton     return ActionPrivate(*this);
139644d93782SGreg Clayton   }
139744d93782SGreg Clayton 
1398b9c1b51eSKate Stone   void SetCannedResult(MenuActionResult result) { m_canned_result = result; }
139944d93782SGreg Clayton 
1400b9c1b51eSKate Stone   Menus &GetSubmenus() { return m_submenus; }
140144d93782SGreg Clayton 
1402b9c1b51eSKate Stone   const Menus &GetSubmenus() const { return m_submenus; }
140344d93782SGreg Clayton 
1404b9c1b51eSKate Stone   int GetSelectedSubmenuIndex() const { return m_selected; }
140544d93782SGreg Clayton 
1406b9c1b51eSKate Stone   void SetSelectedSubmenuIndex(int idx) { m_selected = idx; }
140744d93782SGreg Clayton 
1408b9c1b51eSKate Stone   Type GetType() const { return m_type; }
140944d93782SGreg Clayton 
1410b9c1b51eSKate Stone   int GetStartingColumn() const { return m_start_col; }
141144d93782SGreg Clayton 
1412b9c1b51eSKate Stone   void SetStartingColumn(int col) { m_start_col = col; }
141344d93782SGreg Clayton 
1414b9c1b51eSKate Stone   int GetKeyValue() const { return m_key_value; }
141544d93782SGreg Clayton 
1416b9c1b51eSKate Stone   void SetKeyValue(int key_value) { m_key_value = key_value; }
141744d93782SGreg Clayton 
1418b9c1b51eSKate Stone   std::string &GetName() { return m_name; }
141944d93782SGreg Clayton 
1420b9c1b51eSKate Stone   std::string &GetKeyName() { return m_key_name; }
142144d93782SGreg Clayton 
1422b9c1b51eSKate Stone   int GetDrawWidth() const {
142344d93782SGreg Clayton     return m_max_submenu_name_length + m_max_submenu_key_name_length + 8;
142444d93782SGreg Clayton   }
142544d93782SGreg Clayton 
1426b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
142744d93782SGreg Clayton 
1428b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
142944d93782SGreg Clayton 
143044d93782SGreg Clayton protected:
143144d93782SGreg Clayton   std::string m_name;
143244d93782SGreg Clayton   std::string m_key_name;
143344d93782SGreg Clayton   uint64_t m_identifier;
143444d93782SGreg Clayton   Type m_type;
143544d93782SGreg Clayton   int m_key_value;
143644d93782SGreg Clayton   int m_start_col;
143744d93782SGreg Clayton   int m_max_submenu_name_length;
143844d93782SGreg Clayton   int m_max_submenu_key_name_length;
143944d93782SGreg Clayton   int m_selected;
144044d93782SGreg Clayton   Menu *m_parent;
144144d93782SGreg Clayton   Menus m_submenus;
144244d93782SGreg Clayton   WindowSP m_menu_window_sp;
144344d93782SGreg Clayton   MenuActionResult m_canned_result;
144444d93782SGreg Clayton   MenuDelegateSP m_delegate_sp;
144544d93782SGreg Clayton };
144644d93782SGreg Clayton 
144744d93782SGreg Clayton // Menubar or separator constructor
1448b9c1b51eSKate Stone Menu::Menu(Type type)
1449b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(0), m_type(type), m_key_value(0),
1450b9c1b51eSKate Stone       m_start_col(0), m_max_submenu_name_length(0),
1451b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1452b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1453b9c1b51eSKate Stone       m_delegate_sp() {}
145444d93782SGreg Clayton 
145544d93782SGreg Clayton // Menuitem constructor
1456b9c1b51eSKate Stone Menu::Menu(const char *name, const char *key_name, int key_value,
1457b9c1b51eSKate Stone            uint64_t identifier)
1458b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(identifier), m_type(Type::Invalid),
1459b9c1b51eSKate Stone       m_key_value(key_value), m_start_col(0), m_max_submenu_name_length(0),
1460b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1461b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1462b9c1b51eSKate Stone       m_delegate_sp() {
1463b9c1b51eSKate Stone   if (name && name[0]) {
146444d93782SGreg Clayton     m_name = name;
146544d93782SGreg Clayton     m_type = Type::Item;
146644d93782SGreg Clayton     if (key_name && key_name[0])
146744d93782SGreg Clayton       m_key_name = key_name;
1468b9c1b51eSKate Stone   } else {
146944d93782SGreg Clayton     m_type = Type::Separator;
147044d93782SGreg Clayton   }
147144d93782SGreg Clayton }
147244d93782SGreg Clayton 
1473b9c1b51eSKate Stone void Menu::RecalculateNameLengths() {
147444d93782SGreg Clayton   m_max_submenu_name_length = 0;
147544d93782SGreg Clayton   m_max_submenu_key_name_length = 0;
147644d93782SGreg Clayton   Menus &submenus = GetSubmenus();
147744d93782SGreg Clayton   const size_t num_submenus = submenus.size();
1478b9c1b51eSKate Stone   for (size_t i = 0; i < num_submenus; ++i) {
147944d93782SGreg Clayton     Menu *submenu = submenus[i].get();
14803985c8c6SSaleem Abdulrasool     if (static_cast<size_t>(m_max_submenu_name_length) < submenu->m_name.size())
148144d93782SGreg Clayton       m_max_submenu_name_length = submenu->m_name.size();
1482b9c1b51eSKate Stone     if (static_cast<size_t>(m_max_submenu_key_name_length) <
1483b9c1b51eSKate Stone         submenu->m_key_name.size())
148444d93782SGreg Clayton       m_max_submenu_key_name_length = submenu->m_key_name.size();
148544d93782SGreg Clayton   }
148644d93782SGreg Clayton }
148744d93782SGreg Clayton 
1488b9c1b51eSKate Stone void Menu::AddSubmenu(const MenuSP &menu_sp) {
148944d93782SGreg Clayton   menu_sp->m_parent = this;
14903985c8c6SSaleem Abdulrasool   if (static_cast<size_t>(m_max_submenu_name_length) < menu_sp->m_name.size())
149144d93782SGreg Clayton     m_max_submenu_name_length = menu_sp->m_name.size();
1492b9c1b51eSKate Stone   if (static_cast<size_t>(m_max_submenu_key_name_length) <
1493b9c1b51eSKate Stone       menu_sp->m_key_name.size())
149444d93782SGreg Clayton     m_max_submenu_key_name_length = menu_sp->m_key_name.size();
149544d93782SGreg Clayton   m_submenus.push_back(menu_sp);
149644d93782SGreg Clayton }
149744d93782SGreg Clayton 
1498b9c1b51eSKate Stone void Menu::DrawMenuTitle(Window &window, bool highlight) {
1499b9c1b51eSKate Stone   if (m_type == Type::Separator) {
150044d93782SGreg Clayton     window.MoveCursor(0, window.GetCursorY());
150144d93782SGreg Clayton     window.PutChar(ACS_LTEE);
150244d93782SGreg Clayton     int width = window.GetWidth();
1503b9c1b51eSKate Stone     if (width > 2) {
150444d93782SGreg Clayton       width -= 2;
15053985c8c6SSaleem Abdulrasool       for (int i = 0; i < width; ++i)
150644d93782SGreg Clayton         window.PutChar(ACS_HLINE);
150744d93782SGreg Clayton     }
150844d93782SGreg Clayton     window.PutChar(ACS_RTEE);
1509b9c1b51eSKate Stone   } else {
151044d93782SGreg Clayton     const int shortcut_key = m_key_value;
151144d93782SGreg Clayton     bool underlined_shortcut = false;
151244d93782SGreg Clayton     const attr_t hilgight_attr = A_REVERSE;
151344d93782SGreg Clayton     if (highlight)
151444d93782SGreg Clayton       window.AttributeOn(hilgight_attr);
1515b9c1b51eSKate Stone     if (isprint(shortcut_key)) {
151644d93782SGreg Clayton       size_t lower_pos = m_name.find(tolower(shortcut_key));
151744d93782SGreg Clayton       size_t upper_pos = m_name.find(toupper(shortcut_key));
151844d93782SGreg Clayton       const char *name = m_name.c_str();
151944d93782SGreg Clayton       size_t pos = std::min<size_t>(lower_pos, upper_pos);
1520b9c1b51eSKate Stone       if (pos != std::string::npos) {
152144d93782SGreg Clayton         underlined_shortcut = true;
1522b9c1b51eSKate Stone         if (pos > 0) {
152344d93782SGreg Clayton           window.PutCString(name, pos);
152444d93782SGreg Clayton           name += pos;
152544d93782SGreg Clayton         }
152644d93782SGreg Clayton         const attr_t shortcut_attr = A_UNDERLINE | A_BOLD;
152744d93782SGreg Clayton         window.AttributeOn(shortcut_attr);
152844d93782SGreg Clayton         window.PutChar(name[0]);
152944d93782SGreg Clayton         window.AttributeOff(shortcut_attr);
153044d93782SGreg Clayton         name++;
153144d93782SGreg Clayton         if (name[0])
153244d93782SGreg Clayton           window.PutCString(name);
153344d93782SGreg Clayton       }
153444d93782SGreg Clayton     }
153544d93782SGreg Clayton 
1536b9c1b51eSKate Stone     if (!underlined_shortcut) {
153744d93782SGreg Clayton       window.PutCString(m_name.c_str());
153844d93782SGreg Clayton     }
153944d93782SGreg Clayton 
154044d93782SGreg Clayton     if (highlight)
154144d93782SGreg Clayton       window.AttributeOff(hilgight_attr);
154244d93782SGreg Clayton 
1543b9c1b51eSKate Stone     if (m_key_name.empty()) {
1544b9c1b51eSKate Stone       if (!underlined_shortcut && isprint(m_key_value)) {
154544d93782SGreg Clayton         window.AttributeOn(COLOR_PAIR(3));
154644d93782SGreg Clayton         window.Printf(" (%c)", m_key_value);
154744d93782SGreg Clayton         window.AttributeOff(COLOR_PAIR(3));
154844d93782SGreg Clayton       }
1549b9c1b51eSKate Stone     } else {
155044d93782SGreg Clayton       window.AttributeOn(COLOR_PAIR(3));
155144d93782SGreg Clayton       window.Printf(" (%s)", m_key_name.c_str());
155244d93782SGreg Clayton       window.AttributeOff(COLOR_PAIR(3));
155344d93782SGreg Clayton     }
155444d93782SGreg Clayton   }
155544d93782SGreg Clayton }
155644d93782SGreg Clayton 
1557b9c1b51eSKate Stone bool Menu::WindowDelegateDraw(Window &window, bool force) {
155844d93782SGreg Clayton   Menus &submenus = GetSubmenus();
155944d93782SGreg Clayton   const size_t num_submenus = submenus.size();
156044d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
156144d93782SGreg Clayton   Menu::Type menu_type = GetType();
1562b9c1b51eSKate Stone   switch (menu_type) {
1563b9c1b51eSKate Stone   case Menu::Type::Bar: {
156444d93782SGreg Clayton     window.SetBackground(2);
156544d93782SGreg Clayton     window.MoveCursor(0, 0);
1566b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
156744d93782SGreg Clayton       Menu *menu = submenus[i].get();
156844d93782SGreg Clayton       if (i > 0)
156944d93782SGreg Clayton         window.PutChar(' ');
157044d93782SGreg Clayton       menu->SetStartingColumn(window.GetCursorX());
157144d93782SGreg Clayton       window.PutCString("| ");
157244d93782SGreg Clayton       menu->DrawMenuTitle(window, false);
157344d93782SGreg Clayton     }
157444d93782SGreg Clayton     window.PutCString(" |");
157544d93782SGreg Clayton     window.DeferredRefresh();
1576b9c1b51eSKate Stone   } break;
157744d93782SGreg Clayton 
1578b9c1b51eSKate Stone   case Menu::Type::Item: {
157944d93782SGreg Clayton     int y = 1;
158044d93782SGreg Clayton     int x = 3;
158144d93782SGreg Clayton     // Draw the menu
158244d93782SGreg Clayton     int cursor_x = 0;
158344d93782SGreg Clayton     int cursor_y = 0;
158444d93782SGreg Clayton     window.Erase();
158544d93782SGreg Clayton     window.SetBackground(2);
158644d93782SGreg Clayton     window.Box();
1587b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
1588b9c1b51eSKate Stone       const bool is_selected = (i == static_cast<size_t>(selected_idx));
158944d93782SGreg Clayton       window.MoveCursor(x, y + i);
1590b9c1b51eSKate Stone       if (is_selected) {
159144d93782SGreg Clayton         // Remember where we want the cursor to be
159244d93782SGreg Clayton         cursor_x = x - 1;
159344d93782SGreg Clayton         cursor_y = y + i;
159444d93782SGreg Clayton       }
159544d93782SGreg Clayton       submenus[i]->DrawMenuTitle(window, is_selected);
159644d93782SGreg Clayton     }
159744d93782SGreg Clayton     window.MoveCursor(cursor_x, cursor_y);
159844d93782SGreg Clayton     window.DeferredRefresh();
1599b9c1b51eSKate Stone   } break;
160044d93782SGreg Clayton 
160144d93782SGreg Clayton   default:
160244d93782SGreg Clayton   case Menu::Type::Separator:
160344d93782SGreg Clayton     break;
160444d93782SGreg Clayton   }
160544d93782SGreg Clayton   return true; // Drawing handled...
160644d93782SGreg Clayton }
160744d93782SGreg Clayton 
1608b9c1b51eSKate Stone HandleCharResult Menu::WindowDelegateHandleChar(Window &window, int key) {
160944d93782SGreg Clayton   HandleCharResult result = eKeyNotHandled;
161044d93782SGreg Clayton 
161144d93782SGreg Clayton   Menus &submenus = GetSubmenus();
161244d93782SGreg Clayton   const size_t num_submenus = submenus.size();
161344d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
161444d93782SGreg Clayton   Menu::Type menu_type = GetType();
1615b9c1b51eSKate Stone   if (menu_type == Menu::Type::Bar) {
161644d93782SGreg Clayton     MenuSP run_menu_sp;
1617b9c1b51eSKate Stone     switch (key) {
161844d93782SGreg Clayton     case KEY_DOWN:
161944d93782SGreg Clayton     case KEY_UP:
162044d93782SGreg Clayton       // Show last menu or first menu
16213985c8c6SSaleem Abdulrasool       if (selected_idx < static_cast<int>(num_submenus))
162244d93782SGreg Clayton         run_menu_sp = submenus[selected_idx];
162344d93782SGreg Clayton       else if (!submenus.empty())
162444d93782SGreg Clayton         run_menu_sp = submenus.front();
162544d93782SGreg Clayton       result = eKeyHandled;
162644d93782SGreg Clayton       break;
162744d93782SGreg Clayton 
162844d93782SGreg Clayton     case KEY_RIGHT:
162944d93782SGreg Clayton       ++m_selected;
16303985c8c6SSaleem Abdulrasool       if (m_selected >= static_cast<int>(num_submenus))
163144d93782SGreg Clayton         m_selected = 0;
16323985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
163344d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
163444d93782SGreg Clayton       else if (!submenus.empty())
163544d93782SGreg Clayton         run_menu_sp = submenus.front();
163644d93782SGreg Clayton       result = eKeyHandled;
163744d93782SGreg Clayton       break;
163844d93782SGreg Clayton 
163944d93782SGreg Clayton     case KEY_LEFT:
164044d93782SGreg Clayton       --m_selected;
164144d93782SGreg Clayton       if (m_selected < 0)
164244d93782SGreg Clayton         m_selected = num_submenus - 1;
16433985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
164444d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
164544d93782SGreg Clayton       else if (!submenus.empty())
164644d93782SGreg Clayton         run_menu_sp = submenus.front();
164744d93782SGreg Clayton       result = eKeyHandled;
164844d93782SGreg Clayton       break;
164944d93782SGreg Clayton 
165044d93782SGreg Clayton     default:
1651b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
1652b9c1b51eSKate Stone         if (submenus[i]->GetKeyValue() == key) {
165344d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
165444d93782SGreg Clayton           run_menu_sp = submenus[i];
165544d93782SGreg Clayton           result = eKeyHandled;
165644d93782SGreg Clayton           break;
165744d93782SGreg Clayton         }
165844d93782SGreg Clayton       }
165944d93782SGreg Clayton       break;
166044d93782SGreg Clayton     }
166144d93782SGreg Clayton 
1662b9c1b51eSKate Stone     if (run_menu_sp) {
166305097246SAdrian Prantl       // Run the action on this menu in case we need to populate the menu with
166405097246SAdrian Prantl       // dynamic content and also in case check marks, and any other menu
166505097246SAdrian Prantl       // decorations need to be calculated
166644d93782SGreg Clayton       if (run_menu_sp->Action() == MenuActionResult::Quit)
166744d93782SGreg Clayton         return eQuitApplication;
166844d93782SGreg Clayton 
166944d93782SGreg Clayton       Rect menu_bounds;
167044d93782SGreg Clayton       menu_bounds.origin.x = run_menu_sp->GetStartingColumn();
167144d93782SGreg Clayton       menu_bounds.origin.y = 1;
167244d93782SGreg Clayton       menu_bounds.size.width = run_menu_sp->GetDrawWidth();
167344d93782SGreg Clayton       menu_bounds.size.height = run_menu_sp->GetSubmenus().size() + 2;
167444d93782SGreg Clayton       if (m_menu_window_sp)
167544d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(m_menu_window_sp.get());
167644d93782SGreg Clayton 
1677b9c1b51eSKate Stone       m_menu_window_sp = window.GetParent()->CreateSubWindow(
1678b9c1b51eSKate Stone           run_menu_sp->GetName().c_str(), menu_bounds, true);
167944d93782SGreg Clayton       m_menu_window_sp->SetDelegate(run_menu_sp);
168044d93782SGreg Clayton     }
1681b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Item) {
1682b9c1b51eSKate Stone     switch (key) {
168344d93782SGreg Clayton     case KEY_DOWN:
1684b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
168544d93782SGreg Clayton         const int start_select = m_selected;
1686b9c1b51eSKate Stone         while (++m_selected != start_select) {
16873985c8c6SSaleem Abdulrasool           if (static_cast<size_t>(m_selected) >= num_submenus)
168844d93782SGreg Clayton             m_selected = 0;
168944d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
169044d93782SGreg Clayton             continue;
169144d93782SGreg Clayton           else
169244d93782SGreg Clayton             break;
169344d93782SGreg Clayton         }
169444d93782SGreg Clayton         return eKeyHandled;
169544d93782SGreg Clayton       }
169644d93782SGreg Clayton       break;
169744d93782SGreg Clayton 
169844d93782SGreg Clayton     case KEY_UP:
1699b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
170044d93782SGreg Clayton         const int start_select = m_selected;
1701b9c1b51eSKate Stone         while (--m_selected != start_select) {
17023985c8c6SSaleem Abdulrasool           if (m_selected < static_cast<int>(0))
170344d93782SGreg Clayton             m_selected = num_submenus - 1;
170444d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
170544d93782SGreg Clayton             continue;
170644d93782SGreg Clayton           else
170744d93782SGreg Clayton             break;
170844d93782SGreg Clayton         }
170944d93782SGreg Clayton         return eKeyHandled;
171044d93782SGreg Clayton       }
171144d93782SGreg Clayton       break;
171244d93782SGreg Clayton 
171344d93782SGreg Clayton     case KEY_RETURN:
1714b9c1b51eSKate Stone       if (static_cast<size_t>(selected_idx) < num_submenus) {
171544d93782SGreg Clayton         if (submenus[selected_idx]->Action() == MenuActionResult::Quit)
171644d93782SGreg Clayton           return eQuitApplication;
171744d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(&window);
171844d93782SGreg Clayton         return eKeyHandled;
171944d93782SGreg Clayton       }
172044d93782SGreg Clayton       break;
172144d93782SGreg Clayton 
1722b9c1b51eSKate Stone     case KEY_ESCAPE: // Beware: pressing escape key has 1 to 2 second delay in
1723b9c1b51eSKate Stone                      // case other chars are entered for escaped sequences
172444d93782SGreg Clayton       window.GetParent()->RemoveSubWindow(&window);
172544d93782SGreg Clayton       return eKeyHandled;
172644d93782SGreg Clayton 
172744d93782SGreg Clayton     default:
1728b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
172944d93782SGreg Clayton         Menu *menu = submenus[i].get();
1730b9c1b51eSKate Stone         if (menu->GetKeyValue() == key) {
173144d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
173244d93782SGreg Clayton           window.GetParent()->RemoveSubWindow(&window);
173344d93782SGreg Clayton           if (menu->Action() == MenuActionResult::Quit)
173444d93782SGreg Clayton             return eQuitApplication;
173544d93782SGreg Clayton           return eKeyHandled;
173644d93782SGreg Clayton         }
173744d93782SGreg Clayton       }
173844d93782SGreg Clayton       break;
173944d93782SGreg Clayton     }
1740b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Separator) {
174144d93782SGreg Clayton   }
174244d93782SGreg Clayton   return result;
174344d93782SGreg Clayton }
174444d93782SGreg Clayton 
1745b9c1b51eSKate Stone class Application {
174644d93782SGreg Clayton public:
1747b9c1b51eSKate Stone   Application(FILE *in, FILE *out)
1748b9c1b51eSKate Stone       : m_window_sp(), m_screen(nullptr), m_in(in), m_out(out) {}
174944d93782SGreg Clayton 
1750b9c1b51eSKate Stone   ~Application() {
175144d93782SGreg Clayton     m_window_delegates.clear();
175244d93782SGreg Clayton     m_window_sp.reset();
1753b9c1b51eSKate Stone     if (m_screen) {
175444d93782SGreg Clayton       ::delscreen(m_screen);
1755c5dac77aSEugene Zelenko       m_screen = nullptr;
175644d93782SGreg Clayton     }
175744d93782SGreg Clayton   }
175844d93782SGreg Clayton 
1759b9c1b51eSKate Stone   void Initialize() {
176044d93782SGreg Clayton     ::setlocale(LC_ALL, "");
176144d93782SGreg Clayton     ::setlocale(LC_CTYPE, "");
1762c5dac77aSEugene Zelenko     m_screen = ::newterm(nullptr, m_out, m_in);
176344d93782SGreg Clayton     ::start_color();
176444d93782SGreg Clayton     ::curs_set(0);
176544d93782SGreg Clayton     ::noecho();
176644d93782SGreg Clayton     ::keypad(stdscr, TRUE);
176744d93782SGreg Clayton   }
176844d93782SGreg Clayton 
1769b9c1b51eSKate Stone   void Terminate() { ::endwin(); }
177044d93782SGreg Clayton 
1771b9c1b51eSKate Stone   void Run(Debugger &debugger) {
177244d93782SGreg Clayton     bool done = false;
177344d93782SGreg Clayton     int delay_in_tenths_of_a_second = 1;
177444d93782SGreg Clayton 
177505097246SAdrian Prantl     // Alas the threading model in curses is a bit lame so we need to resort to
177605097246SAdrian Prantl     // polling every 0.5 seconds. We could poll for stdin ourselves and then
177705097246SAdrian Prantl     // pass the keys down but then we need to translate all of the escape
177805097246SAdrian Prantl     // sequences ourselves. So we resort to polling for input because we need
177905097246SAdrian Prantl     // to receive async process events while in this loop.
178044d93782SGreg Clayton 
1781b9c1b51eSKate Stone     halfdelay(delay_in_tenths_of_a_second); // Poll using some number of tenths
1782b9c1b51eSKate Stone                                             // of seconds seconds when calling
1783b9c1b51eSKate Stone                                             // Window::GetChar()
178444d93782SGreg Clayton 
1785b9c1b51eSKate Stone     ListenerSP listener_sp(
1786b9c1b51eSKate Stone         Listener::MakeListener("lldb.IOHandler.curses.Application"));
178744d93782SGreg Clayton     ConstString broadcaster_class_target(Target::GetStaticBroadcasterClass());
178844d93782SGreg Clayton     ConstString broadcaster_class_process(Process::GetStaticBroadcasterClass());
178944d93782SGreg Clayton     ConstString broadcaster_class_thread(Thread::GetStaticBroadcasterClass());
179044d93782SGreg Clayton     debugger.EnableForwardEvents(listener_sp);
179144d93782SGreg Clayton 
179244d93782SGreg Clayton     bool update = true;
179344d93782SGreg Clayton #if defined(__APPLE__)
179444d93782SGreg Clayton     std::deque<int> escape_chars;
179544d93782SGreg Clayton #endif
179644d93782SGreg Clayton 
1797b9c1b51eSKate Stone     while (!done) {
1798b9c1b51eSKate Stone       if (update) {
179944d93782SGreg Clayton         m_window_sp->Draw(false);
180005097246SAdrian Prantl         // All windows should be calling Window::DeferredRefresh() instead of
180105097246SAdrian Prantl         // Window::Refresh() so we can do a single update and avoid any screen
180205097246SAdrian Prantl         // blinking
180344d93782SGreg Clayton         update_panels();
180444d93782SGreg Clayton 
1805b9c1b51eSKate Stone         // Cursor hiding isn't working on MacOSX, so hide it in the top left
1806b9c1b51eSKate Stone         // corner
180744d93782SGreg Clayton         m_window_sp->MoveCursor(0, 0);
180844d93782SGreg Clayton 
180944d93782SGreg Clayton         doupdate();
181044d93782SGreg Clayton         update = false;
181144d93782SGreg Clayton       }
181244d93782SGreg Clayton 
181344d93782SGreg Clayton #if defined(__APPLE__)
181405097246SAdrian Prantl       // Terminal.app doesn't map its function keys correctly, F1-F4 default
181505097246SAdrian Prantl       // to: \033OP, \033OQ, \033OR, \033OS, so lets take care of this here if
1816b9c1b51eSKate Stone       // possible
181744d93782SGreg Clayton       int ch;
181844d93782SGreg Clayton       if (escape_chars.empty())
181944d93782SGreg Clayton         ch = m_window_sp->GetChar();
1820b9c1b51eSKate Stone       else {
182144d93782SGreg Clayton         ch = escape_chars.front();
182244d93782SGreg Clayton         escape_chars.pop_front();
182344d93782SGreg Clayton       }
1824b9c1b51eSKate Stone       if (ch == KEY_ESCAPE) {
182544d93782SGreg Clayton         int ch2 = m_window_sp->GetChar();
1826b9c1b51eSKate Stone         if (ch2 == 'O') {
182744d93782SGreg Clayton           int ch3 = m_window_sp->GetChar();
1828b9c1b51eSKate Stone           switch (ch3) {
1829b9c1b51eSKate Stone           case 'P':
1830b9c1b51eSKate Stone             ch = KEY_F(1);
1831b9c1b51eSKate Stone             break;
1832b9c1b51eSKate Stone           case 'Q':
1833b9c1b51eSKate Stone             ch = KEY_F(2);
1834b9c1b51eSKate Stone             break;
1835b9c1b51eSKate Stone           case 'R':
1836b9c1b51eSKate Stone             ch = KEY_F(3);
1837b9c1b51eSKate Stone             break;
1838b9c1b51eSKate Stone           case 'S':
1839b9c1b51eSKate Stone             ch = KEY_F(4);
1840b9c1b51eSKate Stone             break;
184144d93782SGreg Clayton           default:
184244d93782SGreg Clayton             escape_chars.push_back(ch2);
184344d93782SGreg Clayton             if (ch3 != -1)
184444d93782SGreg Clayton               escape_chars.push_back(ch3);
184544d93782SGreg Clayton             break;
184644d93782SGreg Clayton           }
1847b9c1b51eSKate Stone         } else if (ch2 != -1)
184844d93782SGreg Clayton           escape_chars.push_back(ch2);
184944d93782SGreg Clayton       }
185044d93782SGreg Clayton #else
185144d93782SGreg Clayton       int ch = m_window_sp->GetChar();
185244d93782SGreg Clayton 
185344d93782SGreg Clayton #endif
1854b9c1b51eSKate Stone       if (ch == -1) {
1855b9c1b51eSKate Stone         if (feof(m_in) || ferror(m_in)) {
185644d93782SGreg Clayton           done = true;
1857b9c1b51eSKate Stone         } else {
185844d93782SGreg Clayton           // Just a timeout from using halfdelay(), check for events
185944d93782SGreg Clayton           EventSP event_sp;
1860b9c1b51eSKate Stone           while (listener_sp->PeekAtNextEvent()) {
1861d35031e1SPavel Labath             listener_sp->GetEvent(event_sp, std::chrono::seconds(0));
186244d93782SGreg Clayton 
1863b9c1b51eSKate Stone             if (event_sp) {
186444d93782SGreg Clayton               Broadcaster *broadcaster = event_sp->GetBroadcaster();
1865b9c1b51eSKate Stone               if (broadcaster) {
186644d93782SGreg Clayton                 // uint32_t event_type = event_sp->GetType();
1867b9c1b51eSKate Stone                 ConstString broadcaster_class(
1868b9c1b51eSKate Stone                     broadcaster->GetBroadcasterClass());
1869b9c1b51eSKate Stone                 if (broadcaster_class == broadcaster_class_process) {
1870b9c1b51eSKate Stone                   debugger.GetCommandInterpreter().UpdateExecutionContext(
1871b9c1b51eSKate Stone                       nullptr);
187244d93782SGreg Clayton                   update = true;
187344d93782SGreg Clayton                   continue; // Don't get any key, just update our view
187444d93782SGreg Clayton                 }
187544d93782SGreg Clayton               }
187644d93782SGreg Clayton             }
187744d93782SGreg Clayton           }
187844d93782SGreg Clayton         }
1879b9c1b51eSKate Stone       } else {
188044d93782SGreg Clayton         HandleCharResult key_result = m_window_sp->HandleChar(ch);
1881b9c1b51eSKate Stone         switch (key_result) {
188244d93782SGreg Clayton         case eKeyHandled:
1883c5dac77aSEugene Zelenko           debugger.GetCommandInterpreter().UpdateExecutionContext(nullptr);
188444d93782SGreg Clayton           update = true;
188544d93782SGreg Clayton           break;
188644d93782SGreg Clayton         case eKeyNotHandled:
188744d93782SGreg Clayton           break;
188844d93782SGreg Clayton         case eQuitApplication:
188944d93782SGreg Clayton           done = true;
189044d93782SGreg Clayton           break;
189144d93782SGreg Clayton         }
189244d93782SGreg Clayton       }
189344d93782SGreg Clayton     }
189444d93782SGreg Clayton 
189544d93782SGreg Clayton     debugger.CancelForwardEvents(listener_sp);
189644d93782SGreg Clayton   }
189744d93782SGreg Clayton 
1898b9c1b51eSKate Stone   WindowSP &GetMainWindow() {
189944d93782SGreg Clayton     if (!m_window_sp)
1900796ac80bSJonas Devlieghere       m_window_sp = std::make_shared<Window>("main", stdscr, false);
190144d93782SGreg Clayton     return m_window_sp;
190244d93782SGreg Clayton   }
190344d93782SGreg Clayton 
1904b9c1b51eSKate Stone   WindowDelegates &GetWindowDelegates() { return m_window_delegates; }
190544d93782SGreg Clayton 
190644d93782SGreg Clayton protected:
190744d93782SGreg Clayton   WindowSP m_window_sp;
190844d93782SGreg Clayton   WindowDelegates m_window_delegates;
190944d93782SGreg Clayton   SCREEN *m_screen;
191044d93782SGreg Clayton   FILE *m_in;
191144d93782SGreg Clayton   FILE *m_out;
191244d93782SGreg Clayton };
191344d93782SGreg Clayton 
191444d93782SGreg Clayton } // namespace curses
191544d93782SGreg Clayton 
191644d93782SGreg Clayton using namespace curses;
191744d93782SGreg Clayton 
1918b9c1b51eSKate Stone struct Row {
19198369b28dSGreg Clayton   ValueObjectManager value;
192044d93782SGreg Clayton   Row *parent;
19218369b28dSGreg Clayton   // The process stop ID when the children were calculated.
19228369b28dSGreg Clayton   uint32_t children_stop_id;
192344d93782SGreg Clayton   int row_idx;
192444d93782SGreg Clayton   int x;
192544d93782SGreg Clayton   int y;
192644d93782SGreg Clayton   bool might_have_children;
192744d93782SGreg Clayton   bool expanded;
192844d93782SGreg Clayton   bool calculated_children;
192944d93782SGreg Clayton   std::vector<Row> children;
193044d93782SGreg Clayton 
1931b9c1b51eSKate Stone   Row(const ValueObjectSP &v, Row *p)
19328369b28dSGreg Clayton       : value(v, lldb::eDynamicDontRunTarget, true), parent(p), row_idx(0),
19338369b28dSGreg Clayton         x(1), y(1), might_have_children(v ? v->MightHaveChildren() : false),
1934b9c1b51eSKate Stone         expanded(false), calculated_children(false), children() {}
193544d93782SGreg Clayton 
1936b9c1b51eSKate Stone   size_t GetDepth() const {
193744d93782SGreg Clayton     if (parent)
193844d93782SGreg Clayton       return 1 + parent->GetDepth();
193944d93782SGreg Clayton     return 0;
194044d93782SGreg Clayton   }
194144d93782SGreg Clayton 
1942b9c1b51eSKate Stone   void Expand() {
194344d93782SGreg Clayton     expanded = true;
19448369b28dSGreg Clayton   }
19458369b28dSGreg Clayton 
19468369b28dSGreg Clayton   std::vector<Row> &GetChildren() {
19478369b28dSGreg Clayton     ProcessSP process_sp = value.GetProcessSP();
19488369b28dSGreg Clayton     auto stop_id = process_sp->GetStopID();
19498369b28dSGreg Clayton     if (process_sp && stop_id != children_stop_id) {
19508369b28dSGreg Clayton       children_stop_id = stop_id;
19518369b28dSGreg Clayton       calculated_children = false;
19528369b28dSGreg Clayton     }
1953b9c1b51eSKate Stone     if (!calculated_children) {
19548369b28dSGreg Clayton       children.clear();
195544d93782SGreg Clayton       calculated_children = true;
19568369b28dSGreg Clayton       ValueObjectSP valobj = value.GetSP();
1957b9c1b51eSKate Stone       if (valobj) {
195844d93782SGreg Clayton         const size_t num_children = valobj->GetNumChildren();
1959b9c1b51eSKate Stone         for (size_t i = 0; i < num_children; ++i) {
196044d93782SGreg Clayton           children.push_back(Row(valobj->GetChildAtIndex(i, true), this));
196144d93782SGreg Clayton         }
196244d93782SGreg Clayton       }
196344d93782SGreg Clayton     }
19648369b28dSGreg Clayton     return children;
196544d93782SGreg Clayton   }
196644d93782SGreg Clayton 
19678369b28dSGreg Clayton   void Unexpand() {
19688369b28dSGreg Clayton     expanded = false;
19698369b28dSGreg Clayton     calculated_children = false;
19708369b28dSGreg Clayton     children.clear();
19718369b28dSGreg Clayton   }
197244d93782SGreg Clayton 
1973b9c1b51eSKate Stone   void DrawTree(Window &window) {
197444d93782SGreg Clayton     if (parent)
197544d93782SGreg Clayton       parent->DrawTreeForChild(window, this, 0);
197644d93782SGreg Clayton 
1977b9c1b51eSKate Stone     if (might_have_children) {
197805097246SAdrian Prantl       // It we can get UTF8 characters to work we should try to use the
197905097246SAdrian Prantl       // "symbol" UTF8 string below
198044d93782SGreg Clayton       //            const char *symbol = "";
198144d93782SGreg Clayton       //            if (row.expanded)
198244d93782SGreg Clayton       //                symbol = "\xe2\x96\xbd ";
198344d93782SGreg Clayton       //            else
198444d93782SGreg Clayton       //                symbol = "\xe2\x96\xb7 ";
198544d93782SGreg Clayton       //            window.PutCString (symbol);
198644d93782SGreg Clayton 
198705097246SAdrian Prantl       // The ACS_DARROW and ACS_RARROW don't look very nice they are just a 'v'
198805097246SAdrian Prantl       // or '>' character...
198944d93782SGreg Clayton       //            if (expanded)
199044d93782SGreg Clayton       //                window.PutChar (ACS_DARROW);
199144d93782SGreg Clayton       //            else
199244d93782SGreg Clayton       //                window.PutChar (ACS_RARROW);
199305097246SAdrian Prantl       // Since we can't find any good looking right arrow/down arrow symbols,
199405097246SAdrian Prantl       // just use a diamond...
199544d93782SGreg Clayton       window.PutChar(ACS_DIAMOND);
199644d93782SGreg Clayton       window.PutChar(ACS_HLINE);
199744d93782SGreg Clayton     }
199844d93782SGreg Clayton   }
199944d93782SGreg Clayton 
2000b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, Row *child, uint32_t reverse_depth) {
200144d93782SGreg Clayton     if (parent)
200244d93782SGreg Clayton       parent->DrawTreeForChild(window, this, reverse_depth + 1);
200344d93782SGreg Clayton 
20048369b28dSGreg Clayton     if (&GetChildren().back() == child) {
200544d93782SGreg Clayton       // Last child
2006b9c1b51eSKate Stone       if (reverse_depth == 0) {
200744d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
200844d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2009b9c1b51eSKate Stone       } else {
201044d93782SGreg Clayton         window.PutChar(' ');
201144d93782SGreg Clayton         window.PutChar(' ');
201244d93782SGreg Clayton       }
2013b9c1b51eSKate Stone     } else {
2014b9c1b51eSKate Stone       if (reverse_depth == 0) {
201544d93782SGreg Clayton         window.PutChar(ACS_LTEE);
201644d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2017b9c1b51eSKate Stone       } else {
201844d93782SGreg Clayton         window.PutChar(ACS_VLINE);
201944d93782SGreg Clayton         window.PutChar(' ');
202044d93782SGreg Clayton       }
202144d93782SGreg Clayton     }
202244d93782SGreg Clayton   }
202344d93782SGreg Clayton };
202444d93782SGreg Clayton 
2025b9c1b51eSKate Stone struct DisplayOptions {
202644d93782SGreg Clayton   bool show_types;
202744d93782SGreg Clayton };
202844d93782SGreg Clayton 
202944d93782SGreg Clayton class TreeItem;
203044d93782SGreg Clayton 
2031b9c1b51eSKate Stone class TreeDelegate {
203244d93782SGreg Clayton public:
2033c5dac77aSEugene Zelenko   TreeDelegate() = default;
2034315b6884SEugene Zelenko   virtual ~TreeDelegate() = default;
2035315b6884SEugene Zelenko 
203644d93782SGreg Clayton   virtual void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) = 0;
203744d93782SGreg Clayton   virtual void TreeDelegateGenerateChildren(TreeItem &item) = 0;
2038b9c1b51eSKate Stone   virtual bool TreeDelegateItemSelected(
2039b9c1b51eSKate Stone       TreeItem &item) = 0; // Return true if we need to update views
204044d93782SGreg Clayton };
2041315b6884SEugene Zelenko 
204244d93782SGreg Clayton typedef std::shared_ptr<TreeDelegate> TreeDelegateSP;
204344d93782SGreg Clayton 
2044b9c1b51eSKate Stone class TreeItem {
204544d93782SGreg Clayton public:
2046b9c1b51eSKate Stone   TreeItem(TreeItem *parent, TreeDelegate &delegate, bool might_have_children)
2047b9c1b51eSKate Stone       : m_parent(parent), m_delegate(delegate), m_user_data(nullptr),
2048b9c1b51eSKate Stone         m_identifier(0), m_row_idx(-1), m_children(),
2049b9c1b51eSKate Stone         m_might_have_children(might_have_children), m_is_expanded(false) {}
205044d93782SGreg Clayton 
2051b9c1b51eSKate Stone   TreeItem &operator=(const TreeItem &rhs) {
2052b9c1b51eSKate Stone     if (this != &rhs) {
205344d93782SGreg Clayton       m_parent = rhs.m_parent;
205444d93782SGreg Clayton       m_delegate = rhs.m_delegate;
2055ec990867SGreg Clayton       m_user_data = rhs.m_user_data;
205644d93782SGreg Clayton       m_identifier = rhs.m_identifier;
205744d93782SGreg Clayton       m_row_idx = rhs.m_row_idx;
205844d93782SGreg Clayton       m_children = rhs.m_children;
205944d93782SGreg Clayton       m_might_have_children = rhs.m_might_have_children;
206044d93782SGreg Clayton       m_is_expanded = rhs.m_is_expanded;
206144d93782SGreg Clayton     }
206244d93782SGreg Clayton     return *this;
206344d93782SGreg Clayton   }
206444d93782SGreg Clayton 
2065b9c1b51eSKate Stone   size_t GetDepth() const {
206644d93782SGreg Clayton     if (m_parent)
206744d93782SGreg Clayton       return 1 + m_parent->GetDepth();
206844d93782SGreg Clayton     return 0;
206944d93782SGreg Clayton   }
207044d93782SGreg Clayton 
2071b9c1b51eSKate Stone   int GetRowIndex() const { return m_row_idx; }
207244d93782SGreg Clayton 
2073b9c1b51eSKate Stone   void ClearChildren() { m_children.clear(); }
207444d93782SGreg Clayton 
2075b9c1b51eSKate Stone   void Resize(size_t n, const TreeItem &t) { m_children.resize(n, t); }
207644d93782SGreg Clayton 
2077b9c1b51eSKate Stone   TreeItem &operator[](size_t i) { return m_children[i]; }
207844d93782SGreg Clayton 
2079b9c1b51eSKate Stone   void SetRowIndex(int row_idx) { m_row_idx = row_idx; }
208044d93782SGreg Clayton 
2081b9c1b51eSKate Stone   size_t GetNumChildren() {
208244d93782SGreg Clayton     m_delegate.TreeDelegateGenerateChildren(*this);
208344d93782SGreg Clayton     return m_children.size();
208444d93782SGreg Clayton   }
208544d93782SGreg Clayton 
2086b9c1b51eSKate Stone   void ItemWasSelected() { m_delegate.TreeDelegateItemSelected(*this); }
2087315b6884SEugene Zelenko 
2088b9c1b51eSKate Stone   void CalculateRowIndexes(int &row_idx) {
208944d93782SGreg Clayton     SetRowIndex(row_idx);
209044d93782SGreg Clayton     ++row_idx;
209144d93782SGreg Clayton 
2092ec990867SGreg Clayton     const bool expanded = IsExpanded();
2093ec990867SGreg Clayton 
209405097246SAdrian Prantl     // The root item must calculate its children, or we must calculate the
209505097246SAdrian Prantl     // number of children if the item is expanded
2096c5dac77aSEugene Zelenko     if (m_parent == nullptr || expanded)
209744d93782SGreg Clayton       GetNumChildren();
209844d93782SGreg Clayton 
2099b9c1b51eSKate Stone     for (auto &item : m_children) {
210044d93782SGreg Clayton       if (expanded)
210144d93782SGreg Clayton         item.CalculateRowIndexes(row_idx);
210244d93782SGreg Clayton       else
210344d93782SGreg Clayton         item.SetRowIndex(-1);
210444d93782SGreg Clayton     }
210544d93782SGreg Clayton   }
210644d93782SGreg Clayton 
2107b9c1b51eSKate Stone   TreeItem *GetParent() { return m_parent; }
210844d93782SGreg Clayton 
2109b9c1b51eSKate Stone   bool IsExpanded() const { return m_is_expanded; }
211044d93782SGreg Clayton 
2111b9c1b51eSKate Stone   void Expand() { m_is_expanded = true; }
211244d93782SGreg Clayton 
2113b9c1b51eSKate Stone   void Unexpand() { m_is_expanded = false; }
211444d93782SGreg Clayton 
2115b9c1b51eSKate Stone   bool Draw(Window &window, const int first_visible_row,
2116b9c1b51eSKate Stone             const uint32_t selected_row_idx, int &row_idx, int &num_rows_left) {
211744d93782SGreg Clayton     if (num_rows_left <= 0)
211844d93782SGreg Clayton       return false;
211944d93782SGreg Clayton 
2120b9c1b51eSKate Stone     if (m_row_idx >= first_visible_row) {
212144d93782SGreg Clayton       window.MoveCursor(2, row_idx + 1);
212244d93782SGreg Clayton 
212344d93782SGreg Clayton       if (m_parent)
212444d93782SGreg Clayton         m_parent->DrawTreeForChild(window, this, 0);
212544d93782SGreg Clayton 
2126b9c1b51eSKate Stone       if (m_might_have_children) {
2127b9c1b51eSKate Stone         // It we can get UTF8 characters to work we should try to use the
212805097246SAdrian Prantl         // "symbol" UTF8 string below
212944d93782SGreg Clayton         //            const char *symbol = "";
213044d93782SGreg Clayton         //            if (row.expanded)
213144d93782SGreg Clayton         //                symbol = "\xe2\x96\xbd ";
213244d93782SGreg Clayton         //            else
213344d93782SGreg Clayton         //                symbol = "\xe2\x96\xb7 ";
213444d93782SGreg Clayton         //            window.PutCString (symbol);
213544d93782SGreg Clayton 
213644d93782SGreg Clayton         // The ACS_DARROW and ACS_RARROW don't look very nice they are just a
213744d93782SGreg Clayton         // 'v' or '>' character...
213844d93782SGreg Clayton         //            if (expanded)
213944d93782SGreg Clayton         //                window.PutChar (ACS_DARROW);
214044d93782SGreg Clayton         //            else
214144d93782SGreg Clayton         //                window.PutChar (ACS_RARROW);
214205097246SAdrian Prantl         // Since we can't find any good looking right arrow/down arrow symbols,
214305097246SAdrian Prantl         // just use a diamond...
214444d93782SGreg Clayton         window.PutChar(ACS_DIAMOND);
214544d93782SGreg Clayton         window.PutChar(ACS_HLINE);
214644d93782SGreg Clayton       }
2147b9c1b51eSKate Stone       bool highlight = (selected_row_idx == static_cast<size_t>(m_row_idx)) &&
2148b9c1b51eSKate Stone                        window.IsActive();
214944d93782SGreg Clayton 
215044d93782SGreg Clayton       if (highlight)
215144d93782SGreg Clayton         window.AttributeOn(A_REVERSE);
215244d93782SGreg Clayton 
215344d93782SGreg Clayton       m_delegate.TreeDelegateDrawTreeItem(*this, window);
215444d93782SGreg Clayton 
215544d93782SGreg Clayton       if (highlight)
215644d93782SGreg Clayton         window.AttributeOff(A_REVERSE);
215744d93782SGreg Clayton       ++row_idx;
215844d93782SGreg Clayton       --num_rows_left;
215944d93782SGreg Clayton     }
216044d93782SGreg Clayton 
216144d93782SGreg Clayton     if (num_rows_left <= 0)
216244d93782SGreg Clayton       return false; // We are done drawing...
216344d93782SGreg Clayton 
2164b9c1b51eSKate Stone     if (IsExpanded()) {
2165b9c1b51eSKate Stone       for (auto &item : m_children) {
216605097246SAdrian Prantl         // If we displayed all the rows and item.Draw() returns false we are
216705097246SAdrian Prantl         // done drawing and can exit this for loop
2168b9c1b51eSKate Stone         if (!item.Draw(window, first_visible_row, selected_row_idx, row_idx,
2169b9c1b51eSKate Stone                        num_rows_left))
217044d93782SGreg Clayton           break;
217144d93782SGreg Clayton       }
217244d93782SGreg Clayton     }
217344d93782SGreg Clayton     return num_rows_left >= 0; // Return true if not done drawing yet
217444d93782SGreg Clayton   }
217544d93782SGreg Clayton 
2176b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, TreeItem *child,
2177b9c1b51eSKate Stone                         uint32_t reverse_depth) {
217844d93782SGreg Clayton     if (m_parent)
217944d93782SGreg Clayton       m_parent->DrawTreeForChild(window, this, reverse_depth + 1);
218044d93782SGreg Clayton 
2181b9c1b51eSKate Stone     if (&m_children.back() == child) {
218244d93782SGreg Clayton       // Last child
2183b9c1b51eSKate Stone       if (reverse_depth == 0) {
218444d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
218544d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2186b9c1b51eSKate Stone       } else {
218744d93782SGreg Clayton         window.PutChar(' ');
218844d93782SGreg Clayton         window.PutChar(' ');
218944d93782SGreg Clayton       }
2190b9c1b51eSKate Stone     } else {
2191b9c1b51eSKate Stone       if (reverse_depth == 0) {
219244d93782SGreg Clayton         window.PutChar(ACS_LTEE);
219344d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2194b9c1b51eSKate Stone       } else {
219544d93782SGreg Clayton         window.PutChar(ACS_VLINE);
219644d93782SGreg Clayton         window.PutChar(' ');
219744d93782SGreg Clayton       }
219844d93782SGreg Clayton     }
219944d93782SGreg Clayton   }
220044d93782SGreg Clayton 
2201b9c1b51eSKate Stone   TreeItem *GetItemForRowIndex(uint32_t row_idx) {
22023985c8c6SSaleem Abdulrasool     if (static_cast<uint32_t>(m_row_idx) == row_idx)
220344d93782SGreg Clayton       return this;
220444d93782SGreg Clayton     if (m_children.empty())
2205c5dac77aSEugene Zelenko       return nullptr;
2206b9c1b51eSKate Stone     if (IsExpanded()) {
2207b9c1b51eSKate Stone       for (auto &item : m_children) {
220844d93782SGreg Clayton         TreeItem *selected_item_ptr = item.GetItemForRowIndex(row_idx);
220944d93782SGreg Clayton         if (selected_item_ptr)
221044d93782SGreg Clayton           return selected_item_ptr;
221144d93782SGreg Clayton       }
221244d93782SGreg Clayton     }
2213c5dac77aSEugene Zelenko     return nullptr;
221444d93782SGreg Clayton   }
221544d93782SGreg Clayton 
2216b9c1b51eSKate Stone   void *GetUserData() const { return m_user_data; }
2217ec990867SGreg Clayton 
2218b9c1b51eSKate Stone   void SetUserData(void *user_data) { m_user_data = user_data; }
2219ec990867SGreg Clayton 
2220b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
222144d93782SGreg Clayton 
2222b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
222344d93782SGreg Clayton 
2224b9c1b51eSKate Stone   void SetMightHaveChildren(bool b) { m_might_have_children = b; }
2225ec990867SGreg Clayton 
222644d93782SGreg Clayton protected:
222744d93782SGreg Clayton   TreeItem *m_parent;
222844d93782SGreg Clayton   TreeDelegate &m_delegate;
2229ec990867SGreg Clayton   void *m_user_data;
223044d93782SGreg Clayton   uint64_t m_identifier;
2231b9c1b51eSKate Stone   int m_row_idx; // Zero based visible row index, -1 if not visible or for the
2232b9c1b51eSKate Stone                  // root item
223344d93782SGreg Clayton   std::vector<TreeItem> m_children;
223444d93782SGreg Clayton   bool m_might_have_children;
223544d93782SGreg Clayton   bool m_is_expanded;
223644d93782SGreg Clayton };
223744d93782SGreg Clayton 
2238b9c1b51eSKate Stone class TreeWindowDelegate : public WindowDelegate {
223944d93782SGreg Clayton public:
2240b9c1b51eSKate Stone   TreeWindowDelegate(Debugger &debugger, const TreeDelegateSP &delegate_sp)
2241b9c1b51eSKate Stone       : m_debugger(debugger), m_delegate_sp(delegate_sp),
2242b9c1b51eSKate Stone         m_root(nullptr, *delegate_sp, true), m_selected_item(nullptr),
2243b9c1b51eSKate Stone         m_num_rows(0), m_selected_row_idx(0), m_first_visible_row(0),
2244b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
224544d93782SGreg Clayton 
2246b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
224744d93782SGreg Clayton 
2248b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
2249b9c1b51eSKate Stone     ExecutionContext exe_ctx(
2250b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
225144d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
225244d93782SGreg Clayton 
225344d93782SGreg Clayton     bool display_content = false;
2254b9c1b51eSKate Stone     if (process) {
225544d93782SGreg Clayton       StateType state = process->GetState();
2256b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
225744d93782SGreg Clayton         // We are stopped, so it is ok to
225844d93782SGreg Clayton         display_content = true;
2259b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
226044d93782SGreg Clayton         return true; // Don't do any updating when we are running
226144d93782SGreg Clayton       }
226244d93782SGreg Clayton     }
226344d93782SGreg Clayton 
226444d93782SGreg Clayton     m_min_x = 2;
226544d93782SGreg Clayton     m_min_y = 1;
226644d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
226744d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
226844d93782SGreg Clayton 
226944d93782SGreg Clayton     window.Erase();
227044d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
227144d93782SGreg Clayton 
2272b9c1b51eSKate Stone     if (display_content) {
227344d93782SGreg Clayton       const int num_visible_rows = NumVisibleRows();
227444d93782SGreg Clayton       m_num_rows = 0;
227544d93782SGreg Clayton       m_root.CalculateRowIndexes(m_num_rows);
227644d93782SGreg Clayton 
227705097246SAdrian Prantl       // If we unexpanded while having something selected our total number of
227805097246SAdrian Prantl       // rows is less than the num visible rows, then make sure we show all the
227905097246SAdrian Prantl       // rows by setting the first visible row accordingly.
228044d93782SGreg Clayton       if (m_first_visible_row > 0 && m_num_rows < num_visible_rows)
228144d93782SGreg Clayton         m_first_visible_row = 0;
228244d93782SGreg Clayton 
228344d93782SGreg Clayton       // Make sure the selected row is always visible
228444d93782SGreg Clayton       if (m_selected_row_idx < m_first_visible_row)
228544d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx;
228644d93782SGreg Clayton       else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
228744d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
228844d93782SGreg Clayton 
228944d93782SGreg Clayton       int row_idx = 0;
229044d93782SGreg Clayton       int num_rows_left = num_visible_rows;
2291b9c1b51eSKate Stone       m_root.Draw(window, m_first_visible_row, m_selected_row_idx, row_idx,
2292b9c1b51eSKate Stone                   num_rows_left);
229344d93782SGreg Clayton       // Get the selected row
229444d93782SGreg Clayton       m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
2295b9c1b51eSKate Stone     } else {
2296c5dac77aSEugene Zelenko       m_selected_item = nullptr;
229744d93782SGreg Clayton     }
229844d93782SGreg Clayton 
229944d93782SGreg Clayton     window.DeferredRefresh();
230044d93782SGreg Clayton 
230144d93782SGreg Clayton     return true; // Drawing handled
230244d93782SGreg Clayton   }
230344d93782SGreg Clayton 
2304b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
230544d93782SGreg Clayton     return "Thread window keyboard shortcuts:";
230644d93782SGreg Clayton   }
230744d93782SGreg Clayton 
2308b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
230944d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
231044d93782SGreg Clayton         {KEY_UP, "Select previous item"},
231144d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
231244d93782SGreg Clayton         {KEY_RIGHT, "Expand the selected item"},
2313b9c1b51eSKate Stone         {KEY_LEFT,
2314b9c1b51eSKate Stone          "Unexpand the selected item or select parent if not expanded"},
231544d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
231644d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
231744d93782SGreg Clayton         {'h', "Show help dialog"},
231844d93782SGreg Clayton         {' ', "Toggle item expansion"},
231944d93782SGreg Clayton         {',', "Page up"},
232044d93782SGreg Clayton         {'.', "Page down"},
2321b9c1b51eSKate Stone         {'\0', nullptr}};
232244d93782SGreg Clayton     return g_source_view_key_help;
232344d93782SGreg Clayton   }
232444d93782SGreg Clayton 
2325b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2326b9c1b51eSKate Stone     switch (c) {
232744d93782SGreg Clayton     case ',':
232844d93782SGreg Clayton     case KEY_PPAGE:
232944d93782SGreg Clayton       // Page up key
2330b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
233144d93782SGreg Clayton         if (m_first_visible_row > m_max_y)
233244d93782SGreg Clayton           m_first_visible_row -= m_max_y;
233344d93782SGreg Clayton         else
233444d93782SGreg Clayton           m_first_visible_row = 0;
233544d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
233644d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
233744d93782SGreg Clayton         if (m_selected_item)
233844d93782SGreg Clayton           m_selected_item->ItemWasSelected();
233944d93782SGreg Clayton       }
234044d93782SGreg Clayton       return eKeyHandled;
234144d93782SGreg Clayton 
234244d93782SGreg Clayton     case '.':
234344d93782SGreg Clayton     case KEY_NPAGE:
234444d93782SGreg Clayton       // Page down key
2345b9c1b51eSKate Stone       if (m_num_rows > m_max_y) {
2346b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
234744d93782SGreg Clayton           m_first_visible_row += m_max_y;
234844d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
234944d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
235044d93782SGreg Clayton           if (m_selected_item)
235144d93782SGreg Clayton             m_selected_item->ItemWasSelected();
235244d93782SGreg Clayton         }
235344d93782SGreg Clayton       }
235444d93782SGreg Clayton       return eKeyHandled;
235544d93782SGreg Clayton 
235644d93782SGreg Clayton     case KEY_UP:
2357b9c1b51eSKate Stone       if (m_selected_row_idx > 0) {
235844d93782SGreg Clayton         --m_selected_row_idx;
235944d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
236044d93782SGreg Clayton         if (m_selected_item)
236144d93782SGreg Clayton           m_selected_item->ItemWasSelected();
236244d93782SGreg Clayton       }
236344d93782SGreg Clayton       return eKeyHandled;
2364315b6884SEugene Zelenko 
236544d93782SGreg Clayton     case KEY_DOWN:
2366b9c1b51eSKate Stone       if (m_selected_row_idx + 1 < m_num_rows) {
236744d93782SGreg Clayton         ++m_selected_row_idx;
236844d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
236944d93782SGreg Clayton         if (m_selected_item)
237044d93782SGreg Clayton           m_selected_item->ItemWasSelected();
237144d93782SGreg Clayton       }
237244d93782SGreg Clayton       return eKeyHandled;
237344d93782SGreg Clayton 
237444d93782SGreg Clayton     case KEY_RIGHT:
2375b9c1b51eSKate Stone       if (m_selected_item) {
237644d93782SGreg Clayton         if (!m_selected_item->IsExpanded())
237744d93782SGreg Clayton           m_selected_item->Expand();
237844d93782SGreg Clayton       }
237944d93782SGreg Clayton       return eKeyHandled;
238044d93782SGreg Clayton 
238144d93782SGreg Clayton     case KEY_LEFT:
2382b9c1b51eSKate Stone       if (m_selected_item) {
238344d93782SGreg Clayton         if (m_selected_item->IsExpanded())
238444d93782SGreg Clayton           m_selected_item->Unexpand();
2385b9c1b51eSKate Stone         else if (m_selected_item->GetParent()) {
238644d93782SGreg Clayton           m_selected_row_idx = m_selected_item->GetParent()->GetRowIndex();
238744d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
238844d93782SGreg Clayton           if (m_selected_item)
238944d93782SGreg Clayton             m_selected_item->ItemWasSelected();
239044d93782SGreg Clayton         }
239144d93782SGreg Clayton       }
239244d93782SGreg Clayton       return eKeyHandled;
239344d93782SGreg Clayton 
239444d93782SGreg Clayton     case ' ':
239544d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2396b9c1b51eSKate Stone       if (m_selected_item) {
239744d93782SGreg Clayton         if (m_selected_item->IsExpanded())
239844d93782SGreg Clayton           m_selected_item->Unexpand();
239944d93782SGreg Clayton         else
240044d93782SGreg Clayton           m_selected_item->Expand();
240144d93782SGreg Clayton       }
240244d93782SGreg Clayton       return eKeyHandled;
240344d93782SGreg Clayton 
240444d93782SGreg Clayton     case 'h':
240544d93782SGreg Clayton       window.CreateHelpSubwindow();
240644d93782SGreg Clayton       return eKeyHandled;
240744d93782SGreg Clayton 
240844d93782SGreg Clayton     default:
240944d93782SGreg Clayton       break;
241044d93782SGreg Clayton     }
241144d93782SGreg Clayton     return eKeyNotHandled;
241244d93782SGreg Clayton   }
241344d93782SGreg Clayton 
241444d93782SGreg Clayton protected:
241544d93782SGreg Clayton   Debugger &m_debugger;
241644d93782SGreg Clayton   TreeDelegateSP m_delegate_sp;
241744d93782SGreg Clayton   TreeItem m_root;
241844d93782SGreg Clayton   TreeItem *m_selected_item;
241944d93782SGreg Clayton   int m_num_rows;
242044d93782SGreg Clayton   int m_selected_row_idx;
242144d93782SGreg Clayton   int m_first_visible_row;
242244d93782SGreg Clayton   int m_min_x;
242344d93782SGreg Clayton   int m_min_y;
242444d93782SGreg Clayton   int m_max_x;
242544d93782SGreg Clayton   int m_max_y;
242644d93782SGreg Clayton };
242744d93782SGreg Clayton 
2428b9c1b51eSKate Stone class FrameTreeDelegate : public TreeDelegate {
242944d93782SGreg Clayton public:
2430b9c1b51eSKate Stone   FrameTreeDelegate() : TreeDelegate() {
2431b9c1b51eSKate Stone     FormatEntity::Parse(
2432b9c1b51eSKate Stone         "frame #${frame.index}: {${function.name}${function.pc-offset}}}",
2433554f68d3SGreg Clayton         m_format);
243444d93782SGreg Clayton   }
243544d93782SGreg Clayton 
2436315b6884SEugene Zelenko   ~FrameTreeDelegate() override = default;
243744d93782SGreg Clayton 
2438b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2439ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2440b9c1b51eSKate Stone     if (thread) {
244144d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2442ec990867SGreg Clayton       StackFrameSP frame_sp = thread->GetStackFrameAtIndex(frame_idx);
2443b9c1b51eSKate Stone       if (frame_sp) {
244444d93782SGreg Clayton         StreamString strm;
2445b9c1b51eSKate Stone         const SymbolContext &sc =
2446b9c1b51eSKate Stone             frame_sp->GetSymbolContext(eSymbolContextEverything);
244744d93782SGreg Clayton         ExecutionContext exe_ctx(frame_sp);
2448b9c1b51eSKate Stone         if (FormatEntity::Format(m_format, strm, &sc, &exe_ctx, nullptr,
2449b9c1b51eSKate Stone                                  nullptr, false, false)) {
245044d93782SGreg Clayton           int right_pad = 1;
2451c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
245244d93782SGreg Clayton         }
245344d93782SGreg Clayton       }
245444d93782SGreg Clayton     }
245544d93782SGreg Clayton   }
2456315b6884SEugene Zelenko 
2457b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
245844d93782SGreg Clayton     // No children for frames yet...
245944d93782SGreg Clayton   }
246044d93782SGreg Clayton 
2461b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2462ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2463b9c1b51eSKate Stone     if (thread) {
2464b9c1b51eSKate Stone       thread->GetProcess()->GetThreadList().SetSelectedThreadByID(
2465b9c1b51eSKate Stone           thread->GetID());
246644d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2467ec990867SGreg Clayton       thread->SetSelectedFrameByIndex(frame_idx);
246844d93782SGreg Clayton       return true;
246944d93782SGreg Clayton     }
247044d93782SGreg Clayton     return false;
247144d93782SGreg Clayton   }
2472315b6884SEugene Zelenko 
2473554f68d3SGreg Clayton protected:
2474554f68d3SGreg Clayton   FormatEntity::Entry m_format;
247544d93782SGreg Clayton };
247644d93782SGreg Clayton 
2477b9c1b51eSKate Stone class ThreadTreeDelegate : public TreeDelegate {
247844d93782SGreg Clayton public:
2479b9c1b51eSKate Stone   ThreadTreeDelegate(Debugger &debugger)
2480b9c1b51eSKate Stone       : TreeDelegate(), m_debugger(debugger), m_tid(LLDB_INVALID_THREAD_ID),
2481b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2482b9c1b51eSKate Stone     FormatEntity::Parse("thread #${thread.index}: tid = ${thread.id}{, stop "
2483b9c1b51eSKate Stone                         "reason = ${thread.stop-reason}}",
2484554f68d3SGreg Clayton                         m_format);
248544d93782SGreg Clayton   }
248644d93782SGreg Clayton 
2487315b6884SEugene Zelenko   ~ThreadTreeDelegate() override = default;
248844d93782SGreg Clayton 
2489b9c1b51eSKate Stone   ProcessSP GetProcess() {
2490b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2491b9c1b51eSKate Stone         .GetExecutionContext()
2492b9c1b51eSKate Stone         .GetProcessSP();
2493ec990867SGreg Clayton   }
2494ec990867SGreg Clayton 
2495b9c1b51eSKate Stone   ThreadSP GetThread(const TreeItem &item) {
2496ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2497ec990867SGreg Clayton     if (process_sp)
2498ec990867SGreg Clayton       return process_sp->GetThreadList().FindThreadByID(item.GetIdentifier());
2499ec990867SGreg Clayton     return ThreadSP();
2500ec990867SGreg Clayton   }
2501ec990867SGreg Clayton 
2502b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2503ec990867SGreg Clayton     ThreadSP thread_sp = GetThread(item);
2504b9c1b51eSKate Stone     if (thread_sp) {
250544d93782SGreg Clayton       StreamString strm;
250644d93782SGreg Clayton       ExecutionContext exe_ctx(thread_sp);
2507b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2508b9c1b51eSKate Stone                                nullptr, false, false)) {
250944d93782SGreg Clayton         int right_pad = 1;
2510c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
251144d93782SGreg Clayton       }
251244d93782SGreg Clayton     }
251344d93782SGreg Clayton   }
2514315b6884SEugene Zelenko 
2515b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2516ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2517b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
251844d93782SGreg Clayton       StateType state = process_sp->GetState();
2519b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2520ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2521b9c1b51eSKate Stone         if (thread_sp) {
2522b9c1b51eSKate Stone           if (m_stop_id == process_sp->GetStopID() &&
2523b9c1b51eSKate Stone               thread_sp->GetID() == m_tid)
252444d93782SGreg Clayton             return; // Children are already up to date
2525b9c1b51eSKate Stone           if (!m_frame_delegate_sp) {
252644d93782SGreg Clayton             // Always expand the thread item the first time we show it
2527796ac80bSJonas Devlieghere             m_frame_delegate_sp = std::make_shared<FrameTreeDelegate>();
252844d93782SGreg Clayton           }
252944d93782SGreg Clayton 
253044d93782SGreg Clayton           m_stop_id = process_sp->GetStopID();
253144d93782SGreg Clayton           m_tid = thread_sp->GetID();
253244d93782SGreg Clayton 
253344d93782SGreg Clayton           TreeItem t(&item, *m_frame_delegate_sp, false);
253444d93782SGreg Clayton           size_t num_frames = thread_sp->GetStackFrameCount();
253544d93782SGreg Clayton           item.Resize(num_frames, t);
2536b9c1b51eSKate Stone           for (size_t i = 0; i < num_frames; ++i) {
2537ec990867SGreg Clayton             item[i].SetUserData(thread_sp.get());
253844d93782SGreg Clayton             item[i].SetIdentifier(i);
253944d93782SGreg Clayton           }
254044d93782SGreg Clayton         }
254144d93782SGreg Clayton         return;
254244d93782SGreg Clayton       }
254344d93782SGreg Clayton     }
254444d93782SGreg Clayton     item.ClearChildren();
254544d93782SGreg Clayton   }
254644d93782SGreg Clayton 
2547b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2548ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2549b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2550ec990867SGreg Clayton       StateType state = process_sp->GetState();
2551b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2552ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2553b9c1b51eSKate Stone         if (thread_sp) {
255444d93782SGreg Clayton           ThreadList &thread_list = thread_sp->GetProcess()->GetThreadList();
2555bb19a13cSSaleem Abdulrasool           std::lock_guard<std::recursive_mutex> guard(thread_list.GetMutex());
255644d93782SGreg Clayton           ThreadSP selected_thread_sp = thread_list.GetSelectedThread();
2557b9c1b51eSKate Stone           if (selected_thread_sp->GetID() != thread_sp->GetID()) {
255844d93782SGreg Clayton             thread_list.SetSelectedThreadByID(thread_sp->GetID());
255944d93782SGreg Clayton             return true;
256044d93782SGreg Clayton           }
256144d93782SGreg Clayton         }
2562ec990867SGreg Clayton       }
2563ec990867SGreg Clayton     }
256444d93782SGreg Clayton     return false;
256544d93782SGreg Clayton   }
256644d93782SGreg Clayton 
256744d93782SGreg Clayton protected:
256844d93782SGreg Clayton   Debugger &m_debugger;
256944d93782SGreg Clayton   std::shared_ptr<FrameTreeDelegate> m_frame_delegate_sp;
257044d93782SGreg Clayton   lldb::user_id_t m_tid;
257144d93782SGreg Clayton   uint32_t m_stop_id;
2572554f68d3SGreg Clayton   FormatEntity::Entry m_format;
257344d93782SGreg Clayton };
257444d93782SGreg Clayton 
2575b9c1b51eSKate Stone class ThreadsTreeDelegate : public TreeDelegate {
2576ec990867SGreg Clayton public:
2577b9c1b51eSKate Stone   ThreadsTreeDelegate(Debugger &debugger)
2578b9c1b51eSKate Stone       : TreeDelegate(), m_thread_delegate_sp(), m_debugger(debugger),
2579b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2580554f68d3SGreg Clayton     FormatEntity::Parse("process ${process.id}{, name = ${process.name}}",
2581554f68d3SGreg Clayton                         m_format);
2582ec990867SGreg Clayton   }
2583ec990867SGreg Clayton 
2584315b6884SEugene Zelenko   ~ThreadsTreeDelegate() override = default;
2585ec990867SGreg Clayton 
2586b9c1b51eSKate Stone   ProcessSP GetProcess() {
2587b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2588b9c1b51eSKate Stone         .GetExecutionContext()
2589b9c1b51eSKate Stone         .GetProcessSP();
2590ec990867SGreg Clayton   }
2591ec990867SGreg Clayton 
2592b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2593ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2594b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2595ec990867SGreg Clayton       StreamString strm;
2596ec990867SGreg Clayton       ExecutionContext exe_ctx(process_sp);
2597b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2598b9c1b51eSKate Stone                                nullptr, false, false)) {
2599ec990867SGreg Clayton         int right_pad = 1;
2600c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
2601ec990867SGreg Clayton       }
2602ec990867SGreg Clayton     }
2603ec990867SGreg Clayton   }
2604ec990867SGreg Clayton 
2605b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2606ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2607b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2608ec990867SGreg Clayton       StateType state = process_sp->GetState();
2609b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2610ec990867SGreg Clayton         const uint32_t stop_id = process_sp->GetStopID();
2611ec990867SGreg Clayton         if (m_stop_id == stop_id)
2612ec990867SGreg Clayton           return; // Children are already up to date
2613ec990867SGreg Clayton 
2614ec990867SGreg Clayton         m_stop_id = stop_id;
2615ec990867SGreg Clayton 
2616b9c1b51eSKate Stone         if (!m_thread_delegate_sp) {
2617ec990867SGreg Clayton           // Always expand the thread item the first time we show it
2618ec990867SGreg Clayton           // item.Expand();
2619796ac80bSJonas Devlieghere           m_thread_delegate_sp =
2620796ac80bSJonas Devlieghere               std::make_shared<ThreadTreeDelegate>(m_debugger);
2621ec990867SGreg Clayton         }
2622ec990867SGreg Clayton 
2623ec990867SGreg Clayton         TreeItem t(&item, *m_thread_delegate_sp, false);
2624ec990867SGreg Clayton         ThreadList &threads = process_sp->GetThreadList();
2625bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
2626ec990867SGreg Clayton         size_t num_threads = threads.GetSize();
2627ec990867SGreg Clayton         item.Resize(num_threads, t);
2628b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
2629ec990867SGreg Clayton           item[i].SetIdentifier(threads.GetThreadAtIndex(i)->GetID());
2630ec990867SGreg Clayton           item[i].SetMightHaveChildren(true);
2631ec990867SGreg Clayton         }
2632ec990867SGreg Clayton         return;
2633ec990867SGreg Clayton       }
2634ec990867SGreg Clayton     }
2635ec990867SGreg Clayton     item.ClearChildren();
2636ec990867SGreg Clayton   }
2637ec990867SGreg Clayton 
2638b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override { return false; }
2639ec990867SGreg Clayton 
2640ec990867SGreg Clayton protected:
2641ec990867SGreg Clayton   std::shared_ptr<ThreadTreeDelegate> m_thread_delegate_sp;
2642ec990867SGreg Clayton   Debugger &m_debugger;
2643ec990867SGreg Clayton   uint32_t m_stop_id;
2644554f68d3SGreg Clayton   FormatEntity::Entry m_format;
2645ec990867SGreg Clayton };
2646ec990867SGreg Clayton 
2647b9c1b51eSKate Stone class ValueObjectListDelegate : public WindowDelegate {
264844d93782SGreg Clayton public:
2649b9c1b51eSKate Stone   ValueObjectListDelegate()
26508369b28dSGreg Clayton       : m_rows(), m_selected_row(nullptr),
2651b9c1b51eSKate Stone         m_selected_row_idx(0), m_first_visible_row(0), m_num_rows(0),
2652b9c1b51eSKate Stone         m_max_x(0), m_max_y(0) {}
265344d93782SGreg Clayton 
2654b9c1b51eSKate Stone   ValueObjectListDelegate(ValueObjectList &valobj_list)
26558369b28dSGreg Clayton       : m_rows(), m_selected_row(nullptr),
2656b9c1b51eSKate Stone         m_selected_row_idx(0), m_first_visible_row(0), m_num_rows(0),
2657b9c1b51eSKate Stone         m_max_x(0), m_max_y(0) {
265844d93782SGreg Clayton     SetValues(valobj_list);
265944d93782SGreg Clayton   }
266044d93782SGreg Clayton 
2661315b6884SEugene Zelenko   ~ValueObjectListDelegate() override = default;
266244d93782SGreg Clayton 
2663b9c1b51eSKate Stone   void SetValues(ValueObjectList &valobj_list) {
2664c5dac77aSEugene Zelenko     m_selected_row = nullptr;
266544d93782SGreg Clayton     m_selected_row_idx = 0;
266644d93782SGreg Clayton     m_first_visible_row = 0;
266744d93782SGreg Clayton     m_num_rows = 0;
266844d93782SGreg Clayton     m_rows.clear();
26698369b28dSGreg Clayton     for (auto &valobj_sp : valobj_list.GetObjects())
26708369b28dSGreg Clayton       m_rows.push_back(Row(valobj_sp, nullptr));
267144d93782SGreg Clayton   }
267244d93782SGreg Clayton 
2673b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
267444d93782SGreg Clayton     m_num_rows = 0;
267544d93782SGreg Clayton     m_min_x = 2;
267644d93782SGreg Clayton     m_min_y = 1;
267744d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
267844d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
267944d93782SGreg Clayton 
268044d93782SGreg Clayton     window.Erase();
268144d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
268244d93782SGreg Clayton 
268344d93782SGreg Clayton     const int num_visible_rows = NumVisibleRows();
268444d93782SGreg Clayton     const int num_rows = CalculateTotalNumberRows(m_rows);
268544d93782SGreg Clayton 
268605097246SAdrian Prantl     // If we unexpanded while having something selected our total number of
268705097246SAdrian Prantl     // rows is less than the num visible rows, then make sure we show all the
268805097246SAdrian Prantl     // rows by setting the first visible row accordingly.
268944d93782SGreg Clayton     if (m_first_visible_row > 0 && num_rows < num_visible_rows)
269044d93782SGreg Clayton       m_first_visible_row = 0;
269144d93782SGreg Clayton 
269244d93782SGreg Clayton     // Make sure the selected row is always visible
269344d93782SGreg Clayton     if (m_selected_row_idx < m_first_visible_row)
269444d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx;
269544d93782SGreg Clayton     else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
269644d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
269744d93782SGreg Clayton 
269844d93782SGreg Clayton     DisplayRows(window, m_rows, g_options);
269944d93782SGreg Clayton 
270044d93782SGreg Clayton     window.DeferredRefresh();
270144d93782SGreg Clayton 
270244d93782SGreg Clayton     // Get the selected row
270344d93782SGreg Clayton     m_selected_row = GetRowForRowIndex(m_selected_row_idx);
270405097246SAdrian Prantl     // Keep the cursor on the selected row so the highlight and the cursor are
270505097246SAdrian Prantl     // always on the same line
270644d93782SGreg Clayton     if (m_selected_row)
2707b9c1b51eSKate Stone       window.MoveCursor(m_selected_row->x, m_selected_row->y);
270844d93782SGreg Clayton 
270944d93782SGreg Clayton     return true; // Drawing handled
271044d93782SGreg Clayton   }
271144d93782SGreg Clayton 
2712b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
271344d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
271444d93782SGreg Clayton         {KEY_UP, "Select previous item"},
271544d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
271644d93782SGreg Clayton         {KEY_RIGHT, "Expand selected item"},
271744d93782SGreg Clayton         {KEY_LEFT, "Unexpand selected item or select parent if not expanded"},
271844d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
271944d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
272044d93782SGreg Clayton         {'A', "Format as annotated address"},
272144d93782SGreg Clayton         {'b', "Format as binary"},
272244d93782SGreg Clayton         {'B', "Format as hex bytes with ASCII"},
272344d93782SGreg Clayton         {'c', "Format as character"},
272444d93782SGreg Clayton         {'d', "Format as a signed integer"},
272544d93782SGreg Clayton         {'D', "Format selected value using the default format for the type"},
272644d93782SGreg Clayton         {'f', "Format as float"},
272744d93782SGreg Clayton         {'h', "Show help dialog"},
272844d93782SGreg Clayton         {'i', "Format as instructions"},
272944d93782SGreg Clayton         {'o', "Format as octal"},
273044d93782SGreg Clayton         {'p', "Format as pointer"},
273144d93782SGreg Clayton         {'s', "Format as C string"},
273244d93782SGreg Clayton         {'t', "Toggle showing/hiding type names"},
273344d93782SGreg Clayton         {'u', "Format as an unsigned integer"},
273444d93782SGreg Clayton         {'x', "Format as hex"},
273544d93782SGreg Clayton         {'X', "Format as uppercase hex"},
273644d93782SGreg Clayton         {' ', "Toggle item expansion"},
273744d93782SGreg Clayton         {',', "Page up"},
273844d93782SGreg Clayton         {'.', "Page down"},
2739b9c1b51eSKate Stone         {'\0', nullptr}};
274044d93782SGreg Clayton     return g_source_view_key_help;
274144d93782SGreg Clayton   }
274244d93782SGreg Clayton 
2743b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2744b9c1b51eSKate Stone     switch (c) {
274544d93782SGreg Clayton     case 'x':
274644d93782SGreg Clayton     case 'X':
274744d93782SGreg Clayton     case 'o':
274844d93782SGreg Clayton     case 's':
274944d93782SGreg Clayton     case 'u':
275044d93782SGreg Clayton     case 'd':
275144d93782SGreg Clayton     case 'D':
275244d93782SGreg Clayton     case 'i':
275344d93782SGreg Clayton     case 'A':
275444d93782SGreg Clayton     case 'p':
275544d93782SGreg Clayton     case 'c':
275644d93782SGreg Clayton     case 'b':
275744d93782SGreg Clayton     case 'B':
275844d93782SGreg Clayton     case 'f':
275944d93782SGreg Clayton       // Change the format for the currently selected item
27608369b28dSGreg Clayton       if (m_selected_row) {
27618369b28dSGreg Clayton         auto valobj_sp = m_selected_row->value.GetSP();
27628369b28dSGreg Clayton         if (valobj_sp)
27638369b28dSGreg Clayton           valobj_sp->SetFormat(FormatForChar(c));
27648369b28dSGreg Clayton       }
276544d93782SGreg Clayton       return eKeyHandled;
276644d93782SGreg Clayton 
276744d93782SGreg Clayton     case 't':
276844d93782SGreg Clayton       // Toggle showing type names
276944d93782SGreg Clayton       g_options.show_types = !g_options.show_types;
277044d93782SGreg Clayton       return eKeyHandled;
277144d93782SGreg Clayton 
277244d93782SGreg Clayton     case ',':
277344d93782SGreg Clayton     case KEY_PPAGE:
277444d93782SGreg Clayton       // Page up key
2775b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
27763985c8c6SSaleem Abdulrasool         if (static_cast<int>(m_first_visible_row) > m_max_y)
277744d93782SGreg Clayton           m_first_visible_row -= m_max_y;
277844d93782SGreg Clayton         else
277944d93782SGreg Clayton           m_first_visible_row = 0;
278044d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
278144d93782SGreg Clayton       }
278244d93782SGreg Clayton       return eKeyHandled;
278344d93782SGreg Clayton 
278444d93782SGreg Clayton     case '.':
278544d93782SGreg Clayton     case KEY_NPAGE:
278644d93782SGreg Clayton       // Page down key
2787b9c1b51eSKate Stone       if (m_num_rows > static_cast<size_t>(m_max_y)) {
2788b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
278944d93782SGreg Clayton           m_first_visible_row += m_max_y;
279044d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
279144d93782SGreg Clayton         }
279244d93782SGreg Clayton       }
279344d93782SGreg Clayton       return eKeyHandled;
279444d93782SGreg Clayton 
279544d93782SGreg Clayton     case KEY_UP:
279644d93782SGreg Clayton       if (m_selected_row_idx > 0)
279744d93782SGreg Clayton         --m_selected_row_idx;
279844d93782SGreg Clayton       return eKeyHandled;
2799315b6884SEugene Zelenko 
280044d93782SGreg Clayton     case KEY_DOWN:
280144d93782SGreg Clayton       if (m_selected_row_idx + 1 < m_num_rows)
280244d93782SGreg Clayton         ++m_selected_row_idx;
280344d93782SGreg Clayton       return eKeyHandled;
280444d93782SGreg Clayton 
280544d93782SGreg Clayton     case KEY_RIGHT:
2806b9c1b51eSKate Stone       if (m_selected_row) {
280744d93782SGreg Clayton         if (!m_selected_row->expanded)
280844d93782SGreg Clayton           m_selected_row->Expand();
280944d93782SGreg Clayton       }
281044d93782SGreg Clayton       return eKeyHandled;
281144d93782SGreg Clayton 
281244d93782SGreg Clayton     case KEY_LEFT:
2813b9c1b51eSKate Stone       if (m_selected_row) {
281444d93782SGreg Clayton         if (m_selected_row->expanded)
281544d93782SGreg Clayton           m_selected_row->Unexpand();
281644d93782SGreg Clayton         else if (m_selected_row->parent)
281744d93782SGreg Clayton           m_selected_row_idx = m_selected_row->parent->row_idx;
281844d93782SGreg Clayton       }
281944d93782SGreg Clayton       return eKeyHandled;
282044d93782SGreg Clayton 
282144d93782SGreg Clayton     case ' ':
282244d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2823b9c1b51eSKate Stone       if (m_selected_row) {
282444d93782SGreg Clayton         if (m_selected_row->expanded)
282544d93782SGreg Clayton           m_selected_row->Unexpand();
282644d93782SGreg Clayton         else
282744d93782SGreg Clayton           m_selected_row->Expand();
282844d93782SGreg Clayton       }
282944d93782SGreg Clayton       return eKeyHandled;
283044d93782SGreg Clayton 
283144d93782SGreg Clayton     case 'h':
283244d93782SGreg Clayton       window.CreateHelpSubwindow();
283344d93782SGreg Clayton       return eKeyHandled;
283444d93782SGreg Clayton 
283544d93782SGreg Clayton     default:
283644d93782SGreg Clayton       break;
283744d93782SGreg Clayton     }
283844d93782SGreg Clayton     return eKeyNotHandled;
283944d93782SGreg Clayton   }
284044d93782SGreg Clayton 
284144d93782SGreg Clayton protected:
284244d93782SGreg Clayton   std::vector<Row> m_rows;
284344d93782SGreg Clayton   Row *m_selected_row;
284444d93782SGreg Clayton   uint32_t m_selected_row_idx;
284544d93782SGreg Clayton   uint32_t m_first_visible_row;
284644d93782SGreg Clayton   uint32_t m_num_rows;
284744d93782SGreg Clayton   int m_min_x;
284844d93782SGreg Clayton   int m_min_y;
284944d93782SGreg Clayton   int m_max_x;
285044d93782SGreg Clayton   int m_max_y;
285144d93782SGreg Clayton 
2852b9c1b51eSKate Stone   static Format FormatForChar(int c) {
2853b9c1b51eSKate Stone     switch (c) {
2854b9c1b51eSKate Stone     case 'x':
2855b9c1b51eSKate Stone       return eFormatHex;
2856b9c1b51eSKate Stone     case 'X':
2857b9c1b51eSKate Stone       return eFormatHexUppercase;
2858b9c1b51eSKate Stone     case 'o':
2859b9c1b51eSKate Stone       return eFormatOctal;
2860b9c1b51eSKate Stone     case 's':
2861b9c1b51eSKate Stone       return eFormatCString;
2862b9c1b51eSKate Stone     case 'u':
2863b9c1b51eSKate Stone       return eFormatUnsigned;
2864b9c1b51eSKate Stone     case 'd':
2865b9c1b51eSKate Stone       return eFormatDecimal;
2866b9c1b51eSKate Stone     case 'D':
2867b9c1b51eSKate Stone       return eFormatDefault;
2868b9c1b51eSKate Stone     case 'i':
2869b9c1b51eSKate Stone       return eFormatInstruction;
2870b9c1b51eSKate Stone     case 'A':
2871b9c1b51eSKate Stone       return eFormatAddressInfo;
2872b9c1b51eSKate Stone     case 'p':
2873b9c1b51eSKate Stone       return eFormatPointer;
2874b9c1b51eSKate Stone     case 'c':
2875b9c1b51eSKate Stone       return eFormatChar;
2876b9c1b51eSKate Stone     case 'b':
2877b9c1b51eSKate Stone       return eFormatBinary;
2878b9c1b51eSKate Stone     case 'B':
2879b9c1b51eSKate Stone       return eFormatBytesWithASCII;
2880b9c1b51eSKate Stone     case 'f':
2881b9c1b51eSKate Stone       return eFormatFloat;
288244d93782SGreg Clayton     }
288344d93782SGreg Clayton     return eFormatDefault;
288444d93782SGreg Clayton   }
288544d93782SGreg Clayton 
2886b9c1b51eSKate Stone   bool DisplayRowObject(Window &window, Row &row, DisplayOptions &options,
2887b9c1b51eSKate Stone                         bool highlight, bool last_child) {
28888369b28dSGreg Clayton     ValueObject *valobj = row.value.GetSP().get();
288944d93782SGreg Clayton 
2890c5dac77aSEugene Zelenko     if (valobj == nullptr)
289144d93782SGreg Clayton       return false;
289244d93782SGreg Clayton 
2893b9c1b51eSKate Stone     const char *type_name =
2894b9c1b51eSKate Stone         options.show_types ? valobj->GetTypeName().GetCString() : nullptr;
289544d93782SGreg Clayton     const char *name = valobj->GetName().GetCString();
289644d93782SGreg Clayton     const char *value = valobj->GetValueAsCString();
289744d93782SGreg Clayton     const char *summary = valobj->GetSummaryAsCString();
289844d93782SGreg Clayton 
289944d93782SGreg Clayton     window.MoveCursor(row.x, row.y);
290044d93782SGreg Clayton 
290144d93782SGreg Clayton     row.DrawTree(window);
290244d93782SGreg Clayton 
290344d93782SGreg Clayton     if (highlight)
290444d93782SGreg Clayton       window.AttributeOn(A_REVERSE);
290544d93782SGreg Clayton 
290644d93782SGreg Clayton     if (type_name && type_name[0])
290744d93782SGreg Clayton       window.Printf("(%s) ", type_name);
290844d93782SGreg Clayton 
290944d93782SGreg Clayton     if (name && name[0])
291044d93782SGreg Clayton       window.PutCString(name);
291144d93782SGreg Clayton 
291244d93782SGreg Clayton     attr_t changd_attr = 0;
291344d93782SGreg Clayton     if (valobj->GetValueDidChange())
291444d93782SGreg Clayton       changd_attr = COLOR_PAIR(5) | A_BOLD;
291544d93782SGreg Clayton 
2916b9c1b51eSKate Stone     if (value && value[0]) {
291744d93782SGreg Clayton       window.PutCString(" = ");
291844d93782SGreg Clayton       if (changd_attr)
291944d93782SGreg Clayton         window.AttributeOn(changd_attr);
292044d93782SGreg Clayton       window.PutCString(value);
292144d93782SGreg Clayton       if (changd_attr)
292244d93782SGreg Clayton         window.AttributeOff(changd_attr);
292344d93782SGreg Clayton     }
292444d93782SGreg Clayton 
2925b9c1b51eSKate Stone     if (summary && summary[0]) {
292644d93782SGreg Clayton       window.PutChar(' ');
292744d93782SGreg Clayton       if (changd_attr)
292844d93782SGreg Clayton         window.AttributeOn(changd_attr);
292944d93782SGreg Clayton       window.PutCString(summary);
293044d93782SGreg Clayton       if (changd_attr)
293144d93782SGreg Clayton         window.AttributeOff(changd_attr);
293244d93782SGreg Clayton     }
293344d93782SGreg Clayton 
293444d93782SGreg Clayton     if (highlight)
293544d93782SGreg Clayton       window.AttributeOff(A_REVERSE);
293644d93782SGreg Clayton 
293744d93782SGreg Clayton     return true;
293844d93782SGreg Clayton   }
2939315b6884SEugene Zelenko 
2940b9c1b51eSKate Stone   void DisplayRows(Window &window, std::vector<Row> &rows,
2941b9c1b51eSKate Stone                    DisplayOptions &options) {
294244d93782SGreg Clayton     // >   0x25B7
294344d93782SGreg Clayton     // \/  0x25BD
294444d93782SGreg Clayton 
294544d93782SGreg Clayton     bool window_is_active = window.IsActive();
2946b9c1b51eSKate Stone     for (auto &row : rows) {
294744d93782SGreg Clayton       const bool last_child = row.parent && &rows[rows.size() - 1] == &row;
294844d93782SGreg Clayton       // Save the row index in each Row structure
294944d93782SGreg Clayton       row.row_idx = m_num_rows;
295044d93782SGreg Clayton       if ((m_num_rows >= m_first_visible_row) &&
2951b9c1b51eSKate Stone           ((m_num_rows - m_first_visible_row) <
2952b9c1b51eSKate Stone            static_cast<size_t>(NumVisibleRows()))) {
295344d93782SGreg Clayton         row.x = m_min_x;
295444d93782SGreg Clayton         row.y = m_num_rows - m_first_visible_row + 1;
2955b9c1b51eSKate Stone         if (DisplayRowObject(window, row, options,
2956b9c1b51eSKate Stone                              window_is_active &&
2957b9c1b51eSKate Stone                                  m_num_rows == m_selected_row_idx,
2958b9c1b51eSKate Stone                              last_child)) {
295944d93782SGreg Clayton           ++m_num_rows;
2960b9c1b51eSKate Stone         } else {
296144d93782SGreg Clayton           row.x = 0;
296244d93782SGreg Clayton           row.y = 0;
296344d93782SGreg Clayton         }
2964b9c1b51eSKate Stone       } else {
296544d93782SGreg Clayton         row.x = 0;
296644d93782SGreg Clayton         row.y = 0;
296744d93782SGreg Clayton         ++m_num_rows;
296844d93782SGreg Clayton       }
296944d93782SGreg Clayton 
29708369b28dSGreg Clayton       auto &children = row.GetChildren();
29718369b28dSGreg Clayton       if (row.expanded && !children.empty()) {
29728369b28dSGreg Clayton         DisplayRows(window, children, options);
297344d93782SGreg Clayton       }
297444d93782SGreg Clayton     }
297544d93782SGreg Clayton   }
297644d93782SGreg Clayton 
29778369b28dSGreg Clayton   int CalculateTotalNumberRows(std::vector<Row> &rows) {
297844d93782SGreg Clayton     int row_count = 0;
29798369b28dSGreg Clayton     for (auto &row : rows) {
298044d93782SGreg Clayton       ++row_count;
298144d93782SGreg Clayton       if (row.expanded)
29828369b28dSGreg Clayton         row_count += CalculateTotalNumberRows(row.GetChildren());
298344d93782SGreg Clayton     }
298444d93782SGreg Clayton     return row_count;
298544d93782SGreg Clayton   }
2986315b6884SEugene Zelenko 
2987b9c1b51eSKate Stone   static Row *GetRowForRowIndexImpl(std::vector<Row> &rows, size_t &row_index) {
2988b9c1b51eSKate Stone     for (auto &row : rows) {
298944d93782SGreg Clayton       if (row_index == 0)
299044d93782SGreg Clayton         return &row;
2991b9c1b51eSKate Stone       else {
299244d93782SGreg Clayton         --row_index;
29938369b28dSGreg Clayton         auto &children = row.GetChildren();
29948369b28dSGreg Clayton         if (row.expanded && !children.empty()) {
29958369b28dSGreg Clayton           Row *result = GetRowForRowIndexImpl(children, row_index);
299644d93782SGreg Clayton           if (result)
299744d93782SGreg Clayton             return result;
299844d93782SGreg Clayton         }
299944d93782SGreg Clayton       }
300044d93782SGreg Clayton     }
3001c5dac77aSEugene Zelenko     return nullptr;
300244d93782SGreg Clayton   }
300344d93782SGreg Clayton 
3004b9c1b51eSKate Stone   Row *GetRowForRowIndex(size_t row_index) {
300544d93782SGreg Clayton     return GetRowForRowIndexImpl(m_rows, row_index);
300644d93782SGreg Clayton   }
300744d93782SGreg Clayton 
3008b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
300944d93782SGreg Clayton 
301044d93782SGreg Clayton   static DisplayOptions g_options;
301144d93782SGreg Clayton };
301244d93782SGreg Clayton 
3013b9c1b51eSKate Stone class FrameVariablesWindowDelegate : public ValueObjectListDelegate {
301444d93782SGreg Clayton public:
3015b9c1b51eSKate Stone   FrameVariablesWindowDelegate(Debugger &debugger)
3016b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger),
3017b9c1b51eSKate Stone         m_frame_block(nullptr) {}
301844d93782SGreg Clayton 
3019315b6884SEugene Zelenko   ~FrameVariablesWindowDelegate() override = default;
302044d93782SGreg Clayton 
3021b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
302244d93782SGreg Clayton     return "Frame variable window keyboard shortcuts:";
302344d93782SGreg Clayton   }
302444d93782SGreg Clayton 
3025b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3026b9c1b51eSKate Stone     ExecutionContext exe_ctx(
3027b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
302844d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
3029c5dac77aSEugene Zelenko     Block *frame_block = nullptr;
3030c5dac77aSEugene Zelenko     StackFrame *frame = nullptr;
303144d93782SGreg Clayton 
3032b9c1b51eSKate Stone     if (process) {
303344d93782SGreg Clayton       StateType state = process->GetState();
3034b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
303544d93782SGreg Clayton         frame = exe_ctx.GetFramePtr();
303644d93782SGreg Clayton         if (frame)
303744d93782SGreg Clayton           frame_block = frame->GetFrameBlock();
3038b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
303944d93782SGreg Clayton         return true; // Don't do any updating when we are running
304044d93782SGreg Clayton       }
304144d93782SGreg Clayton     }
304244d93782SGreg Clayton 
304344d93782SGreg Clayton     ValueObjectList local_values;
3044b9c1b51eSKate Stone     if (frame_block) {
304544d93782SGreg Clayton       // Only update the variables if they have changed
3046b9c1b51eSKate Stone       if (m_frame_block != frame_block) {
304744d93782SGreg Clayton         m_frame_block = frame_block;
304844d93782SGreg Clayton 
304944d93782SGreg Clayton         VariableList *locals = frame->GetVariableList(true);
3050b9c1b51eSKate Stone         if (locals) {
305144d93782SGreg Clayton           const DynamicValueType use_dynamic = eDynamicDontRunTarget;
305244d93782SGreg Clayton           const size_t num_locals = locals->GetSize();
3053b9c1b51eSKate Stone           for (size_t i = 0; i < num_locals; ++i) {
3054b9c1b51eSKate Stone             ValueObjectSP value_sp = frame->GetValueObjectForFrameVariable(
3055b9c1b51eSKate Stone                 locals->GetVariableAtIndex(i), use_dynamic);
3056b9c1b51eSKate Stone             if (value_sp) {
3057eb72dc7dSGreg Clayton               ValueObjectSP synthetic_value_sp = value_sp->GetSyntheticValue();
3058eb72dc7dSGreg Clayton               if (synthetic_value_sp)
3059eb72dc7dSGreg Clayton                 local_values.Append(synthetic_value_sp);
3060eb72dc7dSGreg Clayton               else
3061eb72dc7dSGreg Clayton                 local_values.Append(value_sp);
3062eb72dc7dSGreg Clayton             }
3063eb72dc7dSGreg Clayton           }
306444d93782SGreg Clayton           // Update the values
306544d93782SGreg Clayton           SetValues(local_values);
306644d93782SGreg Clayton         }
306744d93782SGreg Clayton       }
3068b9c1b51eSKate Stone     } else {
3069c5dac77aSEugene Zelenko       m_frame_block = nullptr;
307044d93782SGreg Clayton       // Update the values with an empty list if there is no frame
307144d93782SGreg Clayton       SetValues(local_values);
307244d93782SGreg Clayton     }
307344d93782SGreg Clayton 
307444d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
307544d93782SGreg Clayton   }
307644d93782SGreg Clayton 
307744d93782SGreg Clayton protected:
307844d93782SGreg Clayton   Debugger &m_debugger;
307944d93782SGreg Clayton   Block *m_frame_block;
308044d93782SGreg Clayton };
308144d93782SGreg Clayton 
3082b9c1b51eSKate Stone class RegistersWindowDelegate : public ValueObjectListDelegate {
308344d93782SGreg Clayton public:
3084b9c1b51eSKate Stone   RegistersWindowDelegate(Debugger &debugger)
3085b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger) {}
308644d93782SGreg Clayton 
3087315b6884SEugene Zelenko   ~RegistersWindowDelegate() override = default;
308844d93782SGreg Clayton 
3089b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
309044d93782SGreg Clayton     return "Register window keyboard shortcuts:";
309144d93782SGreg Clayton   }
309244d93782SGreg Clayton 
3093b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3094b9c1b51eSKate Stone     ExecutionContext exe_ctx(
3095b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
309644d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
309744d93782SGreg Clayton 
309844d93782SGreg Clayton     ValueObjectList value_list;
3099b9c1b51eSKate Stone     if (frame) {
3100b9c1b51eSKate Stone       if (frame->GetStackID() != m_stack_id) {
310144d93782SGreg Clayton         m_stack_id = frame->GetStackID();
310244d93782SGreg Clayton         RegisterContextSP reg_ctx(frame->GetRegisterContext());
3103b9c1b51eSKate Stone         if (reg_ctx) {
310444d93782SGreg Clayton           const uint32_t num_sets = reg_ctx->GetRegisterSetCount();
3105b9c1b51eSKate Stone           for (uint32_t set_idx = 0; set_idx < num_sets; ++set_idx) {
3106b9c1b51eSKate Stone             value_list.Append(
3107b9c1b51eSKate Stone                 ValueObjectRegisterSet::Create(frame, reg_ctx, set_idx));
310844d93782SGreg Clayton           }
310944d93782SGreg Clayton         }
311044d93782SGreg Clayton         SetValues(value_list);
311144d93782SGreg Clayton       }
3112b9c1b51eSKate Stone     } else {
311344d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
311444d93782SGreg Clayton       if (process && process->IsAlive())
311544d93782SGreg Clayton         return true; // Don't do any updating if we are running
3116b9c1b51eSKate Stone       else {
311705097246SAdrian Prantl         // Update the values with an empty list if there is no process or the
311805097246SAdrian Prantl         // process isn't alive anymore
311944d93782SGreg Clayton         SetValues(value_list);
312044d93782SGreg Clayton       }
312144d93782SGreg Clayton     }
312244d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
312344d93782SGreg Clayton   }
312444d93782SGreg Clayton 
312544d93782SGreg Clayton protected:
312644d93782SGreg Clayton   Debugger &m_debugger;
312744d93782SGreg Clayton   StackID m_stack_id;
312844d93782SGreg Clayton };
312944d93782SGreg Clayton 
3130b9c1b51eSKate Stone static const char *CursesKeyToCString(int ch) {
313144d93782SGreg Clayton   static char g_desc[32];
3132b9c1b51eSKate Stone   if (ch >= KEY_F0 && ch < KEY_F0 + 64) {
313344d93782SGreg Clayton     snprintf(g_desc, sizeof(g_desc), "F%u", ch - KEY_F0);
313444d93782SGreg Clayton     return g_desc;
313544d93782SGreg Clayton   }
3136b9c1b51eSKate Stone   switch (ch) {
3137b9c1b51eSKate Stone   case KEY_DOWN:
3138b9c1b51eSKate Stone     return "down";
3139b9c1b51eSKate Stone   case KEY_UP:
3140b9c1b51eSKate Stone     return "up";
3141b9c1b51eSKate Stone   case KEY_LEFT:
3142b9c1b51eSKate Stone     return "left";
3143b9c1b51eSKate Stone   case KEY_RIGHT:
3144b9c1b51eSKate Stone     return "right";
3145b9c1b51eSKate Stone   case KEY_HOME:
3146b9c1b51eSKate Stone     return "home";
3147b9c1b51eSKate Stone   case KEY_BACKSPACE:
3148b9c1b51eSKate Stone     return "backspace";
3149b9c1b51eSKate Stone   case KEY_DL:
3150b9c1b51eSKate Stone     return "delete-line";
3151b9c1b51eSKate Stone   case KEY_IL:
3152b9c1b51eSKate Stone     return "insert-line";
3153b9c1b51eSKate Stone   case KEY_DC:
3154b9c1b51eSKate Stone     return "delete-char";
3155b9c1b51eSKate Stone   case KEY_IC:
3156b9c1b51eSKate Stone     return "insert-char";
3157b9c1b51eSKate Stone   case KEY_CLEAR:
3158b9c1b51eSKate Stone     return "clear";
3159b9c1b51eSKate Stone   case KEY_EOS:
3160b9c1b51eSKate Stone     return "clear-to-eos";
3161b9c1b51eSKate Stone   case KEY_EOL:
3162b9c1b51eSKate Stone     return "clear-to-eol";
3163b9c1b51eSKate Stone   case KEY_SF:
3164b9c1b51eSKate Stone     return "scroll-forward";
3165b9c1b51eSKate Stone   case KEY_SR:
3166b9c1b51eSKate Stone     return "scroll-backward";
3167b9c1b51eSKate Stone   case KEY_NPAGE:
3168b9c1b51eSKate Stone     return "page-down";
3169b9c1b51eSKate Stone   case KEY_PPAGE:
3170b9c1b51eSKate Stone     return "page-up";
3171b9c1b51eSKate Stone   case KEY_STAB:
3172b9c1b51eSKate Stone     return "set-tab";
3173b9c1b51eSKate Stone   case KEY_CTAB:
3174b9c1b51eSKate Stone     return "clear-tab";
3175b9c1b51eSKate Stone   case KEY_CATAB:
3176b9c1b51eSKate Stone     return "clear-all-tabs";
3177b9c1b51eSKate Stone   case KEY_ENTER:
3178b9c1b51eSKate Stone     return "enter";
3179b9c1b51eSKate Stone   case KEY_PRINT:
3180b9c1b51eSKate Stone     return "print";
3181b9c1b51eSKate Stone   case KEY_LL:
3182b9c1b51eSKate Stone     return "lower-left key";
3183b9c1b51eSKate Stone   case KEY_A1:
3184b9c1b51eSKate Stone     return "upper left of keypad";
3185b9c1b51eSKate Stone   case KEY_A3:
3186b9c1b51eSKate Stone     return "upper right of keypad";
3187b9c1b51eSKate Stone   case KEY_B2:
3188b9c1b51eSKate Stone     return "center of keypad";
3189b9c1b51eSKate Stone   case KEY_C1:
3190b9c1b51eSKate Stone     return "lower left of keypad";
3191b9c1b51eSKate Stone   case KEY_C3:
3192b9c1b51eSKate Stone     return "lower right of keypad";
3193b9c1b51eSKate Stone   case KEY_BTAB:
3194b9c1b51eSKate Stone     return "back-tab key";
3195b9c1b51eSKate Stone   case KEY_BEG:
3196b9c1b51eSKate Stone     return "begin key";
3197b9c1b51eSKate Stone   case KEY_CANCEL:
3198b9c1b51eSKate Stone     return "cancel key";
3199b9c1b51eSKate Stone   case KEY_CLOSE:
3200b9c1b51eSKate Stone     return "close key";
3201b9c1b51eSKate Stone   case KEY_COMMAND:
3202b9c1b51eSKate Stone     return "command key";
3203b9c1b51eSKate Stone   case KEY_COPY:
3204b9c1b51eSKate Stone     return "copy key";
3205b9c1b51eSKate Stone   case KEY_CREATE:
3206b9c1b51eSKate Stone     return "create key";
3207b9c1b51eSKate Stone   case KEY_END:
3208b9c1b51eSKate Stone     return "end key";
3209b9c1b51eSKate Stone   case KEY_EXIT:
3210b9c1b51eSKate Stone     return "exit key";
3211b9c1b51eSKate Stone   case KEY_FIND:
3212b9c1b51eSKate Stone     return "find key";
3213b9c1b51eSKate Stone   case KEY_HELP:
3214b9c1b51eSKate Stone     return "help key";
3215b9c1b51eSKate Stone   case KEY_MARK:
3216b9c1b51eSKate Stone     return "mark key";
3217b9c1b51eSKate Stone   case KEY_MESSAGE:
3218b9c1b51eSKate Stone     return "message key";
3219b9c1b51eSKate Stone   case KEY_MOVE:
3220b9c1b51eSKate Stone     return "move key";
3221b9c1b51eSKate Stone   case KEY_NEXT:
3222b9c1b51eSKate Stone     return "next key";
3223b9c1b51eSKate Stone   case KEY_OPEN:
3224b9c1b51eSKate Stone     return "open key";
3225b9c1b51eSKate Stone   case KEY_OPTIONS:
3226b9c1b51eSKate Stone     return "options key";
3227b9c1b51eSKate Stone   case KEY_PREVIOUS:
3228b9c1b51eSKate Stone     return "previous key";
3229b9c1b51eSKate Stone   case KEY_REDO:
3230b9c1b51eSKate Stone     return "redo key";
3231b9c1b51eSKate Stone   case KEY_REFERENCE:
3232b9c1b51eSKate Stone     return "reference key";
3233b9c1b51eSKate Stone   case KEY_REFRESH:
3234b9c1b51eSKate Stone     return "refresh key";
3235b9c1b51eSKate Stone   case KEY_REPLACE:
3236b9c1b51eSKate Stone     return "replace key";
3237b9c1b51eSKate Stone   case KEY_RESTART:
3238b9c1b51eSKate Stone     return "restart key";
3239b9c1b51eSKate Stone   case KEY_RESUME:
3240b9c1b51eSKate Stone     return "resume key";
3241b9c1b51eSKate Stone   case KEY_SAVE:
3242b9c1b51eSKate Stone     return "save key";
3243b9c1b51eSKate Stone   case KEY_SBEG:
3244b9c1b51eSKate Stone     return "shifted begin key";
3245b9c1b51eSKate Stone   case KEY_SCANCEL:
3246b9c1b51eSKate Stone     return "shifted cancel key";
3247b9c1b51eSKate Stone   case KEY_SCOMMAND:
3248b9c1b51eSKate Stone     return "shifted command key";
3249b9c1b51eSKate Stone   case KEY_SCOPY:
3250b9c1b51eSKate Stone     return "shifted copy key";
3251b9c1b51eSKate Stone   case KEY_SCREATE:
3252b9c1b51eSKate Stone     return "shifted create key";
3253b9c1b51eSKate Stone   case KEY_SDC:
3254b9c1b51eSKate Stone     return "shifted delete-character key";
3255b9c1b51eSKate Stone   case KEY_SDL:
3256b9c1b51eSKate Stone     return "shifted delete-line key";
3257b9c1b51eSKate Stone   case KEY_SELECT:
3258b9c1b51eSKate Stone     return "select key";
3259b9c1b51eSKate Stone   case KEY_SEND:
3260b9c1b51eSKate Stone     return "shifted end key";
3261b9c1b51eSKate Stone   case KEY_SEOL:
3262b9c1b51eSKate Stone     return "shifted clear-to-end-of-line key";
3263b9c1b51eSKate Stone   case KEY_SEXIT:
3264b9c1b51eSKate Stone     return "shifted exit key";
3265b9c1b51eSKate Stone   case KEY_SFIND:
3266b9c1b51eSKate Stone     return "shifted find key";
3267b9c1b51eSKate Stone   case KEY_SHELP:
3268b9c1b51eSKate Stone     return "shifted help key";
3269b9c1b51eSKate Stone   case KEY_SHOME:
3270b9c1b51eSKate Stone     return "shifted home key";
3271b9c1b51eSKate Stone   case KEY_SIC:
3272b9c1b51eSKate Stone     return "shifted insert-character key";
3273b9c1b51eSKate Stone   case KEY_SLEFT:
3274b9c1b51eSKate Stone     return "shifted left-arrow key";
3275b9c1b51eSKate Stone   case KEY_SMESSAGE:
3276b9c1b51eSKate Stone     return "shifted message key";
3277b9c1b51eSKate Stone   case KEY_SMOVE:
3278b9c1b51eSKate Stone     return "shifted move key";
3279b9c1b51eSKate Stone   case KEY_SNEXT:
3280b9c1b51eSKate Stone     return "shifted next key";
3281b9c1b51eSKate Stone   case KEY_SOPTIONS:
3282b9c1b51eSKate Stone     return "shifted options key";
3283b9c1b51eSKate Stone   case KEY_SPREVIOUS:
3284b9c1b51eSKate Stone     return "shifted previous key";
3285b9c1b51eSKate Stone   case KEY_SPRINT:
3286b9c1b51eSKate Stone     return "shifted print key";
3287b9c1b51eSKate Stone   case KEY_SREDO:
3288b9c1b51eSKate Stone     return "shifted redo key";
3289b9c1b51eSKate Stone   case KEY_SREPLACE:
3290b9c1b51eSKate Stone     return "shifted replace key";
3291b9c1b51eSKate Stone   case KEY_SRIGHT:
3292b9c1b51eSKate Stone     return "shifted right-arrow key";
3293b9c1b51eSKate Stone   case KEY_SRSUME:
3294b9c1b51eSKate Stone     return "shifted resume key";
3295b9c1b51eSKate Stone   case KEY_SSAVE:
3296b9c1b51eSKate Stone     return "shifted save key";
3297b9c1b51eSKate Stone   case KEY_SSUSPEND:
3298b9c1b51eSKate Stone     return "shifted suspend key";
3299b9c1b51eSKate Stone   case KEY_SUNDO:
3300b9c1b51eSKate Stone     return "shifted undo key";
3301b9c1b51eSKate Stone   case KEY_SUSPEND:
3302b9c1b51eSKate Stone     return "suspend key";
3303b9c1b51eSKate Stone   case KEY_UNDO:
3304b9c1b51eSKate Stone     return "undo key";
3305b9c1b51eSKate Stone   case KEY_MOUSE:
3306b9c1b51eSKate Stone     return "Mouse event has occurred";
3307b9c1b51eSKate Stone   case KEY_RESIZE:
3308b9c1b51eSKate Stone     return "Terminal resize event";
330927801f4fSBruce Mitchener #ifdef KEY_EVENT
3310b9c1b51eSKate Stone   case KEY_EVENT:
3311b9c1b51eSKate Stone     return "We were interrupted by an event";
331227801f4fSBruce Mitchener #endif
3313b9c1b51eSKate Stone   case KEY_RETURN:
3314b9c1b51eSKate Stone     return "return";
3315b9c1b51eSKate Stone   case ' ':
3316b9c1b51eSKate Stone     return "space";
3317b9c1b51eSKate Stone   case '\t':
3318b9c1b51eSKate Stone     return "tab";
3319b9c1b51eSKate Stone   case KEY_ESCAPE:
3320b9c1b51eSKate Stone     return "escape";
332144d93782SGreg Clayton   default:
332244d93782SGreg Clayton     if (isprint(ch))
332344d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "%c", ch);
332444d93782SGreg Clayton     else
332544d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "\\x%2.2x", ch);
332644d93782SGreg Clayton     return g_desc;
332744d93782SGreg Clayton   }
3328c5dac77aSEugene Zelenko   return nullptr;
332944d93782SGreg Clayton }
333044d93782SGreg Clayton 
3331b9c1b51eSKate Stone HelpDialogDelegate::HelpDialogDelegate(const char *text,
3332b9c1b51eSKate Stone                                        KeyHelp *key_help_array)
3333b9c1b51eSKate Stone     : m_text(), m_first_visible_line(0) {
3334b9c1b51eSKate Stone   if (text && text[0]) {
333544d93782SGreg Clayton     m_text.SplitIntoLines(text);
333644d93782SGreg Clayton     m_text.AppendString("");
333744d93782SGreg Clayton   }
3338b9c1b51eSKate Stone   if (key_help_array) {
3339b9c1b51eSKate Stone     for (KeyHelp *key = key_help_array; key->ch; ++key) {
334044d93782SGreg Clayton       StreamString key_description;
3341b9c1b51eSKate Stone       key_description.Printf("%10s - %s", CursesKeyToCString(key->ch),
3342b9c1b51eSKate Stone                              key->description);
3343c156427dSZachary Turner       m_text.AppendString(key_description.GetString());
334444d93782SGreg Clayton     }
334544d93782SGreg Clayton   }
334644d93782SGreg Clayton }
334744d93782SGreg Clayton 
3348315b6884SEugene Zelenko HelpDialogDelegate::~HelpDialogDelegate() = default;
334944d93782SGreg Clayton 
3350b9c1b51eSKate Stone bool HelpDialogDelegate::WindowDelegateDraw(Window &window, bool force) {
335144d93782SGreg Clayton   window.Erase();
335244d93782SGreg Clayton   const int window_height = window.GetHeight();
335344d93782SGreg Clayton   int x = 2;
335444d93782SGreg Clayton   int y = 1;
335544d93782SGreg Clayton   const int min_y = y;
335644d93782SGreg Clayton   const int max_y = window_height - 1 - y;
33573985c8c6SSaleem Abdulrasool   const size_t num_visible_lines = max_y - min_y + 1;
335844d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
335944d93782SGreg Clayton   const char *bottom_message;
336044d93782SGreg Clayton   if (num_lines <= num_visible_lines)
336144d93782SGreg Clayton     bottom_message = "Press any key to exit";
336244d93782SGreg Clayton   else
336344d93782SGreg Clayton     bottom_message = "Use arrows to scroll, any other key to exit";
336444d93782SGreg Clayton   window.DrawTitleBox(window.GetName(), bottom_message);
3365b9c1b51eSKate Stone   while (y <= max_y) {
336644d93782SGreg Clayton     window.MoveCursor(x, y);
3367b9c1b51eSKate Stone     window.PutCStringTruncated(
3368b9c1b51eSKate Stone         m_text.GetStringAtIndex(m_first_visible_line + y - min_y), 1);
336944d93782SGreg Clayton     ++y;
337044d93782SGreg Clayton   }
337144d93782SGreg Clayton   return true;
337244d93782SGreg Clayton }
337344d93782SGreg Clayton 
3374b9c1b51eSKate Stone HandleCharResult HelpDialogDelegate::WindowDelegateHandleChar(Window &window,
3375b9c1b51eSKate Stone                                                               int key) {
337644d93782SGreg Clayton   bool done = false;
337744d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
337844d93782SGreg Clayton   const size_t num_visible_lines = window.GetHeight() - 2;
337944d93782SGreg Clayton 
3380b9c1b51eSKate Stone   if (num_lines <= num_visible_lines) {
338144d93782SGreg Clayton     done = true;
338205097246SAdrian Prantl     // If we have all lines visible and don't need scrolling, then any key
338305097246SAdrian Prantl     // press will cause us to exit
3384b9c1b51eSKate Stone   } else {
3385b9c1b51eSKate Stone     switch (key) {
338644d93782SGreg Clayton     case KEY_UP:
338744d93782SGreg Clayton       if (m_first_visible_line > 0)
338844d93782SGreg Clayton         --m_first_visible_line;
338944d93782SGreg Clayton       break;
339044d93782SGreg Clayton 
339144d93782SGreg Clayton     case KEY_DOWN:
339244d93782SGreg Clayton       if (m_first_visible_line + num_visible_lines < num_lines)
339344d93782SGreg Clayton         ++m_first_visible_line;
339444d93782SGreg Clayton       break;
339544d93782SGreg Clayton 
339644d93782SGreg Clayton     case KEY_PPAGE:
339744d93782SGreg Clayton     case ',':
3398b9c1b51eSKate Stone       if (m_first_visible_line > 0) {
33993985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_visible_lines)
340044d93782SGreg Clayton           m_first_visible_line -= num_visible_lines;
340144d93782SGreg Clayton         else
340244d93782SGreg Clayton           m_first_visible_line = 0;
340344d93782SGreg Clayton       }
340444d93782SGreg Clayton       break;
3405315b6884SEugene Zelenko 
340644d93782SGreg Clayton     case KEY_NPAGE:
340744d93782SGreg Clayton     case '.':
3408b9c1b51eSKate Stone       if (m_first_visible_line + num_visible_lines < num_lines) {
340944d93782SGreg Clayton         m_first_visible_line += num_visible_lines;
34103985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > num_lines)
341144d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
341244d93782SGreg Clayton       }
341344d93782SGreg Clayton       break;
3414315b6884SEugene Zelenko 
341544d93782SGreg Clayton     default:
341644d93782SGreg Clayton       done = true;
341744d93782SGreg Clayton       break;
341844d93782SGreg Clayton     }
341944d93782SGreg Clayton   }
342044d93782SGreg Clayton   if (done)
342144d93782SGreg Clayton     window.GetParent()->RemoveSubWindow(&window);
342244d93782SGreg Clayton   return eKeyHandled;
342344d93782SGreg Clayton }
342444d93782SGreg Clayton 
3425b9c1b51eSKate Stone class ApplicationDelegate : public WindowDelegate, public MenuDelegate {
342644d93782SGreg Clayton public:
342744d93782SGreg Clayton   enum {
342844d93782SGreg Clayton     eMenuID_LLDB = 1,
342944d93782SGreg Clayton     eMenuID_LLDBAbout,
343044d93782SGreg Clayton     eMenuID_LLDBExit,
343144d93782SGreg Clayton 
343244d93782SGreg Clayton     eMenuID_Target,
343344d93782SGreg Clayton     eMenuID_TargetCreate,
343444d93782SGreg Clayton     eMenuID_TargetDelete,
343544d93782SGreg Clayton 
343644d93782SGreg Clayton     eMenuID_Process,
343744d93782SGreg Clayton     eMenuID_ProcessAttach,
343844d93782SGreg Clayton     eMenuID_ProcessDetach,
343944d93782SGreg Clayton     eMenuID_ProcessLaunch,
344044d93782SGreg Clayton     eMenuID_ProcessContinue,
344144d93782SGreg Clayton     eMenuID_ProcessHalt,
344244d93782SGreg Clayton     eMenuID_ProcessKill,
344344d93782SGreg Clayton 
344444d93782SGreg Clayton     eMenuID_Thread,
344544d93782SGreg Clayton     eMenuID_ThreadStepIn,
344644d93782SGreg Clayton     eMenuID_ThreadStepOver,
344744d93782SGreg Clayton     eMenuID_ThreadStepOut,
344844d93782SGreg Clayton 
344944d93782SGreg Clayton     eMenuID_View,
345044d93782SGreg Clayton     eMenuID_ViewBacktrace,
345144d93782SGreg Clayton     eMenuID_ViewRegisters,
345244d93782SGreg Clayton     eMenuID_ViewSource,
345344d93782SGreg Clayton     eMenuID_ViewVariables,
345444d93782SGreg Clayton 
345544d93782SGreg Clayton     eMenuID_Help,
345644d93782SGreg Clayton     eMenuID_HelpGUIHelp
345744d93782SGreg Clayton   };
345844d93782SGreg Clayton 
3459b9c1b51eSKate Stone   ApplicationDelegate(Application &app, Debugger &debugger)
3460b9c1b51eSKate Stone       : WindowDelegate(), MenuDelegate(), m_app(app), m_debugger(debugger) {}
346144d93782SGreg Clayton 
3462315b6884SEugene Zelenko   ~ApplicationDelegate() override = default;
3463bd5ae6b4SGreg Clayton 
3464b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
346544d93782SGreg Clayton     return false; // Drawing not handled, let standard window drawing happen
346644d93782SGreg Clayton   }
346744d93782SGreg Clayton 
3468b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override {
3469b9c1b51eSKate Stone     switch (key) {
34705fdb09bbSGreg Clayton     case '\t':
347144d93782SGreg Clayton       window.SelectNextWindowAsActive();
347244d93782SGreg Clayton       return eKeyHandled;
34735fdb09bbSGreg Clayton 
34745fdb09bbSGreg Clayton     case 'h':
34755fdb09bbSGreg Clayton       window.CreateHelpSubwindow();
34765fdb09bbSGreg Clayton       return eKeyHandled;
34775fdb09bbSGreg Clayton 
34785fdb09bbSGreg Clayton     case KEY_ESCAPE:
34795fdb09bbSGreg Clayton       return eQuitApplication;
34805fdb09bbSGreg Clayton 
34815fdb09bbSGreg Clayton     default:
34825fdb09bbSGreg Clayton       break;
348344d93782SGreg Clayton     }
348444d93782SGreg Clayton     return eKeyNotHandled;
348544d93782SGreg Clayton   }
348644d93782SGreg Clayton 
3487b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
34885fdb09bbSGreg Clayton     return "Welcome to the LLDB curses GUI.\n\n"
34895fdb09bbSGreg Clayton            "Press the TAB key to change the selected view.\n"
3490b9c1b51eSKate Stone            "Each view has its own keyboard shortcuts, press 'h' to open a "
3491b9c1b51eSKate Stone            "dialog to display them.\n\n"
34925fdb09bbSGreg Clayton            "Common key bindings for all views:";
34935fdb09bbSGreg Clayton   }
34945fdb09bbSGreg Clayton 
3495b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
34965fdb09bbSGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
34975fdb09bbSGreg Clayton         {'\t', "Select next view"},
34985fdb09bbSGreg Clayton         {'h', "Show help dialog with view specific key bindings"},
34995fdb09bbSGreg Clayton         {',', "Page up"},
35005fdb09bbSGreg Clayton         {'.', "Page down"},
35015fdb09bbSGreg Clayton         {KEY_UP, "Select previous"},
35025fdb09bbSGreg Clayton         {KEY_DOWN, "Select next"},
35035fdb09bbSGreg Clayton         {KEY_LEFT, "Unexpand or select parent"},
35045fdb09bbSGreg Clayton         {KEY_RIGHT, "Expand"},
35055fdb09bbSGreg Clayton         {KEY_PPAGE, "Page up"},
35065fdb09bbSGreg Clayton         {KEY_NPAGE, "Page down"},
3507b9c1b51eSKate Stone         {'\0', nullptr}};
35085fdb09bbSGreg Clayton     return g_source_view_key_help;
35095fdb09bbSGreg Clayton   }
35105fdb09bbSGreg Clayton 
3511b9c1b51eSKate Stone   MenuActionResult MenuDelegateAction(Menu &menu) override {
3512b9c1b51eSKate Stone     switch (menu.GetIdentifier()) {
3513b9c1b51eSKate Stone     case eMenuID_ThreadStepIn: {
3514b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3515b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3516b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
351744d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3518b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3519b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
35204b4b2478SJim Ingham           exe_ctx.GetThreadRef().StepIn(true);
352144d93782SGreg Clayton       }
352244d93782SGreg Clayton     }
352344d93782SGreg Clayton       return MenuActionResult::Handled;
352444d93782SGreg Clayton 
3525b9c1b51eSKate Stone     case eMenuID_ThreadStepOut: {
3526b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3527b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3528b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
352944d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3530b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3531b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
353244d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
353344d93782SGreg Clayton       }
353444d93782SGreg Clayton     }
353544d93782SGreg Clayton       return MenuActionResult::Handled;
353644d93782SGreg Clayton 
3537b9c1b51eSKate Stone     case eMenuID_ThreadStepOver: {
3538b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3539b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3540b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
354144d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3542b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3543b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
354444d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOver(true);
354544d93782SGreg Clayton       }
354644d93782SGreg Clayton     }
354744d93782SGreg Clayton       return MenuActionResult::Handled;
354844d93782SGreg Clayton 
3549b9c1b51eSKate Stone     case eMenuID_ProcessContinue: {
3550b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3551b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3552b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
355344d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3554b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3555b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
355644d93782SGreg Clayton           process->Resume();
355744d93782SGreg Clayton       }
355844d93782SGreg Clayton     }
355944d93782SGreg Clayton       return MenuActionResult::Handled;
356044d93782SGreg Clayton 
3561b9c1b51eSKate Stone     case eMenuID_ProcessKill: {
3562b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3563b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3564b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
356544d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
356644d93782SGreg Clayton         if (process && process->IsAlive())
3567ede3193bSJason Molenda           process->Destroy(false);
356844d93782SGreg Clayton       }
356944d93782SGreg Clayton     }
357044d93782SGreg Clayton       return MenuActionResult::Handled;
357144d93782SGreg Clayton 
3572b9c1b51eSKate Stone     case eMenuID_ProcessHalt: {
3573b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3574b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3575b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
357644d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
357744d93782SGreg Clayton         if (process && process->IsAlive())
357844d93782SGreg Clayton           process->Halt();
357944d93782SGreg Clayton       }
358044d93782SGreg Clayton     }
358144d93782SGreg Clayton       return MenuActionResult::Handled;
358244d93782SGreg Clayton 
3583b9c1b51eSKate Stone     case eMenuID_ProcessDetach: {
3584b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3585b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3586b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
358744d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
358844d93782SGreg Clayton         if (process && process->IsAlive())
358944d93782SGreg Clayton           process->Detach(false);
359044d93782SGreg Clayton       }
359144d93782SGreg Clayton     }
359244d93782SGreg Clayton       return MenuActionResult::Handled;
359344d93782SGreg Clayton 
3594b9c1b51eSKate Stone     case eMenuID_Process: {
3595b9c1b51eSKate Stone       // Populate the menu with all of the threads if the process is stopped
359605097246SAdrian Prantl       // when the Process menu gets selected and is about to display its
359705097246SAdrian Prantl       // submenu.
359844d93782SGreg Clayton       Menus &submenus = menu.GetSubmenus();
3599b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3600b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
360144d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
3602b9c1b51eSKate Stone       if (process && process->IsAlive() &&
3603b9c1b51eSKate Stone           StateIsStoppedState(process->GetState(), true)) {
360444d93782SGreg Clayton         if (submenus.size() == 7)
360544d93782SGreg Clayton           menu.AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
360644d93782SGreg Clayton         else if (submenus.size() > 8)
360744d93782SGreg Clayton           submenus.erase(submenus.begin() + 8, submenus.end());
360844d93782SGreg Clayton 
360944d93782SGreg Clayton         ThreadList &threads = process->GetThreadList();
3610bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
361144d93782SGreg Clayton         size_t num_threads = threads.GetSize();
3612b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
361344d93782SGreg Clayton           ThreadSP thread_sp = threads.GetThreadAtIndex(i);
361444d93782SGreg Clayton           char menu_char = '\0';
361544d93782SGreg Clayton           if (i < 9)
361644d93782SGreg Clayton             menu_char = '1' + i;
361744d93782SGreg Clayton           StreamString thread_menu_title;
361844d93782SGreg Clayton           thread_menu_title.Printf("Thread %u", thread_sp->GetIndexID());
361944d93782SGreg Clayton           const char *thread_name = thread_sp->GetName();
362044d93782SGreg Clayton           if (thread_name && thread_name[0])
362144d93782SGreg Clayton             thread_menu_title.Printf(" %s", thread_name);
3622b9c1b51eSKate Stone           else {
362344d93782SGreg Clayton             const char *queue_name = thread_sp->GetQueueName();
362444d93782SGreg Clayton             if (queue_name && queue_name[0])
362544d93782SGreg Clayton               thread_menu_title.Printf(" %s", queue_name);
362644d93782SGreg Clayton           }
3627b9c1b51eSKate Stone           menu.AddSubmenu(
3628c156427dSZachary Turner               MenuSP(new Menu(thread_menu_title.GetString().str().c_str(),
3629c156427dSZachary Turner                               nullptr, menu_char, thread_sp->GetID())));
363044d93782SGreg Clayton         }
3631b9c1b51eSKate Stone       } else if (submenus.size() > 7) {
363205097246SAdrian Prantl         // Remove the separator and any other thread submenu items that were
363305097246SAdrian Prantl         // previously added
363444d93782SGreg Clayton         submenus.erase(submenus.begin() + 7, submenus.end());
363544d93782SGreg Clayton       }
3636b9c1b51eSKate Stone       // Since we are adding and removing items we need to recalculate the name
3637b9c1b51eSKate Stone       // lengths
363844d93782SGreg Clayton       menu.RecalculateNameLengths();
363944d93782SGreg Clayton     }
364044d93782SGreg Clayton       return MenuActionResult::Handled;
364144d93782SGreg Clayton 
3642b9c1b51eSKate Stone     case eMenuID_ViewVariables: {
364344d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
364444d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
364544d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
364644d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
364744d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
364844d93782SGreg Clayton 
3649b9c1b51eSKate Stone       if (variables_window_sp) {
365044d93782SGreg Clayton         const Rect variables_bounds = variables_window_sp->GetBounds();
365144d93782SGreg Clayton 
365244d93782SGreg Clayton         main_window_sp->RemoveSubWindow(variables_window_sp.get());
365344d93782SGreg Clayton 
3654b9c1b51eSKate Stone         if (registers_window_sp) {
3655b9c1b51eSKate Stone           // We have a registers window, so give all the area back to the
3656b9c1b51eSKate Stone           // registers window
365744d93782SGreg Clayton           Rect registers_bounds = variables_bounds;
365844d93782SGreg Clayton           registers_bounds.size.width = source_bounds.size.width;
365944d93782SGreg Clayton           registers_window_sp->SetBounds(registers_bounds);
3660b9c1b51eSKate Stone         } else {
366105097246SAdrian Prantl           // We have no registers window showing so give the bottom area back
366205097246SAdrian Prantl           // to the source view
366344d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3664b9c1b51eSKate Stone                                    source_bounds.size.height +
3665b9c1b51eSKate Stone                                        variables_bounds.size.height);
366644d93782SGreg Clayton         }
3667b9c1b51eSKate Stone       } else {
366844d93782SGreg Clayton         Rect new_variables_rect;
3669b9c1b51eSKate Stone         if (registers_window_sp) {
367044d93782SGreg Clayton           // We have a registers window so split the area of the registers
367144d93782SGreg Clayton           // window into two columns where the left hand side will be the
367244d93782SGreg Clayton           // variables and the right hand side will be the registers
367344d93782SGreg Clayton           const Rect variables_bounds = registers_window_sp->GetBounds();
367444d93782SGreg Clayton           Rect new_registers_rect;
3675b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_variables_rect,
3676b9c1b51eSKate Stone                                                    new_registers_rect);
367744d93782SGreg Clayton           registers_window_sp->SetBounds(new_registers_rect);
3678b9c1b51eSKate Stone         } else {
367944d93782SGreg Clayton           // No variables window, grab the bottom part of the source window
368044d93782SGreg Clayton           Rect new_source_rect;
3681b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3682b9c1b51eSKate Stone                                                   new_variables_rect);
368344d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
368444d93782SGreg Clayton         }
3685b9c1b51eSKate Stone         WindowSP new_window_sp = main_window_sp->CreateSubWindow(
3686b9c1b51eSKate Stone             "Variables", new_variables_rect, false);
3687b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3688b9c1b51eSKate Stone             WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
368944d93782SGreg Clayton       }
369044d93782SGreg Clayton       touchwin(stdscr);
369144d93782SGreg Clayton     }
369244d93782SGreg Clayton       return MenuActionResult::Handled;
369344d93782SGreg Clayton 
3694b9c1b51eSKate Stone     case eMenuID_ViewRegisters: {
369544d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
369644d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
369744d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
369844d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
369944d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
370044d93782SGreg Clayton 
3701b9c1b51eSKate Stone       if (registers_window_sp) {
3702b9c1b51eSKate Stone         if (variables_window_sp) {
370344d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
370444d93782SGreg Clayton 
3705b9c1b51eSKate Stone           // We have a variables window, so give all the area back to the
3706b9c1b51eSKate Stone           // variables window
3707b9c1b51eSKate Stone           variables_window_sp->Resize(variables_bounds.size.width +
3708b9c1b51eSKate Stone                                           registers_window_sp->GetWidth(),
370944d93782SGreg Clayton                                       variables_bounds.size.height);
3710b9c1b51eSKate Stone         } else {
371105097246SAdrian Prantl           // We have no variables window showing so give the bottom area back
371205097246SAdrian Prantl           // to the source view
371344d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3714b9c1b51eSKate Stone                                    source_bounds.size.height +
3715b9c1b51eSKate Stone                                        registers_window_sp->GetHeight());
371644d93782SGreg Clayton         }
371744d93782SGreg Clayton         main_window_sp->RemoveSubWindow(registers_window_sp.get());
3718b9c1b51eSKate Stone       } else {
371944d93782SGreg Clayton         Rect new_regs_rect;
3720b9c1b51eSKate Stone         if (variables_window_sp) {
372105097246SAdrian Prantl           // We have a variables window, split it into two columns where the
372205097246SAdrian Prantl           // left hand side will be the variables and the right hand side will
372305097246SAdrian Prantl           // be the registers
372444d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
372544d93782SGreg Clayton           Rect new_vars_rect;
3726b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_vars_rect,
3727b9c1b51eSKate Stone                                                    new_regs_rect);
372844d93782SGreg Clayton           variables_window_sp->SetBounds(new_vars_rect);
3729b9c1b51eSKate Stone         } else {
373044d93782SGreg Clayton           // No registers window, grab the bottom part of the source window
373144d93782SGreg Clayton           Rect new_source_rect;
3732b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3733b9c1b51eSKate Stone                                                   new_regs_rect);
373444d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
373544d93782SGreg Clayton         }
3736b9c1b51eSKate Stone         WindowSP new_window_sp =
3737b9c1b51eSKate Stone             main_window_sp->CreateSubWindow("Registers", new_regs_rect, false);
3738b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3739b9c1b51eSKate Stone             WindowDelegateSP(new RegistersWindowDelegate(m_debugger)));
374044d93782SGreg Clayton       }
374144d93782SGreg Clayton       touchwin(stdscr);
374244d93782SGreg Clayton     }
374344d93782SGreg Clayton       return MenuActionResult::Handled;
374444d93782SGreg Clayton 
374544d93782SGreg Clayton     case eMenuID_HelpGUIHelp:
37465fdb09bbSGreg Clayton       m_app.GetMainWindow()->CreateHelpSubwindow();
374744d93782SGreg Clayton       return MenuActionResult::Handled;
374844d93782SGreg Clayton 
374944d93782SGreg Clayton     default:
375044d93782SGreg Clayton       break;
375144d93782SGreg Clayton     }
375244d93782SGreg Clayton 
375344d93782SGreg Clayton     return MenuActionResult::NotHandled;
375444d93782SGreg Clayton   }
3755b9c1b51eSKate Stone 
375644d93782SGreg Clayton protected:
375744d93782SGreg Clayton   Application &m_app;
375844d93782SGreg Clayton   Debugger &m_debugger;
375944d93782SGreg Clayton };
376044d93782SGreg Clayton 
3761b9c1b51eSKate Stone class StatusBarWindowDelegate : public WindowDelegate {
376244d93782SGreg Clayton public:
3763b9c1b51eSKate Stone   StatusBarWindowDelegate(Debugger &debugger) : m_debugger(debugger) {
3764b9c1b51eSKate Stone     FormatEntity::Parse("Thread: ${thread.id%tid}", m_format);
376544d93782SGreg Clayton   }
376644d93782SGreg Clayton 
3767315b6884SEugene Zelenko   ~StatusBarWindowDelegate() override = default;
3768bd5ae6b4SGreg Clayton 
3769b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3770b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3771b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
377244d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
377344d93782SGreg Clayton     Thread *thread = exe_ctx.GetThreadPtr();
377444d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
377544d93782SGreg Clayton     window.Erase();
377644d93782SGreg Clayton     window.SetBackground(2);
377744d93782SGreg Clayton     window.MoveCursor(0, 0);
3778b9c1b51eSKate Stone     if (process) {
377944d93782SGreg Clayton       const StateType state = process->GetState();
3780b9c1b51eSKate Stone       window.Printf("Process: %5" PRIu64 " %10s", process->GetID(),
3781b9c1b51eSKate Stone                     StateAsCString(state));
378244d93782SGreg Clayton 
3783b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
37845b031ebcSEd Maste         StreamString strm;
3785b9c1b51eSKate Stone         if (thread && FormatEntity::Format(m_format, strm, nullptr, &exe_ctx,
3786b9c1b51eSKate Stone                                            nullptr, nullptr, false, false)) {
378744d93782SGreg Clayton           window.MoveCursor(40, 0);
3788c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), 1);
37895b031ebcSEd Maste         }
379044d93782SGreg Clayton 
379144d93782SGreg Clayton         window.MoveCursor(60, 0);
379244d93782SGreg Clayton         if (frame)
3793b9c1b51eSKate Stone           window.Printf("Frame: %3u  PC = 0x%16.16" PRIx64,
3794b9c1b51eSKate Stone                         frame->GetFrameIndex(),
3795b9c1b51eSKate Stone                         frame->GetFrameCodeAddress().GetOpcodeLoadAddress(
3796b9c1b51eSKate Stone                             exe_ctx.GetTargetPtr()));
3797b9c1b51eSKate Stone       } else if (state == eStateExited) {
379844d93782SGreg Clayton         const char *exit_desc = process->GetExitDescription();
379944d93782SGreg Clayton         const int exit_status = process->GetExitStatus();
380044d93782SGreg Clayton         if (exit_desc && exit_desc[0])
380144d93782SGreg Clayton           window.Printf(" with status = %i (%s)", exit_status, exit_desc);
380244d93782SGreg Clayton         else
380344d93782SGreg Clayton           window.Printf(" with status = %i", exit_status);
380444d93782SGreg Clayton       }
380544d93782SGreg Clayton     }
380644d93782SGreg Clayton     window.DeferredRefresh();
380744d93782SGreg Clayton     return true;
380844d93782SGreg Clayton   }
380944d93782SGreg Clayton 
381044d93782SGreg Clayton protected:
381144d93782SGreg Clayton   Debugger &m_debugger;
3812554f68d3SGreg Clayton   FormatEntity::Entry m_format;
381344d93782SGreg Clayton };
381444d93782SGreg Clayton 
3815b9c1b51eSKate Stone class SourceFileWindowDelegate : public WindowDelegate {
381644d93782SGreg Clayton public:
3817b9c1b51eSKate Stone   SourceFileWindowDelegate(Debugger &debugger)
3818b9c1b51eSKate Stone       : WindowDelegate(), m_debugger(debugger), m_sc(), m_file_sp(),
3819b9c1b51eSKate Stone         m_disassembly_scope(nullptr), m_disassembly_sp(), m_disassembly_range(),
3820b9c1b51eSKate Stone         m_title(), m_line_width(4), m_selected_line(0), m_pc_line(0),
3821b9c1b51eSKate Stone         m_stop_id(0), m_frame_idx(UINT32_MAX), m_first_visible_line(0),
3822b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
382344d93782SGreg Clayton 
3824315b6884SEugene Zelenko   ~SourceFileWindowDelegate() override = default;
382544d93782SGreg Clayton 
3826b9c1b51eSKate Stone   void Update(const SymbolContext &sc) { m_sc = sc; }
382744d93782SGreg Clayton 
3828b9c1b51eSKate Stone   uint32_t NumVisibleLines() const { return m_max_y - m_min_y; }
382944d93782SGreg Clayton 
3830b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
383144d93782SGreg Clayton     return "Source/Disassembly window keyboard shortcuts:";
383244d93782SGreg Clayton   }
383344d93782SGreg Clayton 
3834b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
383544d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
383644d93782SGreg Clayton         {KEY_RETURN, "Run to selected line with one shot breakpoint"},
383744d93782SGreg Clayton         {KEY_UP, "Select previous source line"},
383844d93782SGreg Clayton         {KEY_DOWN, "Select next source line"},
383944d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
384044d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
384144d93782SGreg Clayton         {'b', "Set breakpoint on selected source/disassembly line"},
384244d93782SGreg Clayton         {'c', "Continue process"},
384344d93782SGreg Clayton         {'d', "Detach and resume process"},
384444d93782SGreg Clayton         {'D', "Detach with process suspended"},
384544d93782SGreg Clayton         {'h', "Show help dialog"},
384644d93782SGreg Clayton         {'k', "Kill process"},
384744d93782SGreg Clayton         {'n', "Step over (source line)"},
384844d93782SGreg Clayton         {'N', "Step over (single instruction)"},
384944d93782SGreg Clayton         {'o', "Step out"},
385044d93782SGreg Clayton         {'s', "Step in (source line)"},
385144d93782SGreg Clayton         {'S', "Step in (single instruction)"},
385244d93782SGreg Clayton         {',', "Page up"},
385344d93782SGreg Clayton         {'.', "Page down"},
3854b9c1b51eSKate Stone         {'\0', nullptr}};
385544d93782SGreg Clayton     return g_source_view_key_help;
385644d93782SGreg Clayton   }
385744d93782SGreg Clayton 
3858b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3859b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3860b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
386144d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
3862c5dac77aSEugene Zelenko     Thread *thread = nullptr;
386344d93782SGreg Clayton 
386444d93782SGreg Clayton     bool update_location = false;
3865b9c1b51eSKate Stone     if (process) {
386644d93782SGreg Clayton       StateType state = process->GetState();
3867b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
386844d93782SGreg Clayton         // We are stopped, so it is ok to
386944d93782SGreg Clayton         update_location = true;
387044d93782SGreg Clayton       }
387144d93782SGreg Clayton     }
387244d93782SGreg Clayton 
387344d93782SGreg Clayton     m_min_x = 1;
3874ec990867SGreg Clayton     m_min_y = 2;
387544d93782SGreg Clayton     m_max_x = window.GetMaxX() - 1;
387644d93782SGreg Clayton     m_max_y = window.GetMaxY() - 1;
387744d93782SGreg Clayton 
387844d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
387944d93782SGreg Clayton     StackFrameSP frame_sp;
388044d93782SGreg Clayton     bool set_selected_line_to_pc = false;
388144d93782SGreg Clayton 
3882b9c1b51eSKate Stone     if (update_location) {
388344d93782SGreg Clayton       const bool process_alive = process ? process->IsAlive() : false;
388444d93782SGreg Clayton       bool thread_changed = false;
3885b9c1b51eSKate Stone       if (process_alive) {
388644d93782SGreg Clayton         thread = exe_ctx.GetThreadPtr();
3887b9c1b51eSKate Stone         if (thread) {
388844d93782SGreg Clayton           frame_sp = thread->GetSelectedFrame();
388944d93782SGreg Clayton           auto tid = thread->GetID();
389044d93782SGreg Clayton           thread_changed = tid != m_tid;
389144d93782SGreg Clayton           m_tid = tid;
3892b9c1b51eSKate Stone         } else {
3893b9c1b51eSKate Stone           if (m_tid != LLDB_INVALID_THREAD_ID) {
389444d93782SGreg Clayton             thread_changed = true;
389544d93782SGreg Clayton             m_tid = LLDB_INVALID_THREAD_ID;
389644d93782SGreg Clayton           }
389744d93782SGreg Clayton         }
389844d93782SGreg Clayton       }
389944d93782SGreg Clayton       const uint32_t stop_id = process ? process->GetStopID() : 0;
390044d93782SGreg Clayton       const bool stop_id_changed = stop_id != m_stop_id;
390144d93782SGreg Clayton       bool frame_changed = false;
390244d93782SGreg Clayton       m_stop_id = stop_id;
3903ec990867SGreg Clayton       m_title.Clear();
3904b9c1b51eSKate Stone       if (frame_sp) {
390544d93782SGreg Clayton         m_sc = frame_sp->GetSymbolContext(eSymbolContextEverything);
3906b9c1b51eSKate Stone         if (m_sc.module_sp) {
3907b9c1b51eSKate Stone           m_title.Printf(
3908b9c1b51eSKate Stone               "%s", m_sc.module_sp->GetFileSpec().GetFilename().GetCString());
3909ec990867SGreg Clayton           ConstString func_name = m_sc.GetFunctionName();
3910ec990867SGreg Clayton           if (func_name)
3911ec990867SGreg Clayton             m_title.Printf("`%s", func_name.GetCString());
3912ec990867SGreg Clayton         }
391344d93782SGreg Clayton         const uint32_t frame_idx = frame_sp->GetFrameIndex();
391444d93782SGreg Clayton         frame_changed = frame_idx != m_frame_idx;
391544d93782SGreg Clayton         m_frame_idx = frame_idx;
3916b9c1b51eSKate Stone       } else {
391744d93782SGreg Clayton         m_sc.Clear(true);
391844d93782SGreg Clayton         frame_changed = m_frame_idx != UINT32_MAX;
391944d93782SGreg Clayton         m_frame_idx = UINT32_MAX;
392044d93782SGreg Clayton       }
392144d93782SGreg Clayton 
3922b9c1b51eSKate Stone       const bool context_changed =
3923b9c1b51eSKate Stone           thread_changed || frame_changed || stop_id_changed;
392444d93782SGreg Clayton 
3925b9c1b51eSKate Stone       if (process_alive) {
3926b9c1b51eSKate Stone         if (m_sc.line_entry.IsValid()) {
392744d93782SGreg Clayton           m_pc_line = m_sc.line_entry.line;
392844d93782SGreg Clayton           if (m_pc_line != UINT32_MAX)
392944d93782SGreg Clayton             --m_pc_line; // Convert to zero based line number...
393044d93782SGreg Clayton           // Update the selected line if the stop ID changed...
393144d93782SGreg Clayton           if (context_changed)
393244d93782SGreg Clayton             m_selected_line = m_pc_line;
393344d93782SGreg Clayton 
3934b9c1b51eSKate Stone           if (m_file_sp && m_file_sp->FileSpecMatches(m_sc.line_entry.file)) {
393505097246SAdrian Prantl             // Same file, nothing to do, we should either have the lines or not
393605097246SAdrian Prantl             // (source file missing)
3937b9c1b51eSKate Stone             if (m_selected_line >= static_cast<size_t>(m_first_visible_line)) {
393844d93782SGreg Clayton               if (m_selected_line >= m_first_visible_line + num_visible_lines)
393944d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
3940b9c1b51eSKate Stone             } else {
394144d93782SGreg Clayton               if (m_selected_line > 10)
394244d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
394344d93782SGreg Clayton               else
394444d93782SGreg Clayton                 m_first_visible_line = 0;
394544d93782SGreg Clayton             }
3946b9c1b51eSKate Stone           } else {
394744d93782SGreg Clayton             // File changed, set selected line to the line with the PC
394844d93782SGreg Clayton             m_selected_line = m_pc_line;
3949b9c1b51eSKate Stone             m_file_sp =
3950b9c1b51eSKate Stone                 m_debugger.GetSourceManager().GetFile(m_sc.line_entry.file);
3951b9c1b51eSKate Stone             if (m_file_sp) {
395244d93782SGreg Clayton               const size_t num_lines = m_file_sp->GetNumLines();
3953eebf32faSPavel Labath               m_line_width = 1;
395444d93782SGreg Clayton               for (size_t n = num_lines; n >= 10; n = n / 10)
395544d93782SGreg Clayton                 ++m_line_width;
395644d93782SGreg Clayton 
3957b9c1b51eSKate Stone               if (num_lines < num_visible_lines ||
3958b9c1b51eSKate Stone                   m_selected_line < num_visible_lines)
395944d93782SGreg Clayton                 m_first_visible_line = 0;
396044d93782SGreg Clayton               else
396144d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
396244d93782SGreg Clayton             }
396344d93782SGreg Clayton           }
3964b9c1b51eSKate Stone         } else {
396544d93782SGreg Clayton           m_file_sp.reset();
396644d93782SGreg Clayton         }
396744d93782SGreg Clayton 
3968b9c1b51eSKate Stone         if (!m_file_sp || m_file_sp->GetNumLines() == 0) {
396944d93782SGreg Clayton           // Show disassembly
397044d93782SGreg Clayton           bool prefer_file_cache = false;
3971b9c1b51eSKate Stone           if (m_sc.function) {
3972b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.function) {
397344d93782SGreg Clayton               m_disassembly_scope = m_sc.function;
3974b9c1b51eSKate Stone               m_disassembly_sp = m_sc.function->GetInstructions(
3975b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3976b9c1b51eSKate Stone               if (m_disassembly_sp) {
397744d93782SGreg Clayton                 set_selected_line_to_pc = true;
397844d93782SGreg Clayton                 m_disassembly_range = m_sc.function->GetAddressRange();
3979b9c1b51eSKate Stone               } else {
398044d93782SGreg Clayton                 m_disassembly_range.Clear();
398144d93782SGreg Clayton               }
3982b9c1b51eSKate Stone             } else {
398344d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
398444d93782SGreg Clayton             }
3985b9c1b51eSKate Stone           } else if (m_sc.symbol) {
3986b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.symbol) {
398744d93782SGreg Clayton               m_disassembly_scope = m_sc.symbol;
3988b9c1b51eSKate Stone               m_disassembly_sp = m_sc.symbol->GetInstructions(
3989b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3990b9c1b51eSKate Stone               if (m_disassembly_sp) {
399144d93782SGreg Clayton                 set_selected_line_to_pc = true;
3992b9c1b51eSKate Stone                 m_disassembly_range.GetBaseAddress() =
3993b9c1b51eSKate Stone                     m_sc.symbol->GetAddress();
399444d93782SGreg Clayton                 m_disassembly_range.SetByteSize(m_sc.symbol->GetByteSize());
3995b9c1b51eSKate Stone               } else {
399644d93782SGreg Clayton                 m_disassembly_range.Clear();
399744d93782SGreg Clayton               }
3998b9c1b51eSKate Stone             } else {
399944d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
400044d93782SGreg Clayton             }
400144d93782SGreg Clayton           }
400244d93782SGreg Clayton         }
4003b9c1b51eSKate Stone       } else {
400444d93782SGreg Clayton         m_pc_line = UINT32_MAX;
400544d93782SGreg Clayton       }
400644d93782SGreg Clayton     }
400744d93782SGreg Clayton 
4008ec990867SGreg Clayton     const int window_width = window.GetWidth();
400944d93782SGreg Clayton     window.Erase();
401044d93782SGreg Clayton     window.DrawTitleBox("Sources");
4011b9c1b51eSKate Stone     if (!m_title.GetString().empty()) {
4012ec990867SGreg Clayton       window.AttributeOn(A_REVERSE);
4013ec990867SGreg Clayton       window.MoveCursor(1, 1);
4014ec990867SGreg Clayton       window.PutChar(' ');
4015c156427dSZachary Turner       window.PutCStringTruncated(m_title.GetString().str().c_str(), 1);
4016ec990867SGreg Clayton       int x = window.GetCursorX();
4017b9c1b51eSKate Stone       if (x < window_width - 1) {
4018ec990867SGreg Clayton         window.Printf("%*s", window_width - x - 1, "");
4019ec990867SGreg Clayton       }
4020ec990867SGreg Clayton       window.AttributeOff(A_REVERSE);
4021ec990867SGreg Clayton     }
402244d93782SGreg Clayton 
402344d93782SGreg Clayton     Target *target = exe_ctx.GetTargetPtr();
402444d93782SGreg Clayton     const size_t num_source_lines = GetNumSourceLines();
4025b9c1b51eSKate Stone     if (num_source_lines > 0) {
402644d93782SGreg Clayton       // Display source
402744d93782SGreg Clayton       BreakpointLines bp_lines;
4028b9c1b51eSKate Stone       if (target) {
402944d93782SGreg Clayton         BreakpointList &bp_list = target->GetBreakpointList();
403044d93782SGreg Clayton         const size_t num_bps = bp_list.GetSize();
4031b9c1b51eSKate Stone         for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
403244d93782SGreg Clayton           BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
403344d93782SGreg Clayton           const size_t num_bps_locs = bp_sp->GetNumLocations();
4034b9c1b51eSKate Stone           for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
4035b9c1b51eSKate Stone             BreakpointLocationSP bp_loc_sp =
4036b9c1b51eSKate Stone                 bp_sp->GetLocationAtIndex(bp_loc_idx);
403744d93782SGreg Clayton             LineEntry bp_loc_line_entry;
4038b9c1b51eSKate Stone             if (bp_loc_sp->GetAddress().CalculateSymbolContextLineEntry(
4039b9c1b51eSKate Stone                     bp_loc_line_entry)) {
4040b9c1b51eSKate Stone               if (m_file_sp->GetFileSpec() == bp_loc_line_entry.file) {
404144d93782SGreg Clayton                 bp_lines.insert(bp_loc_line_entry.line);
404244d93782SGreg Clayton               }
404344d93782SGreg Clayton             }
404444d93782SGreg Clayton           }
404544d93782SGreg Clayton         }
404644d93782SGreg Clayton       }
404744d93782SGreg Clayton 
404844d93782SGreg Clayton       const attr_t selected_highlight_attr = A_REVERSE;
404944d93782SGreg Clayton       const attr_t pc_highlight_attr = COLOR_PAIR(1);
405044d93782SGreg Clayton 
4051b9c1b51eSKate Stone       for (size_t i = 0; i < num_visible_lines; ++i) {
405244d93782SGreg Clayton         const uint32_t curr_line = m_first_visible_line + i;
4053b9c1b51eSKate Stone         if (curr_line < num_source_lines) {
4054ec990867SGreg Clayton           const int line_y = m_min_y + i;
405544d93782SGreg Clayton           window.MoveCursor(1, line_y);
405644d93782SGreg Clayton           const bool is_pc_line = curr_line == m_pc_line;
405744d93782SGreg Clayton           const bool line_is_selected = m_selected_line == curr_line;
405844d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
405944d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
406044d93782SGreg Clayton           attr_t highlight_attr = 0;
406144d93782SGreg Clayton           attr_t bp_attr = 0;
406244d93782SGreg Clayton           if (is_pc_line)
406344d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
406444d93782SGreg Clayton           else if (line_is_selected)
406544d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
406644d93782SGreg Clayton 
406744d93782SGreg Clayton           if (bp_lines.find(curr_line + 1) != bp_lines.end())
406844d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
406944d93782SGreg Clayton 
407044d93782SGreg Clayton           if (bp_attr)
407144d93782SGreg Clayton             window.AttributeOn(bp_attr);
407244d93782SGreg Clayton 
4073eebf32faSPavel Labath           window.Printf(" %*u ", m_line_width, curr_line + 1);
407444d93782SGreg Clayton 
407544d93782SGreg Clayton           if (bp_attr)
407644d93782SGreg Clayton             window.AttributeOff(bp_attr);
407744d93782SGreg Clayton 
407844d93782SGreg Clayton           window.PutChar(ACS_VLINE);
407944d93782SGreg Clayton           // Mark the line with the PC with a diamond
408044d93782SGreg Clayton           if (is_pc_line)
408144d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
408244d93782SGreg Clayton           else
408344d93782SGreg Clayton             window.PutChar(' ');
408444d93782SGreg Clayton 
408544d93782SGreg Clayton           if (highlight_attr)
408644d93782SGreg Clayton             window.AttributeOn(highlight_attr);
4087b9c1b51eSKate Stone           const uint32_t line_len =
4088b9c1b51eSKate Stone               m_file_sp->GetLineLength(curr_line + 1, false);
408944d93782SGreg Clayton           if (line_len > 0)
409044d93782SGreg Clayton             window.PutCString(m_file_sp->PeekLineData(curr_line + 1), line_len);
409144d93782SGreg Clayton 
4092b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4093b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
409444d93782SGreg Clayton             StopInfoSP stop_info_sp;
409544d93782SGreg Clayton             if (thread)
409644d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4097b9c1b51eSKate Stone             if (stop_info_sp) {
409844d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4099b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
410044d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4101ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
410244d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4103b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4104b9c1b51eSKate Stone                 // line_y);
4105b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4106b9c1b51eSKate Stone                               stop_description);
410744d93782SGreg Clayton               }
4108b9c1b51eSKate Stone             } else {
4109ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
411044d93782SGreg Clayton             }
411144d93782SGreg Clayton           }
411244d93782SGreg Clayton           if (highlight_attr)
411344d93782SGreg Clayton             window.AttributeOff(highlight_attr);
4114b9c1b51eSKate Stone         } else {
411544d93782SGreg Clayton           break;
411644d93782SGreg Clayton         }
411744d93782SGreg Clayton       }
4118b9c1b51eSKate Stone     } else {
411944d93782SGreg Clayton       size_t num_disassembly_lines = GetNumDisassemblyLines();
4120b9c1b51eSKate Stone       if (num_disassembly_lines > 0) {
412144d93782SGreg Clayton         // Display disassembly
412244d93782SGreg Clayton         BreakpointAddrs bp_file_addrs;
412344d93782SGreg Clayton         Target *target = exe_ctx.GetTargetPtr();
4124b9c1b51eSKate Stone         if (target) {
412544d93782SGreg Clayton           BreakpointList &bp_list = target->GetBreakpointList();
412644d93782SGreg Clayton           const size_t num_bps = bp_list.GetSize();
4127b9c1b51eSKate Stone           for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
412844d93782SGreg Clayton             BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
412944d93782SGreg Clayton             const size_t num_bps_locs = bp_sp->GetNumLocations();
4130b9c1b51eSKate Stone             for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs;
4131b9c1b51eSKate Stone                  ++bp_loc_idx) {
4132b9c1b51eSKate Stone               BreakpointLocationSP bp_loc_sp =
4133b9c1b51eSKate Stone                   bp_sp->GetLocationAtIndex(bp_loc_idx);
413444d93782SGreg Clayton               LineEntry bp_loc_line_entry;
4135b9c1b51eSKate Stone               const lldb::addr_t file_addr =
4136b9c1b51eSKate Stone                   bp_loc_sp->GetAddress().GetFileAddress();
4137b9c1b51eSKate Stone               if (file_addr != LLDB_INVALID_ADDRESS) {
413844d93782SGreg Clayton                 if (m_disassembly_range.ContainsFileAddress(file_addr))
413944d93782SGreg Clayton                   bp_file_addrs.insert(file_addr);
414044d93782SGreg Clayton               }
414144d93782SGreg Clayton             }
414244d93782SGreg Clayton           }
414344d93782SGreg Clayton         }
414444d93782SGreg Clayton 
414544d93782SGreg Clayton         const attr_t selected_highlight_attr = A_REVERSE;
414644d93782SGreg Clayton         const attr_t pc_highlight_attr = COLOR_PAIR(1);
414744d93782SGreg Clayton 
414844d93782SGreg Clayton         StreamString strm;
414944d93782SGreg Clayton 
415044d93782SGreg Clayton         InstructionList &insts = m_disassembly_sp->GetInstructionList();
415144d93782SGreg Clayton         Address pc_address;
415244d93782SGreg Clayton 
415344d93782SGreg Clayton         if (frame_sp)
415444d93782SGreg Clayton           pc_address = frame_sp->GetFrameCodeAddress();
4155b9c1b51eSKate Stone         const uint32_t pc_idx =
4156b9c1b51eSKate Stone             pc_address.IsValid()
4157b9c1b51eSKate Stone                 ? insts.GetIndexOfInstructionAtAddress(pc_address)
4158b9c1b51eSKate Stone                 : UINT32_MAX;
4159b9c1b51eSKate Stone         if (set_selected_line_to_pc) {
416044d93782SGreg Clayton           m_selected_line = pc_idx;
416144d93782SGreg Clayton         }
416244d93782SGreg Clayton 
416344d93782SGreg Clayton         const uint32_t non_visible_pc_offset = (num_visible_lines / 5);
41643985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_disassembly_lines)
416544d93782SGreg Clayton           m_first_visible_line = 0;
416644d93782SGreg Clayton 
4167b9c1b51eSKate Stone         if (pc_idx < num_disassembly_lines) {
41683985c8c6SSaleem Abdulrasool           if (pc_idx < static_cast<uint32_t>(m_first_visible_line) ||
416944d93782SGreg Clayton               pc_idx >= m_first_visible_line + num_visible_lines)
417044d93782SGreg Clayton             m_first_visible_line = pc_idx - non_visible_pc_offset;
417144d93782SGreg Clayton         }
417244d93782SGreg Clayton 
4173b9c1b51eSKate Stone         for (size_t i = 0; i < num_visible_lines; ++i) {
417444d93782SGreg Clayton           const uint32_t inst_idx = m_first_visible_line + i;
417544d93782SGreg Clayton           Instruction *inst = insts.GetInstructionAtIndex(inst_idx).get();
417644d93782SGreg Clayton           if (!inst)
417744d93782SGreg Clayton             break;
417844d93782SGreg Clayton 
4179ec990867SGreg Clayton           const int line_y = m_min_y + i;
4180ec990867SGreg Clayton           window.MoveCursor(1, line_y);
418144d93782SGreg Clayton           const bool is_pc_line = frame_sp && inst_idx == pc_idx;
418244d93782SGreg Clayton           const bool line_is_selected = m_selected_line == inst_idx;
418344d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
418444d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
418544d93782SGreg Clayton           attr_t highlight_attr = 0;
418644d93782SGreg Clayton           attr_t bp_attr = 0;
418744d93782SGreg Clayton           if (is_pc_line)
418844d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
418944d93782SGreg Clayton           else if (line_is_selected)
419044d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
419144d93782SGreg Clayton 
4192b9c1b51eSKate Stone           if (bp_file_addrs.find(inst->GetAddress().GetFileAddress()) !=
4193b9c1b51eSKate Stone               bp_file_addrs.end())
419444d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
419544d93782SGreg Clayton 
419644d93782SGreg Clayton           if (bp_attr)
419744d93782SGreg Clayton             window.AttributeOn(bp_attr);
419844d93782SGreg Clayton 
4199324a1036SSaleem Abdulrasool           window.Printf(" 0x%16.16llx ",
4200b9c1b51eSKate Stone                         static_cast<unsigned long long>(
4201b9c1b51eSKate Stone                             inst->GetAddress().GetLoadAddress(target)));
420244d93782SGreg Clayton 
420344d93782SGreg Clayton           if (bp_attr)
420444d93782SGreg Clayton             window.AttributeOff(bp_attr);
420544d93782SGreg Clayton 
420644d93782SGreg Clayton           window.PutChar(ACS_VLINE);
420744d93782SGreg Clayton           // Mark the line with the PC with a diamond
420844d93782SGreg Clayton           if (is_pc_line)
420944d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
421044d93782SGreg Clayton           else
421144d93782SGreg Clayton             window.PutChar(' ');
421244d93782SGreg Clayton 
421344d93782SGreg Clayton           if (highlight_attr)
421444d93782SGreg Clayton             window.AttributeOn(highlight_attr);
421544d93782SGreg Clayton 
421644d93782SGreg Clayton           const char *mnemonic = inst->GetMnemonic(&exe_ctx);
421744d93782SGreg Clayton           const char *operands = inst->GetOperands(&exe_ctx);
421844d93782SGreg Clayton           const char *comment = inst->GetComment(&exe_ctx);
421944d93782SGreg Clayton 
4220c5dac77aSEugene Zelenko           if (mnemonic != nullptr && mnemonic[0] == '\0')
4221c5dac77aSEugene Zelenko             mnemonic = nullptr;
4222c5dac77aSEugene Zelenko           if (operands != nullptr && operands[0] == '\0')
4223c5dac77aSEugene Zelenko             operands = nullptr;
4224c5dac77aSEugene Zelenko           if (comment != nullptr && comment[0] == '\0')
4225c5dac77aSEugene Zelenko             comment = nullptr;
422644d93782SGreg Clayton 
422744d93782SGreg Clayton           strm.Clear();
422844d93782SGreg Clayton 
4229c5dac77aSEugene Zelenko           if (mnemonic != nullptr && operands != nullptr && comment != nullptr)
423044d93782SGreg Clayton             strm.Printf("%-8s %-25s ; %s", mnemonic, operands, comment);
4231c5dac77aSEugene Zelenko           else if (mnemonic != nullptr && operands != nullptr)
423244d93782SGreg Clayton             strm.Printf("%-8s %s", mnemonic, operands);
4233c5dac77aSEugene Zelenko           else if (mnemonic != nullptr)
423444d93782SGreg Clayton             strm.Printf("%s", mnemonic);
423544d93782SGreg Clayton 
423644d93782SGreg Clayton           int right_pad = 1;
4237c156427dSZachary Turner           window.PutCStringTruncated(strm.GetData(), right_pad);
423844d93782SGreg Clayton 
4239b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4240b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
424144d93782SGreg Clayton             StopInfoSP stop_info_sp;
424244d93782SGreg Clayton             if (thread)
424344d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4244b9c1b51eSKate Stone             if (stop_info_sp) {
424544d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4246b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
424744d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4248ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
424944d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4250b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4251b9c1b51eSKate Stone                 // line_y);
4252b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4253b9c1b51eSKate Stone                               stop_description);
425444d93782SGreg Clayton               }
4255b9c1b51eSKate Stone             } else {
4256ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
425744d93782SGreg Clayton             }
425844d93782SGreg Clayton           }
425944d93782SGreg Clayton           if (highlight_attr)
426044d93782SGreg Clayton             window.AttributeOff(highlight_attr);
426144d93782SGreg Clayton         }
426244d93782SGreg Clayton       }
426344d93782SGreg Clayton     }
426444d93782SGreg Clayton     window.DeferredRefresh();
426544d93782SGreg Clayton     return true; // Drawing handled
426644d93782SGreg Clayton   }
426744d93782SGreg Clayton 
4268b9c1b51eSKate Stone   size_t GetNumLines() {
426944d93782SGreg Clayton     size_t num_lines = GetNumSourceLines();
427044d93782SGreg Clayton     if (num_lines == 0)
427144d93782SGreg Clayton       num_lines = GetNumDisassemblyLines();
427244d93782SGreg Clayton     return num_lines;
427344d93782SGreg Clayton   }
427444d93782SGreg Clayton 
4275b9c1b51eSKate Stone   size_t GetNumSourceLines() const {
427644d93782SGreg Clayton     if (m_file_sp)
427744d93782SGreg Clayton       return m_file_sp->GetNumLines();
427844d93782SGreg Clayton     return 0;
427944d93782SGreg Clayton   }
4280315b6884SEugene Zelenko 
4281b9c1b51eSKate Stone   size_t GetNumDisassemblyLines() const {
428244d93782SGreg Clayton     if (m_disassembly_sp)
428344d93782SGreg Clayton       return m_disassembly_sp->GetInstructionList().GetSize();
428444d93782SGreg Clayton     return 0;
428544d93782SGreg Clayton   }
428644d93782SGreg Clayton 
4287b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
428844d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
428944d93782SGreg Clayton     const size_t num_lines = GetNumLines();
429044d93782SGreg Clayton 
4291b9c1b51eSKate Stone     switch (c) {
429244d93782SGreg Clayton     case ',':
429344d93782SGreg Clayton     case KEY_PPAGE:
429444d93782SGreg Clayton       // Page up key
42953985c8c6SSaleem Abdulrasool       if (static_cast<uint32_t>(m_first_visible_line) > num_visible_lines)
429644d93782SGreg Clayton         m_first_visible_line -= num_visible_lines;
429744d93782SGreg Clayton       else
429844d93782SGreg Clayton         m_first_visible_line = 0;
429944d93782SGreg Clayton       m_selected_line = m_first_visible_line;
430044d93782SGreg Clayton       return eKeyHandled;
430144d93782SGreg Clayton 
430244d93782SGreg Clayton     case '.':
430344d93782SGreg Clayton     case KEY_NPAGE:
430444d93782SGreg Clayton       // Page down key
430544d93782SGreg Clayton       {
430644d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < num_lines)
430744d93782SGreg Clayton           m_first_visible_line += num_visible_lines;
430844d93782SGreg Clayton         else if (num_lines < num_visible_lines)
430944d93782SGreg Clayton           m_first_visible_line = 0;
431044d93782SGreg Clayton         else
431144d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
431244d93782SGreg Clayton         m_selected_line = m_first_visible_line;
431344d93782SGreg Clayton       }
431444d93782SGreg Clayton       return eKeyHandled;
431544d93782SGreg Clayton 
431644d93782SGreg Clayton     case KEY_UP:
4317b9c1b51eSKate Stone       if (m_selected_line > 0) {
431844d93782SGreg Clayton         m_selected_line--;
43193985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > m_selected_line)
432044d93782SGreg Clayton           m_first_visible_line = m_selected_line;
432144d93782SGreg Clayton       }
432244d93782SGreg Clayton       return eKeyHandled;
432344d93782SGreg Clayton 
432444d93782SGreg Clayton     case KEY_DOWN:
4325b9c1b51eSKate Stone       if (m_selected_line + 1 < num_lines) {
432644d93782SGreg Clayton         m_selected_line++;
432744d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < m_selected_line)
432844d93782SGreg Clayton           m_first_visible_line++;
432944d93782SGreg Clayton       }
433044d93782SGreg Clayton       return eKeyHandled;
433144d93782SGreg Clayton 
433244d93782SGreg Clayton     case '\r':
433344d93782SGreg Clayton     case '\n':
433444d93782SGreg Clayton     case KEY_ENTER:
433544d93782SGreg Clayton       // Set a breakpoint and run to the line using a one shot breakpoint
4336b9c1b51eSKate Stone       if (GetNumSourceLines() > 0) {
4337b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4338b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4339b9c1b51eSKate Stone         if (exe_ctx.HasProcessScope() && exe_ctx.GetProcessRef().IsAlive()) {
4340b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4341b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
434244d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4343b9c1b51eSKate Stone               m_selected_line +
4344b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4345431b1584SAdrian Prantl               0,     // Unspecified column.
43462411167fSJim Ingham               0,     // No offset
434744d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
434844d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
434944d93782SGreg Clayton               false,               // internal
4350055ad9beSIlia K               false,               // request_hardware
4351055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
435244d93782SGreg Clayton           // Make breakpoint one shot
435344d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
435444d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
435544d93782SGreg Clayton         }
4356b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4357b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4358b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4359b9c1b51eSKate Stone                                       .get();
4360b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4361b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4362b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
436344d93782SGreg Clayton           Address addr = inst->GetAddress();
4364b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4365b9c1b51eSKate Stone               addr,   // lldb_private::Address
436644d93782SGreg Clayton               false,  // internal
436744d93782SGreg Clayton               false); // request_hardware
436844d93782SGreg Clayton           // Make breakpoint one shot
436944d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
437044d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
437144d93782SGreg Clayton         }
437244d93782SGreg Clayton       }
437344d93782SGreg Clayton       return eKeyHandled;
437444d93782SGreg Clayton 
437544d93782SGreg Clayton     case 'b': // 'b' == toggle breakpoint on currently selected line
4376b9c1b51eSKate Stone       if (m_selected_line < GetNumSourceLines()) {
4377b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4378b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4379b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
4380b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4381b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
438244d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4383b9c1b51eSKate Stone               m_selected_line +
4384b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4385431b1584SAdrian Prantl               0,     // No column specified.
43862411167fSJim Ingham               0,     // No offset
438744d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
438844d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
438944d93782SGreg Clayton               false,               // internal
4390055ad9beSIlia K               false,               // request_hardware
4391055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
439244d93782SGreg Clayton         }
4393b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4394b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4395b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4396b9c1b51eSKate Stone                                       .get();
4397b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4398b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4399b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
440044d93782SGreg Clayton           Address addr = inst->GetAddress();
4401b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4402b9c1b51eSKate Stone               addr,   // lldb_private::Address
440344d93782SGreg Clayton               false,  // internal
440444d93782SGreg Clayton               false); // request_hardware
440544d93782SGreg Clayton         }
440644d93782SGreg Clayton       }
440744d93782SGreg Clayton       return eKeyHandled;
440844d93782SGreg Clayton 
440944d93782SGreg Clayton     case 'd': // 'd' == detach and let run
441044d93782SGreg Clayton     case 'D': // 'D' == detach and keep stopped
441144d93782SGreg Clayton     {
4412b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4413b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
441444d93782SGreg Clayton       if (exe_ctx.HasProcessScope())
441544d93782SGreg Clayton         exe_ctx.GetProcessRef().Detach(c == 'D');
441644d93782SGreg Clayton     }
441744d93782SGreg Clayton       return eKeyHandled;
441844d93782SGreg Clayton 
441944d93782SGreg Clayton     case 'k':
442044d93782SGreg Clayton       // 'k' == kill
442144d93782SGreg Clayton       {
4422b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4423b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
442444d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
4425ede3193bSJason Molenda           exe_ctx.GetProcessRef().Destroy(false);
442644d93782SGreg Clayton       }
442744d93782SGreg Clayton       return eKeyHandled;
442844d93782SGreg Clayton 
442944d93782SGreg Clayton     case 'c':
443044d93782SGreg Clayton       // 'c' == continue
443144d93782SGreg Clayton       {
4432b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4433b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
443444d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
443544d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
443644d93782SGreg Clayton       }
443744d93782SGreg Clayton       return eKeyHandled;
443844d93782SGreg Clayton 
443944d93782SGreg Clayton     case 'o':
444044d93782SGreg Clayton       // 'o' == step out
444144d93782SGreg Clayton       {
4442b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4443b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4444b9c1b51eSKate Stone         if (exe_ctx.HasThreadScope() &&
4445b9c1b51eSKate Stone             StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
444644d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
444744d93782SGreg Clayton         }
444844d93782SGreg Clayton       }
444944d93782SGreg Clayton       return eKeyHandled;
4450315b6884SEugene Zelenko 
445144d93782SGreg Clayton     case 'n': // 'n' == step over
445244d93782SGreg Clayton     case 'N': // 'N' == step over instruction
445344d93782SGreg Clayton     {
4454b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4455b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
4456b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope() &&
4457b9c1b51eSKate Stone           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
445844d93782SGreg Clayton         bool source_step = (c == 'n');
445944d93782SGreg Clayton         exe_ctx.GetThreadRef().StepOver(source_step);
446044d93782SGreg Clayton       }
446144d93782SGreg Clayton     }
446244d93782SGreg Clayton       return eKeyHandled;
4463315b6884SEugene Zelenko 
446444d93782SGreg Clayton     case 's': // 's' == step into
446544d93782SGreg Clayton     case 'S': // 'S' == step into instruction
446644d93782SGreg Clayton     {
4467b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4468b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
4469b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope() &&
4470b9c1b51eSKate Stone           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
447144d93782SGreg Clayton         bool source_step = (c == 's');
44724b4b2478SJim Ingham         exe_ctx.GetThreadRef().StepIn(source_step);
447344d93782SGreg Clayton       }
447444d93782SGreg Clayton     }
447544d93782SGreg Clayton       return eKeyHandled;
447644d93782SGreg Clayton 
447744d93782SGreg Clayton     case 'h':
447844d93782SGreg Clayton       window.CreateHelpSubwindow();
447944d93782SGreg Clayton       return eKeyHandled;
448044d93782SGreg Clayton 
448144d93782SGreg Clayton     default:
448244d93782SGreg Clayton       break;
448344d93782SGreg Clayton     }
448444d93782SGreg Clayton     return eKeyNotHandled;
448544d93782SGreg Clayton   }
448644d93782SGreg Clayton 
448744d93782SGreg Clayton protected:
448844d93782SGreg Clayton   typedef std::set<uint32_t> BreakpointLines;
448944d93782SGreg Clayton   typedef std::set<lldb::addr_t> BreakpointAddrs;
449044d93782SGreg Clayton 
449144d93782SGreg Clayton   Debugger &m_debugger;
449244d93782SGreg Clayton   SymbolContext m_sc;
449344d93782SGreg Clayton   SourceManager::FileSP m_file_sp;
449444d93782SGreg Clayton   SymbolContextScope *m_disassembly_scope;
449544d93782SGreg Clayton   lldb::DisassemblerSP m_disassembly_sp;
449644d93782SGreg Clayton   AddressRange m_disassembly_range;
4497ec990867SGreg Clayton   StreamString m_title;
449844d93782SGreg Clayton   lldb::user_id_t m_tid;
449944d93782SGreg Clayton   int m_line_width;
450044d93782SGreg Clayton   uint32_t m_selected_line; // The selected line
450144d93782SGreg Clayton   uint32_t m_pc_line;       // The line with the PC
450244d93782SGreg Clayton   uint32_t m_stop_id;
450344d93782SGreg Clayton   uint32_t m_frame_idx;
450444d93782SGreg Clayton   int m_first_visible_line;
450544d93782SGreg Clayton   int m_min_x;
450644d93782SGreg Clayton   int m_min_y;
450744d93782SGreg Clayton   int m_max_x;
450844d93782SGreg Clayton   int m_max_y;
450944d93782SGreg Clayton };
451044d93782SGreg Clayton 
451144d93782SGreg Clayton DisplayOptions ValueObjectListDelegate::g_options = {true};
451244d93782SGreg Clayton 
4513b9c1b51eSKate Stone IOHandlerCursesGUI::IOHandlerCursesGUI(Debugger &debugger)
4514b9c1b51eSKate Stone     : IOHandler(debugger, IOHandler::Type::Curses) {}
451544d93782SGreg Clayton 
4516b9c1b51eSKate Stone void IOHandlerCursesGUI::Activate() {
451744d93782SGreg Clayton   IOHandler::Activate();
4518b9c1b51eSKate Stone   if (!m_app_ap) {
451944d93782SGreg Clayton     m_app_ap.reset(new Application(GetInputFILE(), GetOutputFILE()));
452044d93782SGreg Clayton 
452144d93782SGreg Clayton     // This is both a window and a menu delegate
4522b9c1b51eSKate Stone     std::shared_ptr<ApplicationDelegate> app_delegate_sp(
4523b9c1b51eSKate Stone         new ApplicationDelegate(*m_app_ap, m_debugger));
452444d93782SGreg Clayton 
4525b9c1b51eSKate Stone     MenuDelegateSP app_menu_delegate_sp =
4526b9c1b51eSKate Stone         std::static_pointer_cast<MenuDelegate>(app_delegate_sp);
4527b9c1b51eSKate Stone     MenuSP lldb_menu_sp(
4528b9c1b51eSKate Stone         new Menu("LLDB", "F1", KEY_F(1), ApplicationDelegate::eMenuID_LLDB));
4529b9c1b51eSKate Stone     MenuSP exit_menuitem_sp(
4530b9c1b51eSKate Stone         new Menu("Exit", nullptr, 'x', ApplicationDelegate::eMenuID_LLDBExit));
453144d93782SGreg Clayton     exit_menuitem_sp->SetCannedResult(MenuActionResult::Quit);
4532b9c1b51eSKate Stone     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(
4533b9c1b51eSKate Stone         "About LLDB", nullptr, 'a', ApplicationDelegate::eMenuID_LLDBAbout)));
453444d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
453544d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(exit_menuitem_sp);
453644d93782SGreg Clayton 
4537b9c1b51eSKate Stone     MenuSP target_menu_sp(new Menu("Target", "F2", KEY_F(2),
4538b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Target));
4539b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4540b9c1b51eSKate Stone         "Create", nullptr, 'c', ApplicationDelegate::eMenuID_TargetCreate)));
4541b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4542b9c1b51eSKate Stone         "Delete", nullptr, 'd', ApplicationDelegate::eMenuID_TargetDelete)));
454344d93782SGreg Clayton 
4544b9c1b51eSKate Stone     MenuSP process_menu_sp(new Menu("Process", "F3", KEY_F(3),
4545b9c1b51eSKate Stone                                     ApplicationDelegate::eMenuID_Process));
4546b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4547b9c1b51eSKate Stone         "Attach", nullptr, 'a', ApplicationDelegate::eMenuID_ProcessAttach)));
4548b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4549b9c1b51eSKate Stone         "Detach", nullptr, 'd', ApplicationDelegate::eMenuID_ProcessDetach)));
4550b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4551b9c1b51eSKate Stone         "Launch", nullptr, 'l', ApplicationDelegate::eMenuID_ProcessLaunch)));
455244d93782SGreg Clayton     process_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
4553b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(
4554b9c1b51eSKate Stone         MenuSP(new Menu("Continue", nullptr, 'c',
4555b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ProcessContinue)));
4556b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4557b9c1b51eSKate Stone         "Halt", nullptr, 'h', ApplicationDelegate::eMenuID_ProcessHalt)));
4558b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4559b9c1b51eSKate Stone         "Kill", nullptr, 'k', ApplicationDelegate::eMenuID_ProcessKill)));
456044d93782SGreg Clayton 
4561b9c1b51eSKate Stone     MenuSP thread_menu_sp(new Menu("Thread", "F4", KEY_F(4),
4562b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Thread));
4563b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4564b9c1b51eSKate Stone         "Step In", nullptr, 'i', ApplicationDelegate::eMenuID_ThreadStepIn)));
4565b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(
4566b9c1b51eSKate Stone         MenuSP(new Menu("Step Over", nullptr, 'v',
4567b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ThreadStepOver)));
4568b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4569b9c1b51eSKate Stone         "Step Out", nullptr, 'o', ApplicationDelegate::eMenuID_ThreadStepOut)));
457044d93782SGreg Clayton 
4571b9c1b51eSKate Stone     MenuSP view_menu_sp(
4572b9c1b51eSKate Stone         new Menu("View", "F5", KEY_F(5), ApplicationDelegate::eMenuID_View));
4573b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4574b9c1b51eSKate Stone         MenuSP(new Menu("Backtrace", nullptr, 'b',
4575b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewBacktrace)));
4576b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4577b9c1b51eSKate Stone         MenuSP(new Menu("Registers", nullptr, 'r',
4578b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewRegisters)));
4579b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(MenuSP(new Menu(
4580b9c1b51eSKate Stone         "Source", nullptr, 's', ApplicationDelegate::eMenuID_ViewSource)));
4581b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4582b9c1b51eSKate Stone         MenuSP(new Menu("Variables", nullptr, 'v',
4583b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewVariables)));
458444d93782SGreg Clayton 
4585b9c1b51eSKate Stone     MenuSP help_menu_sp(
4586b9c1b51eSKate Stone         new Menu("Help", "F6", KEY_F(6), ApplicationDelegate::eMenuID_Help));
4587b9c1b51eSKate Stone     help_menu_sp->AddSubmenu(MenuSP(new Menu(
4588b9c1b51eSKate Stone         "GUI Help", nullptr, 'g', ApplicationDelegate::eMenuID_HelpGUIHelp)));
458944d93782SGreg Clayton 
459044d93782SGreg Clayton     m_app_ap->Initialize();
459144d93782SGreg Clayton     WindowSP &main_window_sp = m_app_ap->GetMainWindow();
459244d93782SGreg Clayton 
459344d93782SGreg Clayton     MenuSP menubar_sp(new Menu(Menu::Type::Bar));
459444d93782SGreg Clayton     menubar_sp->AddSubmenu(lldb_menu_sp);
459544d93782SGreg Clayton     menubar_sp->AddSubmenu(target_menu_sp);
459644d93782SGreg Clayton     menubar_sp->AddSubmenu(process_menu_sp);
459744d93782SGreg Clayton     menubar_sp->AddSubmenu(thread_menu_sp);
459844d93782SGreg Clayton     menubar_sp->AddSubmenu(view_menu_sp);
459944d93782SGreg Clayton     menubar_sp->AddSubmenu(help_menu_sp);
460044d93782SGreg Clayton     menubar_sp->SetDelegate(app_menu_delegate_sp);
460144d93782SGreg Clayton 
460244d93782SGreg Clayton     Rect content_bounds = main_window_sp->GetFrame();
460344d93782SGreg Clayton     Rect menubar_bounds = content_bounds.MakeMenuBar();
460444d93782SGreg Clayton     Rect status_bounds = content_bounds.MakeStatusBar();
460544d93782SGreg Clayton     Rect source_bounds;
460644d93782SGreg Clayton     Rect variables_bounds;
460744d93782SGreg Clayton     Rect threads_bounds;
460844d93782SGreg Clayton     Rect source_variables_bounds;
4609b9c1b51eSKate Stone     content_bounds.VerticalSplitPercentage(0.80, source_variables_bounds,
4610b9c1b51eSKate Stone                                            threads_bounds);
4611b9c1b51eSKate Stone     source_variables_bounds.HorizontalSplitPercentage(0.70, source_bounds,
4612b9c1b51eSKate Stone                                                       variables_bounds);
461344d93782SGreg Clayton 
4614b9c1b51eSKate Stone     WindowSP menubar_window_sp =
4615b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Menubar", menubar_bounds, false);
461605097246SAdrian Prantl     // Let the menubar get keys if the active window doesn't handle the keys
461705097246SAdrian Prantl     // that are typed so it can respond to menubar key presses.
4618b9c1b51eSKate Stone     menubar_window_sp->SetCanBeActive(
4619b9c1b51eSKate Stone         false); // Don't let the menubar become the active window
462044d93782SGreg Clayton     menubar_window_sp->SetDelegate(menubar_sp);
462144d93782SGreg Clayton 
4622b9c1b51eSKate Stone     WindowSP source_window_sp(
4623b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Source", source_bounds, true));
4624b9c1b51eSKate Stone     WindowSP variables_window_sp(
4625b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Variables", variables_bounds, false));
4626b9c1b51eSKate Stone     WindowSP threads_window_sp(
4627b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Threads", threads_bounds, false));
4628b9c1b51eSKate Stone     WindowSP status_window_sp(
46296bb7e21fSPavel Labath         main_window_sp->CreateSubWindow("Status", status_bounds, false));
4630b9c1b51eSKate Stone     status_window_sp->SetCanBeActive(
4631b9c1b51eSKate Stone         false); // Don't let the status bar become the active window
4632b9c1b51eSKate Stone     main_window_sp->SetDelegate(
4633b9c1b51eSKate Stone         std::static_pointer_cast<WindowDelegate>(app_delegate_sp));
4634b9c1b51eSKate Stone     source_window_sp->SetDelegate(
4635b9c1b51eSKate Stone         WindowDelegateSP(new SourceFileWindowDelegate(m_debugger)));
4636b9c1b51eSKate Stone     variables_window_sp->SetDelegate(
4637b9c1b51eSKate Stone         WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
4638ec990867SGreg Clayton     TreeDelegateSP thread_delegate_sp(new ThreadsTreeDelegate(m_debugger));
4639b9c1b51eSKate Stone     threads_window_sp->SetDelegate(WindowDelegateSP(
4640b9c1b51eSKate Stone         new TreeWindowDelegate(m_debugger, thread_delegate_sp)));
4641b9c1b51eSKate Stone     status_window_sp->SetDelegate(
4642b9c1b51eSKate Stone         WindowDelegateSP(new StatusBarWindowDelegate(m_debugger)));
464344d93782SGreg Clayton 
46445fdb09bbSGreg Clayton     // Show the main help window once the first time the curses GUI is launched
46455fdb09bbSGreg Clayton     static bool g_showed_help = false;
4646b9c1b51eSKate Stone     if (!g_showed_help) {
46475fdb09bbSGreg Clayton       g_showed_help = true;
46485fdb09bbSGreg Clayton       main_window_sp->CreateHelpSubwindow();
46495fdb09bbSGreg Clayton     }
46505fdb09bbSGreg Clayton 
465144d93782SGreg Clayton     init_pair(1, COLOR_WHITE, COLOR_BLUE);
465244d93782SGreg Clayton     init_pair(2, COLOR_BLACK, COLOR_WHITE);
465344d93782SGreg Clayton     init_pair(3, COLOR_MAGENTA, COLOR_WHITE);
465444d93782SGreg Clayton     init_pair(4, COLOR_MAGENTA, COLOR_BLACK);
465544d93782SGreg Clayton     init_pair(5, COLOR_RED, COLOR_BLACK);
465644d93782SGreg Clayton   }
465744d93782SGreg Clayton }
465844d93782SGreg Clayton 
4659b9c1b51eSKate Stone void IOHandlerCursesGUI::Deactivate() { m_app_ap->Terminate(); }
466044d93782SGreg Clayton 
4661b9c1b51eSKate Stone void IOHandlerCursesGUI::Run() {
466244d93782SGreg Clayton   m_app_ap->Run(m_debugger);
466344d93782SGreg Clayton   SetIsDone(true);
466444d93782SGreg Clayton }
466544d93782SGreg Clayton 
4666315b6884SEugene Zelenko IOHandlerCursesGUI::~IOHandlerCursesGUI() = default;
466744d93782SGreg Clayton 
4668b9c1b51eSKate Stone void IOHandlerCursesGUI::Cancel() {}
466944d93782SGreg Clayton 
4670b9c1b51eSKate Stone bool IOHandlerCursesGUI::Interrupt() { return false; }
467144d93782SGreg Clayton 
4672b9c1b51eSKate Stone void IOHandlerCursesGUI::GotEOF() {}
467344d93782SGreg Clayton 
4674315b6884SEugene Zelenko #endif // LLDB_DISABLE_CURSES
4675