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
79340b0309SGreg Clayton                 0)              // Flags
80b9c1b51eSKate Stone {}
8144d93782SGreg Clayton 
82b9c1b51eSKate Stone IOHandler::IOHandler(Debugger &debugger, IOHandler::Type type,
8344d93782SGreg Clayton                      const lldb::StreamFileSP &input_sp,
8444d93782SGreg Clayton                      const lldb::StreamFileSP &output_sp,
85b9c1b51eSKate Stone                      const lldb::StreamFileSP &error_sp, uint32_t flags)
86b9c1b51eSKate Stone     : m_debugger(debugger), m_input_sp(input_sp), m_output_sp(output_sp),
87b9c1b51eSKate Stone       m_error_sp(error_sp), m_popped(false), m_flags(flags), m_type(type),
88b9c1b51eSKate Stone       m_user_data(nullptr), m_done(false), m_active(false) {
8944d93782SGreg Clayton   // If any files are not specified, then adopt them from the top input reader.
9044d93782SGreg Clayton   if (!m_input_sp || !m_output_sp || !m_error_sp)
91b9c1b51eSKate Stone     debugger.AdoptTopIOHandlerFilesIfInvalid(m_input_sp, m_output_sp,
9244d93782SGreg Clayton                                              m_error_sp);
9344d93782SGreg Clayton }
9444d93782SGreg Clayton 
95315b6884SEugene Zelenko IOHandler::~IOHandler() = default;
9644d93782SGreg Clayton 
97b9c1b51eSKate Stone int IOHandler::GetInputFD() {
98c5dac77aSEugene Zelenko   return (m_input_sp ? m_input_sp->GetFile().GetDescriptor() : -1);
9944d93782SGreg Clayton }
10044d93782SGreg Clayton 
101b9c1b51eSKate Stone int IOHandler::GetOutputFD() {
102c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetDescriptor() : -1);
10344d93782SGreg Clayton }
10444d93782SGreg Clayton 
105b9c1b51eSKate Stone int IOHandler::GetErrorFD() {
106c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetDescriptor() : -1);
10744d93782SGreg Clayton }
10844d93782SGreg Clayton 
109b9c1b51eSKate Stone FILE *IOHandler::GetInputFILE() {
110c5dac77aSEugene Zelenko   return (m_input_sp ? m_input_sp->GetFile().GetStream() : nullptr);
11144d93782SGreg Clayton }
11244d93782SGreg Clayton 
113b9c1b51eSKate Stone FILE *IOHandler::GetOutputFILE() {
114c5dac77aSEugene Zelenko   return (m_output_sp ? m_output_sp->GetFile().GetStream() : nullptr);
11544d93782SGreg Clayton }
11644d93782SGreg Clayton 
117b9c1b51eSKate Stone FILE *IOHandler::GetErrorFILE() {
118c5dac77aSEugene Zelenko   return (m_error_sp ? m_error_sp->GetFile().GetStream() : nullptr);
11944d93782SGreg Clayton }
12044d93782SGreg Clayton 
121b9c1b51eSKate Stone StreamFileSP &IOHandler::GetInputStreamFile() { return m_input_sp; }
12244d93782SGreg Clayton 
123b9c1b51eSKate Stone StreamFileSP &IOHandler::GetOutputStreamFile() { return m_output_sp; }
12444d93782SGreg Clayton 
125b9c1b51eSKate Stone StreamFileSP &IOHandler::GetErrorStreamFile() { return m_error_sp; }
12644d93782SGreg Clayton 
127b9c1b51eSKate Stone bool IOHandler::GetIsInteractive() {
128340b0309SGreg Clayton   return GetInputStreamFile()->GetFile().GetIsInteractive();
129340b0309SGreg Clayton }
130340b0309SGreg Clayton 
131b9c1b51eSKate Stone bool IOHandler::GetIsRealTerminal() {
132340b0309SGreg Clayton   return GetInputStreamFile()->GetFile().GetIsRealTerminal();
133340b0309SGreg Clayton }
13444d93782SGreg Clayton 
135b9c1b51eSKate Stone void IOHandler::SetPopped(bool b) { m_popped.SetValue(b, eBroadcastOnChange); }
136e30f11d9SKate Stone 
137b9c1b51eSKate Stone void IOHandler::WaitForPop() { m_popped.WaitForValueEqualTo(true); }
138e30f11d9SKate Stone 
139b9c1b51eSKate Stone void IOHandlerStack::PrintAsync(Stream *stream, const char *s, size_t len) {
140b9c1b51eSKate Stone   if (stream) {
14116ff8604SSaleem Abdulrasool     std::lock_guard<std::recursive_mutex> guard(m_mutex);
1424446487dSPavel Labath     if (m_top)
1434446487dSPavel Labath       m_top->PrintAsync(stream, s, len);
1444446487dSPavel Labath   }
1454446487dSPavel Labath }
1464446487dSPavel Labath 
1477a120c8bSZachary Turner IOHandlerConfirm::IOHandlerConfirm(Debugger &debugger, llvm::StringRef prompt,
148b9c1b51eSKate Stone                                    bool default_response)
149b9c1b51eSKate Stone     : IOHandlerEditline(
150b9c1b51eSKate Stone           debugger, IOHandler::Type::Confirm,
151c5dac77aSEugene Zelenko           nullptr, // nullptr editline_name means no history loaded/saved
152514d8cd8SZachary Turner           llvm::StringRef(), // No prompt
153514d8cd8SZachary Turner           llvm::StringRef(), // No continuation prompt
15444d93782SGreg Clayton           false,             // Multi-line
155e30f11d9SKate Stone           false, // Don't colorize the prompt (i.e. the confirm message.)
156b9c1b51eSKate Stone           0, *this),
157b9c1b51eSKate Stone       m_default_response(default_response), m_user_response(default_response) {
15844d93782SGreg Clayton   StreamString prompt_stream;
15944d93782SGreg Clayton   prompt_stream.PutCString(prompt);
16044d93782SGreg Clayton   if (m_default_response)
16144d93782SGreg Clayton     prompt_stream.Printf(": [Y/n] ");
16244d93782SGreg Clayton   else
16344d93782SGreg Clayton     prompt_stream.Printf(": [y/N] ");
16444d93782SGreg Clayton 
165514d8cd8SZachary Turner   SetPrompt(prompt_stream.GetString());
16644d93782SGreg Clayton }
16744d93782SGreg Clayton 
168315b6884SEugene Zelenko IOHandlerConfirm::~IOHandlerConfirm() = default;
16944d93782SGreg Clayton 
1707f88829cSRaphael Isemann int IOHandlerConfirm::IOHandlerComplete(
1717f88829cSRaphael Isemann     IOHandler &io_handler, const char *current_line, const char *cursor,
1727f88829cSRaphael Isemann     const char *last_char, int skip_first_n_matches, int max_matches,
1737f88829cSRaphael Isemann     StringList &matches, StringList &descriptions) {
174b9c1b51eSKate Stone   if (current_line == cursor) {
175b9c1b51eSKate Stone     if (m_default_response) {
17644d93782SGreg Clayton       matches.AppendString("y");
177b9c1b51eSKate Stone     } else {
17844d93782SGreg Clayton       matches.AppendString("n");
17944d93782SGreg Clayton     }
18044d93782SGreg Clayton   }
18144d93782SGreg Clayton   return matches.GetSize();
18244d93782SGreg Clayton }
18344d93782SGreg Clayton 
184b9c1b51eSKate Stone void IOHandlerConfirm::IOHandlerInputComplete(IOHandler &io_handler,
185b9c1b51eSKate Stone                                               std::string &line) {
186b9c1b51eSKate Stone   if (line.empty()) {
18744d93782SGreg Clayton     // User just hit enter, set the response to the default
18844d93782SGreg Clayton     m_user_response = m_default_response;
18944d93782SGreg Clayton     io_handler.SetIsDone(true);
19044d93782SGreg Clayton     return;
19144d93782SGreg Clayton   }
19244d93782SGreg Clayton 
193b9c1b51eSKate Stone   if (line.size() == 1) {
194b9c1b51eSKate Stone     switch (line[0]) {
19544d93782SGreg Clayton     case 'y':
19644d93782SGreg Clayton     case 'Y':
19744d93782SGreg Clayton       m_user_response = true;
19844d93782SGreg Clayton       io_handler.SetIsDone(true);
19944d93782SGreg Clayton       return;
20044d93782SGreg Clayton     case 'n':
20144d93782SGreg Clayton     case 'N':
20244d93782SGreg Clayton       m_user_response = false;
20344d93782SGreg Clayton       io_handler.SetIsDone(true);
20444d93782SGreg Clayton       return;
20544d93782SGreg Clayton     default:
20644d93782SGreg Clayton       break;
20744d93782SGreg Clayton     }
20844d93782SGreg Clayton   }
20944d93782SGreg Clayton 
210b9c1b51eSKate Stone   if (line == "yes" || line == "YES" || line == "Yes") {
21144d93782SGreg Clayton     m_user_response = true;
21244d93782SGreg Clayton     io_handler.SetIsDone(true);
213b9c1b51eSKate Stone   } else if (line == "no" || line == "NO" || line == "No") {
21444d93782SGreg Clayton     m_user_response = false;
21544d93782SGreg Clayton     io_handler.SetIsDone(true);
21644d93782SGreg Clayton   }
21744d93782SGreg Clayton }
21844d93782SGreg Clayton 
2197f88829cSRaphael Isemann int IOHandlerDelegate::IOHandlerComplete(
2207f88829cSRaphael Isemann     IOHandler &io_handler, const char *current_line, const char *cursor,
2217f88829cSRaphael Isemann     const char *last_char, int skip_first_n_matches, int max_matches,
2227f88829cSRaphael Isemann     StringList &matches, StringList &descriptions) {
223b9c1b51eSKate Stone   switch (m_completion) {
22444d93782SGreg Clayton   case Completion::None:
22544d93782SGreg Clayton     break;
22644d93782SGreg Clayton 
22744d93782SGreg Clayton   case Completion::LLDBCommand:
228b9c1b51eSKate Stone     return io_handler.GetDebugger().GetCommandInterpreter().HandleCompletion(
229b9c1b51eSKate Stone         current_line, cursor, last_char, skip_first_n_matches, max_matches,
2307f88829cSRaphael Isemann         matches, descriptions);
231b9c1b51eSKate Stone   case Completion::Expression: {
2327f88829cSRaphael Isemann     CompletionResult result;
233a2e76c0bSRaphael Isemann     CompletionRequest request(current_line, current_line - cursor,
2347f88829cSRaphael Isemann                               skip_first_n_matches, max_matches, result);
235b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
236b9c1b51eSKate Stone         io_handler.GetDebugger().GetCommandInterpreter(),
237a2e76c0bSRaphael Isemann         CommandCompletions::eVariablePathCompletion, request, nullptr);
2387f88829cSRaphael Isemann     result.GetMatches(matches);
2397f88829cSRaphael Isemann     result.GetDescriptions(descriptions);
24044d93782SGreg Clayton 
2411a6d7ab5SRaphael Isemann     size_t num_matches = request.GetNumberOfMatches();
242b9c1b51eSKate Stone     if (num_matches > 0) {
24344d93782SGreg Clayton       std::string common_prefix;
2441a6d7ab5SRaphael Isemann       matches.LongestCommonPrefix(common_prefix);
245a2e76c0bSRaphael Isemann       const size_t partial_name_len = request.GetCursorArgumentPrefix().size();
24644d93782SGreg Clayton 
24705097246SAdrian Prantl       // If we matched a unique single command, add a space... Only do this if
24805097246SAdrian Prantl       // the completer told us this was a complete word, however...
249a2e76c0bSRaphael Isemann       if (num_matches == 1 && request.GetWordComplete()) {
25044d93782SGreg Clayton         common_prefix.push_back(' ');
25144d93782SGreg Clayton       }
25244d93782SGreg Clayton       common_prefix.erase(0, partial_name_len);
25344d93782SGreg Clayton       matches.InsertStringAtIndex(0, std::move(common_prefix));
25444d93782SGreg Clayton     }
25544d93782SGreg Clayton     return num_matches;
256b9c1b51eSKate Stone   } break;
25744d93782SGreg Clayton   }
25844d93782SGreg Clayton 
25944d93782SGreg Clayton   return 0;
26044d93782SGreg Clayton }
26144d93782SGreg Clayton 
262b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
263b9c1b51eSKate Stone     Debugger &debugger, IOHandler::Type type,
26444d93782SGreg Clayton     const char *editline_name, // Used for saving history files
265514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
266514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
267514d8cd8SZachary Turner     IOHandlerDelegate &delegate)
268b9c1b51eSKate Stone     : IOHandlerEditline(debugger, type,
26944d93782SGreg Clayton                         StreamFileSP(), // Inherit input from top input reader
27044d93782SGreg Clayton                         StreamFileSP(), // Inherit output from top input reader
27144d93782SGreg Clayton                         StreamFileSP(), // Inherit error from top input reader
272340b0309SGreg Clayton                         0,              // Flags
27344d93782SGreg Clayton                         editline_name,  // Used for saving history files
274b9c1b51eSKate Stone                         prompt, continuation_prompt, multi_line, color_prompts,
275b9c1b51eSKate Stone                         line_number_start, delegate) {}
27644d93782SGreg Clayton 
277b9c1b51eSKate Stone IOHandlerEditline::IOHandlerEditline(
278b9c1b51eSKate Stone     Debugger &debugger, IOHandler::Type type,
279b9c1b51eSKate Stone     const lldb::StreamFileSP &input_sp, const lldb::StreamFileSP &output_sp,
280b9c1b51eSKate Stone     const lldb::StreamFileSP &error_sp, uint32_t flags,
28144d93782SGreg Clayton     const char *editline_name, // Used for saving history files
282514d8cd8SZachary Turner     llvm::StringRef prompt, llvm::StringRef continuation_prompt,
283514d8cd8SZachary Turner     bool multi_line, bool color_prompts, uint32_t line_number_start,
284514d8cd8SZachary Turner     IOHandlerDelegate &delegate)
285b9c1b51eSKate Stone     : IOHandler(debugger, type, input_sp, output_sp, error_sp, flags),
286cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
28744d93782SGreg Clayton       m_editline_ap(),
288cacde7dfSTodd Fiala #endif
289b9c1b51eSKate Stone       m_delegate(delegate), m_prompt(), m_continuation_prompt(),
290b9c1b51eSKate Stone       m_current_lines_ptr(nullptr), m_base_line_number(line_number_start),
291b9c1b51eSKate Stone       m_curr_line_idx(UINT32_MAX), m_multi_line(multi_line),
292b9c1b51eSKate Stone       m_color_prompts(color_prompts), m_interrupt_exits(true),
293b9c1b51eSKate Stone       m_editing(false) {
29444d93782SGreg Clayton   SetPrompt(prompt);
29544d93782SGreg Clayton 
296cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
297914b8d98SDeepak Panickal   bool use_editline = false;
298340b0309SGreg Clayton 
299340b0309SGreg Clayton   use_editline = m_input_sp->GetFile().GetIsRealTerminal();
30044d93782SGreg Clayton 
301b9c1b51eSKate Stone   if (use_editline) {
302b9c1b51eSKate Stone     m_editline_ap.reset(new Editline(editline_name, GetInputFILE(),
303b9c1b51eSKate Stone                                      GetOutputFILE(), GetErrorFILE(),
304e30f11d9SKate Stone                                      m_color_prompts));
305e30f11d9SKate Stone     m_editline_ap->SetIsInputCompleteCallback(IsInputCompleteCallback, this);
30644d93782SGreg Clayton     m_editline_ap->SetAutoCompleteCallback(AutoCompleteCallback, this);
307e30f11d9SKate Stone     // See if the delegate supports fixing indentation
308e30f11d9SKate Stone     const char *indent_chars = delegate.IOHandlerGetFixIndentationCharacters();
309b9c1b51eSKate Stone     if (indent_chars) {
310b9c1b51eSKate Stone       // The delegate does support indentation, hook it up so when any
31105097246SAdrian Prantl       // indentation character is typed, the delegate gets a chance to fix it
312b9c1b51eSKate Stone       m_editline_ap->SetFixIndentationCallback(FixIndentationCallback, this,
313b9c1b51eSKate Stone                                                indent_chars);
314e30f11d9SKate Stone     }
31544d93782SGreg Clayton   }
316cacde7dfSTodd Fiala #endif
317e30f11d9SKate Stone   SetBaseLineNumber(m_base_line_number);
318514d8cd8SZachary Turner   SetPrompt(prompt);
319e30f11d9SKate Stone   SetContinuationPrompt(continuation_prompt);
32044d93782SGreg Clayton }
32144d93782SGreg Clayton 
322b9c1b51eSKate Stone IOHandlerEditline::~IOHandlerEditline() {
323cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
32444d93782SGreg Clayton   m_editline_ap.reset();
325cacde7dfSTodd Fiala #endif
32644d93782SGreg Clayton }
32744d93782SGreg Clayton 
328b9c1b51eSKate Stone void IOHandlerEditline::Activate() {
329e30f11d9SKate Stone   IOHandler::Activate();
330e30f11d9SKate Stone   m_delegate.IOHandlerActivated(*this);
331e30f11d9SKate Stone }
332e30f11d9SKate Stone 
333b9c1b51eSKate Stone void IOHandlerEditline::Deactivate() {
334e30f11d9SKate Stone   IOHandler::Deactivate();
335e30f11d9SKate Stone   m_delegate.IOHandlerDeactivated(*this);
336e30f11d9SKate Stone }
337e30f11d9SKate Stone 
338b9c1b51eSKate Stone bool IOHandlerEditline::GetLine(std::string &line, bool &interrupted) {
339cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
340b9c1b51eSKate Stone   if (m_editline_ap) {
341e30f11d9SKate Stone     return m_editline_ap->GetLine(line, interrupted);
342b9c1b51eSKate Stone   } else {
343cacde7dfSTodd Fiala #endif
34444d93782SGreg Clayton     line.clear();
34544d93782SGreg Clayton 
34644d93782SGreg Clayton     FILE *in = GetInputFILE();
347b9c1b51eSKate Stone     if (in) {
348b9c1b51eSKate Stone       if (GetIsInteractive()) {
349c5dac77aSEugene Zelenko         const char *prompt = nullptr;
350e30f11d9SKate Stone 
351e30f11d9SKate Stone         if (m_multi_line && m_curr_line_idx > 0)
352e30f11d9SKate Stone           prompt = GetContinuationPrompt();
353e30f11d9SKate Stone 
354c5dac77aSEugene Zelenko         if (prompt == nullptr)
355e30f11d9SKate Stone           prompt = GetPrompt();
356e30f11d9SKate Stone 
357b9c1b51eSKate Stone         if (prompt && prompt[0]) {
35844d93782SGreg Clayton           FILE *out = GetOutputFILE();
359b9c1b51eSKate Stone           if (out) {
36044d93782SGreg Clayton             ::fprintf(out, "%s", prompt);
36144d93782SGreg Clayton             ::fflush(out);
36244d93782SGreg Clayton           }
36344d93782SGreg Clayton         }
36444d93782SGreg Clayton       }
36544d93782SGreg Clayton       char buffer[256];
36644d93782SGreg Clayton       bool done = false;
3670f86e6e7SGreg Clayton       bool got_line = false;
368e034a04eSGreg Clayton       m_editing = true;
369b9c1b51eSKate Stone       while (!done) {
370b9c1b51eSKate Stone         if (fgets(buffer, sizeof(buffer), in) == nullptr) {
371c7797accSGreg Clayton           const int saved_errno = errno;
372c9cf5798SGreg Clayton           if (feof(in))
37344d93782SGreg Clayton             done = true;
374b9c1b51eSKate Stone           else if (ferror(in)) {
375c7797accSGreg Clayton             if (saved_errno != EINTR)
376c7797accSGreg Clayton               done = true;
377c7797accSGreg Clayton           }
378b9c1b51eSKate Stone         } else {
3790f86e6e7SGreg Clayton           got_line = true;
38044d93782SGreg Clayton           size_t buffer_len = strlen(buffer);
38144d93782SGreg Clayton           assert(buffer[buffer_len] == '\0');
38244d93782SGreg Clayton           char last_char = buffer[buffer_len - 1];
383b9c1b51eSKate Stone           if (last_char == '\r' || last_char == '\n') {
38444d93782SGreg Clayton             done = true;
38544d93782SGreg Clayton             // Strip trailing newlines
386b9c1b51eSKate Stone             while (last_char == '\r' || last_char == '\n') {
38744d93782SGreg Clayton               --buffer_len;
38844d93782SGreg Clayton               if (buffer_len == 0)
38944d93782SGreg Clayton                 break;
39044d93782SGreg Clayton               last_char = buffer[buffer_len - 1];
39144d93782SGreg Clayton             }
39244d93782SGreg Clayton           }
39344d93782SGreg Clayton           line.append(buffer, buffer_len);
39444d93782SGreg Clayton         }
39544d93782SGreg Clayton       }
396e034a04eSGreg Clayton       m_editing = false;
39705097246SAdrian Prantl       // We might have gotten a newline on a line by itself make sure to return
39805097246SAdrian Prantl       // true in this case.
3990f86e6e7SGreg Clayton       return got_line;
400b9c1b51eSKate Stone     } else {
40144d93782SGreg Clayton       // No more input file, we are done...
40244d93782SGreg Clayton       SetIsDone(true);
40344d93782SGreg Clayton     }
404340b0309SGreg Clayton     return false;
405cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
40644d93782SGreg Clayton   }
407cacde7dfSTodd Fiala #endif
40844d93782SGreg Clayton }
40944d93782SGreg Clayton 
410cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
411b9c1b51eSKate Stone bool IOHandlerEditline::IsInputCompleteCallback(Editline *editline,
41244d93782SGreg Clayton                                                 StringList &lines,
413b9c1b51eSKate Stone                                                 void *baton) {
41444d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
415b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerIsInputComplete(*editline_reader,
416b9c1b51eSKate Stone                                                               lines);
417e30f11d9SKate Stone }
418e30f11d9SKate Stone 
419b9c1b51eSKate Stone int IOHandlerEditline::FixIndentationCallback(Editline *editline,
420e30f11d9SKate Stone                                               const StringList &lines,
421e30f11d9SKate Stone                                               int cursor_position,
422b9c1b51eSKate Stone                                               void *baton) {
423e30f11d9SKate Stone   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
424b9c1b51eSKate Stone   return editline_reader->m_delegate.IOHandlerFixIndentation(
425b9c1b51eSKate Stone       *editline_reader, lines, cursor_position);
42644d93782SGreg Clayton }
42744d93782SGreg Clayton 
4287f88829cSRaphael Isemann int IOHandlerEditline::AutoCompleteCallback(
4297f88829cSRaphael Isemann     const char *current_line, const char *cursor, const char *last_char,
4307f88829cSRaphael Isemann     int skip_first_n_matches, int max_matches, StringList &matches,
4317f88829cSRaphael Isemann     StringList &descriptions, void *baton) {
43244d93782SGreg Clayton   IOHandlerEditline *editline_reader = (IOHandlerEditline *)baton;
43344d93782SGreg Clayton   if (editline_reader)
434b9c1b51eSKate Stone     return editline_reader->m_delegate.IOHandlerComplete(
435b9c1b51eSKate Stone         *editline_reader, current_line, cursor, last_char, skip_first_n_matches,
4367f88829cSRaphael Isemann         max_matches, matches, descriptions);
43744d93782SGreg Clayton   return 0;
43844d93782SGreg Clayton }
439cacde7dfSTodd Fiala #endif
44044d93782SGreg Clayton 
441b9c1b51eSKate Stone const char *IOHandlerEditline::GetPrompt() {
442cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
443b9c1b51eSKate Stone   if (m_editline_ap) {
44444d93782SGreg Clayton     return m_editline_ap->GetPrompt();
445b9c1b51eSKate Stone   } else {
446cacde7dfSTodd Fiala #endif
447cacde7dfSTodd Fiala     if (m_prompt.empty())
448c5dac77aSEugene Zelenko       return nullptr;
449cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
450cacde7dfSTodd Fiala   }
451cacde7dfSTodd Fiala #endif
45244d93782SGreg Clayton   return m_prompt.c_str();
45344d93782SGreg Clayton }
45444d93782SGreg Clayton 
455514d8cd8SZachary Turner bool IOHandlerEditline::SetPrompt(llvm::StringRef prompt) {
456514d8cd8SZachary Turner   m_prompt = prompt;
457514d8cd8SZachary Turner 
458cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
45944d93782SGreg Clayton   if (m_editline_ap)
460c5dac77aSEugene Zelenko     m_editline_ap->SetPrompt(m_prompt.empty() ? nullptr : m_prompt.c_str());
461cacde7dfSTodd Fiala #endif
46244d93782SGreg Clayton   return true;
46344d93782SGreg Clayton }
46444d93782SGreg Clayton 
465b9c1b51eSKate Stone const char *IOHandlerEditline::GetContinuationPrompt() {
466b9c1b51eSKate Stone   return (m_continuation_prompt.empty() ? nullptr
467b9c1b51eSKate Stone                                         : m_continuation_prompt.c_str());
468e30f11d9SKate Stone }
469e30f11d9SKate Stone 
470514d8cd8SZachary Turner void IOHandlerEditline::SetContinuationPrompt(llvm::StringRef prompt) {
471514d8cd8SZachary Turner   m_continuation_prompt = prompt;
472e30f11d9SKate Stone 
473d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
474e30f11d9SKate Stone   if (m_editline_ap)
475b9c1b51eSKate Stone     m_editline_ap->SetContinuationPrompt(m_continuation_prompt.empty()
476b9c1b51eSKate Stone                                              ? nullptr
477b9c1b51eSKate Stone                                              : m_continuation_prompt.c_str());
478d553d00cSZachary Turner #endif
479e30f11d9SKate Stone }
480e30f11d9SKate Stone 
481b9c1b51eSKate Stone void IOHandlerEditline::SetBaseLineNumber(uint32_t line) {
482f6913cd7SGreg Clayton   m_base_line_number = line;
483f6913cd7SGreg Clayton }
484e30f11d9SKate Stone 
485b9c1b51eSKate Stone uint32_t IOHandlerEditline::GetCurrentLineIndex() const {
486d553d00cSZachary Turner #ifndef LLDB_DISABLE_LIBEDIT
487e30f11d9SKate Stone   if (m_editline_ap)
488e30f11d9SKate Stone     return m_editline_ap->GetCurrentLine();
489e30f11d9SKate Stone #endif
490e30f11d9SKate Stone   return m_curr_line_idx;
491e30f11d9SKate Stone }
492e30f11d9SKate Stone 
493b9c1b51eSKate Stone bool IOHandlerEditline::GetLines(StringList &lines, bool &interrupted) {
494e30f11d9SKate Stone   m_current_lines_ptr = &lines;
495e30f11d9SKate Stone 
49644d93782SGreg Clayton   bool success = false;
497cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
498b9c1b51eSKate Stone   if (m_editline_ap) {
499e30f11d9SKate Stone     return m_editline_ap->GetLines(m_base_line_number, lines, interrupted);
500b9c1b51eSKate Stone   } else {
501cacde7dfSTodd Fiala #endif
502e30f11d9SKate Stone     bool done = false;
50397206d57SZachary Turner     Status error;
50444d93782SGreg Clayton 
505b9c1b51eSKate Stone     while (!done) {
506f6913cd7SGreg Clayton       // Show line numbers if we are asked to
50744d93782SGreg Clayton       std::string line;
508b9c1b51eSKate Stone       if (m_base_line_number > 0 && GetIsInteractive()) {
509f6913cd7SGreg Clayton         FILE *out = GetOutputFILE();
510f6913cd7SGreg Clayton         if (out)
511b9c1b51eSKate Stone           ::fprintf(out, "%u%s", m_base_line_number + (uint32_t)lines.GetSize(),
512b9c1b51eSKate Stone                     GetPrompt() == nullptr ? " " : "");
513f6913cd7SGreg Clayton       }
514f6913cd7SGreg Clayton 
515e30f11d9SKate Stone       m_curr_line_idx = lines.GetSize();
516e30f11d9SKate Stone 
517f0066ad0SGreg Clayton       bool interrupted = false;
518b9c1b51eSKate Stone       if (GetLine(line, interrupted) && !interrupted) {
51944d93782SGreg Clayton         lines.AppendString(line);
520e30f11d9SKate Stone         done = m_delegate.IOHandlerIsInputComplete(*this, lines);
521b9c1b51eSKate Stone       } else {
522e30f11d9SKate Stone         done = true;
52344d93782SGreg Clayton       }
52444d93782SGreg Clayton     }
52544d93782SGreg Clayton     success = lines.GetSize() > 0;
526cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
52744d93782SGreg Clayton   }
528cacde7dfSTodd Fiala #endif
52944d93782SGreg Clayton   return success;
53044d93782SGreg Clayton }
53144d93782SGreg Clayton 
53205097246SAdrian Prantl // Each IOHandler gets to run until it is done. It should read data from the
53305097246SAdrian Prantl // "in" and place output into "out" and "err and return when done.
534b9c1b51eSKate Stone void IOHandlerEditline::Run() {
53544d93782SGreg Clayton   std::string line;
536b9c1b51eSKate Stone   while (IsActive()) {
537f0066ad0SGreg Clayton     bool interrupted = false;
538b9c1b51eSKate Stone     if (m_multi_line) {
53944d93782SGreg Clayton       StringList lines;
540b9c1b51eSKate Stone       if (GetLines(lines, interrupted)) {
541b9c1b51eSKate Stone         if (interrupted) {
542e30f11d9SKate Stone           m_done = m_interrupt_exits;
543e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
544e30f11d9SKate Stone 
545b9c1b51eSKate Stone         } else {
54644d93782SGreg Clayton           line = lines.CopyList();
54744d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
54844d93782SGreg Clayton         }
549b9c1b51eSKate Stone       } else {
55044d93782SGreg Clayton         m_done = true;
55144d93782SGreg Clayton       }
552b9c1b51eSKate Stone     } else {
553b9c1b51eSKate Stone       if (GetLine(line, interrupted)) {
554e30f11d9SKate Stone         if (interrupted)
555e30f11d9SKate Stone           m_delegate.IOHandlerInputInterrupted(*this, line);
556e30f11d9SKate Stone         else
55744d93782SGreg Clayton           m_delegate.IOHandlerInputComplete(*this, line);
558b9c1b51eSKate Stone       } else {
55944d93782SGreg Clayton         m_done = true;
56044d93782SGreg Clayton       }
56144d93782SGreg Clayton     }
56244d93782SGreg Clayton   }
56344d93782SGreg Clayton }
56444d93782SGreg Clayton 
565b9c1b51eSKate Stone void IOHandlerEditline::Cancel() {
566cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
567e68f5d6bSGreg Clayton   if (m_editline_ap)
5684446487dSPavel Labath     m_editline_ap->Cancel();
569cacde7dfSTodd Fiala #endif
570e68f5d6bSGreg Clayton }
571e68f5d6bSGreg Clayton 
572b9c1b51eSKate Stone bool IOHandlerEditline::Interrupt() {
573f0066ad0SGreg Clayton   // Let the delgate handle it first
574f0066ad0SGreg Clayton   if (m_delegate.IOHandlerInterrupt(*this))
575f0066ad0SGreg Clayton     return true;
576f0066ad0SGreg Clayton 
577cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
57844d93782SGreg Clayton   if (m_editline_ap)
579f0066ad0SGreg Clayton     return m_editline_ap->Interrupt();
580cacde7dfSTodd Fiala #endif
581f0066ad0SGreg Clayton   return false;
58244d93782SGreg Clayton }
58344d93782SGreg Clayton 
584b9c1b51eSKate Stone void IOHandlerEditline::GotEOF() {
585cacde7dfSTodd Fiala #ifndef LLDB_DISABLE_LIBEDIT
58644d93782SGreg Clayton   if (m_editline_ap)
58744d93782SGreg Clayton     m_editline_ap->Interrupt();
588cacde7dfSTodd Fiala #endif
58944d93782SGreg Clayton }
59044d93782SGreg Clayton 
591b9c1b51eSKate Stone void IOHandlerEditline::PrintAsync(Stream *stream, const char *s, size_t len) {
5924446487dSPavel Labath #ifndef LLDB_DISABLE_LIBEDIT
5934446487dSPavel Labath   if (m_editline_ap)
5944446487dSPavel Labath     m_editline_ap->PrintAsync(stream, s, len);
5954446487dSPavel Labath   else
5964446487dSPavel Labath #endif
597fab31220STed Woodward   {
598fab31220STed Woodward #ifdef _MSC_VER
599341e4789SDawn Perchik     const char *prompt = GetPrompt();
600b9c1b51eSKate Stone     if (prompt) {
601fab31220STed Woodward       // Back up over previous prompt using Windows API
602fab31220STed Woodward       CONSOLE_SCREEN_BUFFER_INFO screen_buffer_info;
603fab31220STed Woodward       HANDLE console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
604fab31220STed Woodward       GetConsoleScreenBufferInfo(console_handle, &screen_buffer_info);
605fab31220STed Woodward       COORD coord = screen_buffer_info.dwCursorPosition;
606fab31220STed Woodward       coord.X -= strlen(prompt);
607fab31220STed Woodward       if (coord.X < 0)
608fab31220STed Woodward         coord.X = 0;
609fab31220STed Woodward       SetConsoleCursorPosition(console_handle, coord);
610fab31220STed Woodward     }
611fab31220STed Woodward #endif
6124446487dSPavel Labath     IOHandler::PrintAsync(stream, s, len);
613341e4789SDawn Perchik #ifdef _MSC_VER
614fab31220STed Woodward     if (prompt)
615b9c1b51eSKate Stone       IOHandler::PrintAsync(GetOutputStreamFile().get(), prompt,
616b9c1b51eSKate Stone                             strlen(prompt));
617341e4789SDawn Perchik #endif
618fab31220STed Woodward   }
6194446487dSPavel Labath }
6204446487dSPavel Labath 
62105097246SAdrian Prantl // we may want curses to be disabled for some builds for instance, windows
622914b8d98SDeepak Panickal #ifndef LLDB_DISABLE_CURSES
623914b8d98SDeepak Panickal 
62444d93782SGreg Clayton #define KEY_RETURN 10
62544d93782SGreg Clayton #define KEY_ESCAPE 27
62644d93782SGreg Clayton 
627b9c1b51eSKate Stone namespace curses {
62844d93782SGreg Clayton class Menu;
62944d93782SGreg Clayton class MenuDelegate;
63044d93782SGreg Clayton class Window;
63144d93782SGreg Clayton class WindowDelegate;
63244d93782SGreg Clayton typedef std::shared_ptr<Menu> MenuSP;
63344d93782SGreg Clayton typedef std::shared_ptr<MenuDelegate> MenuDelegateSP;
63444d93782SGreg Clayton typedef std::shared_ptr<Window> WindowSP;
63544d93782SGreg Clayton typedef std::shared_ptr<WindowDelegate> WindowDelegateSP;
63644d93782SGreg Clayton typedef std::vector<MenuSP> Menus;
63744d93782SGreg Clayton typedef std::vector<WindowSP> Windows;
63844d93782SGreg Clayton typedef std::vector<WindowDelegateSP> WindowDelegates;
63944d93782SGreg Clayton 
64044d93782SGreg Clayton #if 0
64144d93782SGreg Clayton type summary add -s "x=${var.x}, y=${var.y}" curses::Point
64244d93782SGreg Clayton type summary add -s "w=${var.width}, h=${var.height}" curses::Size
64344d93782SGreg Clayton type summary add -s "${var.origin%S} ${var.size%S}" curses::Rect
64444d93782SGreg Clayton #endif
645315b6884SEugene Zelenko 
646b9c1b51eSKate Stone struct Point {
64744d93782SGreg Clayton   int x;
64844d93782SGreg Clayton   int y;
64944d93782SGreg Clayton 
650b9c1b51eSKate Stone   Point(int _x = 0, int _y = 0) : x(_x), y(_y) {}
65144d93782SGreg Clayton 
652b9c1b51eSKate Stone   void Clear() {
65344d93782SGreg Clayton     x = 0;
65444d93782SGreg Clayton     y = 0;
65544d93782SGreg Clayton   }
65644d93782SGreg Clayton 
657b9c1b51eSKate Stone   Point &operator+=(const Point &rhs) {
65844d93782SGreg Clayton     x += rhs.x;
65944d93782SGreg Clayton     y += rhs.y;
66044d93782SGreg Clayton     return *this;
66144d93782SGreg Clayton   }
66244d93782SGreg Clayton 
663b9c1b51eSKate Stone   void Dump() { printf("(x=%i, y=%i)\n", x, y); }
66444d93782SGreg Clayton };
66544d93782SGreg Clayton 
666b9c1b51eSKate Stone bool operator==(const Point &lhs, const Point &rhs) {
66744d93782SGreg Clayton   return lhs.x == rhs.x && lhs.y == rhs.y;
66844d93782SGreg Clayton }
669315b6884SEugene Zelenko 
670b9c1b51eSKate Stone bool operator!=(const Point &lhs, const Point &rhs) {
67144d93782SGreg Clayton   return lhs.x != rhs.x || lhs.y != rhs.y;
67244d93782SGreg Clayton }
67344d93782SGreg Clayton 
674b9c1b51eSKate Stone struct Size {
67544d93782SGreg Clayton   int width;
67644d93782SGreg Clayton   int height;
677b9c1b51eSKate Stone   Size(int w = 0, int h = 0) : width(w), height(h) {}
67844d93782SGreg Clayton 
679b9c1b51eSKate Stone   void Clear() {
68044d93782SGreg Clayton     width = 0;
68144d93782SGreg Clayton     height = 0;
68244d93782SGreg Clayton   }
68344d93782SGreg Clayton 
684b9c1b51eSKate Stone   void Dump() { printf("(w=%i, h=%i)\n", width, height); }
68544d93782SGreg Clayton };
68644d93782SGreg Clayton 
687b9c1b51eSKate Stone bool operator==(const Size &lhs, const Size &rhs) {
68844d93782SGreg Clayton   return lhs.width == rhs.width && lhs.height == rhs.height;
68944d93782SGreg Clayton }
690315b6884SEugene Zelenko 
691b9c1b51eSKate Stone bool operator!=(const Size &lhs, const Size &rhs) {
69244d93782SGreg Clayton   return lhs.width != rhs.width || lhs.height != rhs.height;
69344d93782SGreg Clayton }
69444d93782SGreg Clayton 
695b9c1b51eSKate Stone struct Rect {
69644d93782SGreg Clayton   Point origin;
69744d93782SGreg Clayton   Size size;
69844d93782SGreg Clayton 
699b9c1b51eSKate Stone   Rect() : origin(), size() {}
70044d93782SGreg Clayton 
701b9c1b51eSKate Stone   Rect(const Point &p, const Size &s) : origin(p), size(s) {}
70244d93782SGreg Clayton 
703b9c1b51eSKate Stone   void Clear() {
70444d93782SGreg Clayton     origin.Clear();
70544d93782SGreg Clayton     size.Clear();
70644d93782SGreg Clayton   }
70744d93782SGreg Clayton 
708b9c1b51eSKate Stone   void Dump() {
709b9c1b51eSKate Stone     printf("(x=%i, y=%i), w=%i, h=%i)\n", origin.x, origin.y, size.width,
710b9c1b51eSKate Stone            size.height);
71144d93782SGreg Clayton   }
71244d93782SGreg Clayton 
713b9c1b51eSKate Stone   void Inset(int w, int h) {
71444d93782SGreg Clayton     if (size.width > w * 2)
71544d93782SGreg Clayton       size.width -= w * 2;
71644d93782SGreg Clayton     origin.x += w;
71744d93782SGreg Clayton 
71844d93782SGreg Clayton     if (size.height > h * 2)
71944d93782SGreg Clayton       size.height -= h * 2;
72044d93782SGreg Clayton     origin.y += h;
72144d93782SGreg Clayton   }
722315b6884SEugene Zelenko 
72305097246SAdrian Prantl   // Return a status bar rectangle which is the last line of this rectangle.
72405097246SAdrian Prantl   // This rectangle will be modified to not include the status bar area.
725b9c1b51eSKate Stone   Rect MakeStatusBar() {
72644d93782SGreg Clayton     Rect status_bar;
727b9c1b51eSKate Stone     if (size.height > 1) {
72844d93782SGreg Clayton       status_bar.origin.x = origin.x;
72944d93782SGreg Clayton       status_bar.origin.y = size.height;
73044d93782SGreg Clayton       status_bar.size.width = size.width;
73144d93782SGreg Clayton       status_bar.size.height = 1;
73244d93782SGreg Clayton       --size.height;
73344d93782SGreg Clayton     }
73444d93782SGreg Clayton     return status_bar;
73544d93782SGreg Clayton   }
73644d93782SGreg Clayton 
73705097246SAdrian Prantl   // Return a menubar rectangle which is the first line of this rectangle. This
73805097246SAdrian Prantl   // rectangle will be modified to not include the menubar area.
739b9c1b51eSKate Stone   Rect MakeMenuBar() {
74044d93782SGreg Clayton     Rect menubar;
741b9c1b51eSKate Stone     if (size.height > 1) {
74244d93782SGreg Clayton       menubar.origin.x = origin.x;
74344d93782SGreg Clayton       menubar.origin.y = origin.y;
74444d93782SGreg Clayton       menubar.size.width = size.width;
74544d93782SGreg Clayton       menubar.size.height = 1;
74644d93782SGreg Clayton       ++origin.y;
74744d93782SGreg Clayton       --size.height;
74844d93782SGreg Clayton     }
74944d93782SGreg Clayton     return menubar;
75044d93782SGreg Clayton   }
75144d93782SGreg Clayton 
752b9c1b51eSKate Stone   void HorizontalSplitPercentage(float top_percentage, Rect &top,
753b9c1b51eSKate Stone                                  Rect &bottom) const {
75444d93782SGreg Clayton     float top_height = top_percentage * size.height;
75544d93782SGreg Clayton     HorizontalSplit(top_height, top, bottom);
75644d93782SGreg Clayton   }
75744d93782SGreg Clayton 
758b9c1b51eSKate Stone   void HorizontalSplit(int top_height, Rect &top, Rect &bottom) const {
75944d93782SGreg Clayton     top = *this;
760b9c1b51eSKate Stone     if (top_height < size.height) {
76144d93782SGreg Clayton       top.size.height = top_height;
76244d93782SGreg Clayton       bottom.origin.x = origin.x;
76344d93782SGreg Clayton       bottom.origin.y = origin.y + top.size.height;
76444d93782SGreg Clayton       bottom.size.width = size.width;
76544d93782SGreg Clayton       bottom.size.height = size.height - top.size.height;
766b9c1b51eSKate Stone     } else {
76744d93782SGreg Clayton       bottom.Clear();
76844d93782SGreg Clayton     }
76944d93782SGreg Clayton   }
77044d93782SGreg Clayton 
771b9c1b51eSKate Stone   void VerticalSplitPercentage(float left_percentage, Rect &left,
772b9c1b51eSKate Stone                                Rect &right) const {
77344d93782SGreg Clayton     float left_width = left_percentage * size.width;
77444d93782SGreg Clayton     VerticalSplit(left_width, left, right);
77544d93782SGreg Clayton   }
77644d93782SGreg Clayton 
777b9c1b51eSKate Stone   void VerticalSplit(int left_width, Rect &left, Rect &right) const {
77844d93782SGreg Clayton     left = *this;
779b9c1b51eSKate Stone     if (left_width < size.width) {
78044d93782SGreg Clayton       left.size.width = left_width;
78144d93782SGreg Clayton       right.origin.x = origin.x + left.size.width;
78244d93782SGreg Clayton       right.origin.y = origin.y;
78344d93782SGreg Clayton       right.size.width = size.width - left.size.width;
78444d93782SGreg Clayton       right.size.height = size.height;
785b9c1b51eSKate Stone     } else {
78644d93782SGreg Clayton       right.Clear();
78744d93782SGreg Clayton     }
78844d93782SGreg Clayton   }
78944d93782SGreg Clayton };
79044d93782SGreg Clayton 
791b9c1b51eSKate Stone bool operator==(const Rect &lhs, const Rect &rhs) {
79244d93782SGreg Clayton   return lhs.origin == rhs.origin && lhs.size == rhs.size;
79344d93782SGreg Clayton }
794315b6884SEugene Zelenko 
795b9c1b51eSKate Stone bool operator!=(const Rect &lhs, const Rect &rhs) {
79644d93782SGreg Clayton   return lhs.origin != rhs.origin || lhs.size != rhs.size;
79744d93782SGreg Clayton }
79844d93782SGreg Clayton 
799b9c1b51eSKate Stone enum HandleCharResult {
80044d93782SGreg Clayton   eKeyNotHandled = 0,
80144d93782SGreg Clayton   eKeyHandled = 1,
80244d93782SGreg Clayton   eQuitApplication = 2
80344d93782SGreg Clayton };
80444d93782SGreg Clayton 
805b9c1b51eSKate Stone enum class MenuActionResult {
80644d93782SGreg Clayton   Handled,
80744d93782SGreg Clayton   NotHandled,
80844d93782SGreg Clayton   Quit // Exit all menus and quit
80944d93782SGreg Clayton };
81044d93782SGreg Clayton 
811b9c1b51eSKate Stone struct KeyHelp {
81244d93782SGreg Clayton   int ch;
81344d93782SGreg Clayton   const char *description;
81444d93782SGreg Clayton };
81544d93782SGreg Clayton 
816b9c1b51eSKate Stone class WindowDelegate {
81744d93782SGreg Clayton public:
818b9c1b51eSKate Stone   virtual ~WindowDelegate() = default;
81944d93782SGreg Clayton 
820b9c1b51eSKate Stone   virtual bool WindowDelegateDraw(Window &window, bool force) {
82144d93782SGreg Clayton     return false; // Drawing not handled
82244d93782SGreg Clayton   }
82344d93782SGreg Clayton 
824b9c1b51eSKate Stone   virtual HandleCharResult WindowDelegateHandleChar(Window &window, int key) {
82544d93782SGreg Clayton     return eKeyNotHandled;
82644d93782SGreg Clayton   }
82744d93782SGreg Clayton 
828b9c1b51eSKate Stone   virtual const char *WindowDelegateGetHelpText() { return nullptr; }
82944d93782SGreg Clayton 
830b9c1b51eSKate Stone   virtual KeyHelp *WindowDelegateGetKeyHelp() { return nullptr; }
83144d93782SGreg Clayton };
83244d93782SGreg Clayton 
833b9c1b51eSKate Stone class HelpDialogDelegate : public WindowDelegate {
83444d93782SGreg Clayton public:
83544d93782SGreg Clayton   HelpDialogDelegate(const char *text, KeyHelp *key_help_array);
83644d93782SGreg Clayton 
837bd5ae6b4SGreg Clayton   ~HelpDialogDelegate() override;
83844d93782SGreg Clayton 
839b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
84044d93782SGreg Clayton 
841b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
84244d93782SGreg Clayton 
843b9c1b51eSKate Stone   size_t GetNumLines() const { return m_text.GetSize(); }
84444d93782SGreg Clayton 
845b9c1b51eSKate Stone   size_t GetMaxLineLength() const { return m_text.GetMaxStringLength(); }
84644d93782SGreg Clayton 
84744d93782SGreg Clayton protected:
84844d93782SGreg Clayton   StringList m_text;
84944d93782SGreg Clayton   int m_first_visible_line;
85044d93782SGreg Clayton };
85144d93782SGreg Clayton 
852b9c1b51eSKate Stone class Window {
85344d93782SGreg Clayton public:
854b9c1b51eSKate Stone   Window(const char *name)
855b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
856b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
857b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(false),
858b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {}
85944d93782SGreg Clayton 
860b9c1b51eSKate Stone   Window(const char *name, WINDOW *w, bool del = true)
861b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_panel(nullptr), m_parent(nullptr),
862b9c1b51eSKate Stone         m_subwindows(), m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
863b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(del),
864b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
86544d93782SGreg Clayton     if (w)
86644d93782SGreg Clayton       Reset(w);
86744d93782SGreg Clayton   }
86844d93782SGreg Clayton 
869b9c1b51eSKate Stone   Window(const char *name, const Rect &bounds)
870b9c1b51eSKate Stone       : m_name(name), m_window(nullptr), m_parent(nullptr), m_subwindows(),
871b9c1b51eSKate Stone         m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
872b9c1b51eSKate Stone         m_prev_active_window_idx(UINT32_MAX), m_delete(true),
873b9c1b51eSKate Stone         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
874b9c1b51eSKate Stone     Reset(::newwin(bounds.size.height, bounds.size.width, bounds.origin.y,
875b9c1b51eSKate Stone                    bounds.origin.y));
87644d93782SGreg Clayton   }
87744d93782SGreg Clayton 
878b9c1b51eSKate Stone   virtual ~Window() {
87944d93782SGreg Clayton     RemoveSubWindows();
88044d93782SGreg Clayton     Reset();
88144d93782SGreg Clayton   }
88244d93782SGreg Clayton 
883b9c1b51eSKate Stone   void Reset(WINDOW *w = nullptr, bool del = true) {
88444d93782SGreg Clayton     if (m_window == w)
88544d93782SGreg Clayton       return;
88644d93782SGreg Clayton 
887b9c1b51eSKate Stone     if (m_panel) {
88844d93782SGreg Clayton       ::del_panel(m_panel);
889c5dac77aSEugene Zelenko       m_panel = nullptr;
89044d93782SGreg Clayton     }
891b9c1b51eSKate Stone     if (m_window && m_delete) {
89244d93782SGreg Clayton       ::delwin(m_window);
893c5dac77aSEugene Zelenko       m_window = nullptr;
89444d93782SGreg Clayton       m_delete = false;
89544d93782SGreg Clayton     }
896b9c1b51eSKate Stone     if (w) {
89744d93782SGreg Clayton       m_window = w;
89844d93782SGreg Clayton       m_panel = ::new_panel(m_window);
89944d93782SGreg Clayton       m_delete = del;
90044d93782SGreg Clayton     }
90144d93782SGreg Clayton   }
90244d93782SGreg Clayton 
90344d93782SGreg Clayton   void AttributeOn(attr_t attr) { ::wattron(m_window, attr); }
90444d93782SGreg Clayton   void AttributeOff(attr_t attr) { ::wattroff(m_window, attr); }
905b9c1b51eSKate Stone   void Box(chtype v_char = ACS_VLINE, chtype h_char = ACS_HLINE) {
906b9c1b51eSKate Stone     ::box(m_window, v_char, h_char);
907b9c1b51eSKate Stone   }
90844d93782SGreg Clayton   void Clear() { ::wclear(m_window); }
90944d93782SGreg Clayton   void Erase() { ::werase(m_window); }
910b9c1b51eSKate Stone   Rect GetBounds() {
911b9c1b51eSKate Stone     return Rect(GetParentOrigin(), GetSize());
912b9c1b51eSKate Stone   } // Get the rectangle in our parent window
91344d93782SGreg Clayton   int GetChar() { return ::wgetch(m_window); }
91444d93782SGreg Clayton   int GetCursorX() { return getcurx(m_window); }
91544d93782SGreg Clayton   int GetCursorY() { return getcury(m_window); }
916b9c1b51eSKate Stone   Rect GetFrame() {
917b9c1b51eSKate Stone     return Rect(Point(), GetSize());
918b9c1b51eSKate Stone   } // Get our rectangle in our own coordinate system
91944d93782SGreg Clayton   Point GetParentOrigin() { return Point(GetParentX(), GetParentY()); }
92044d93782SGreg Clayton   Size GetSize() { return Size(GetWidth(), GetHeight()); }
92144d93782SGreg Clayton   int GetParentX() { return getparx(m_window); }
92244d93782SGreg Clayton   int GetParentY() { return getpary(m_window); }
92344d93782SGreg Clayton   int GetMaxX() { return getmaxx(m_window); }
92444d93782SGreg Clayton   int GetMaxY() { return getmaxy(m_window); }
92544d93782SGreg Clayton   int GetWidth() { return GetMaxX(); }
92644d93782SGreg Clayton   int GetHeight() { return GetMaxY(); }
92744d93782SGreg Clayton   void MoveCursor(int x, int y) { ::wmove(m_window, y, x); }
92844d93782SGreg Clayton   void MoveWindow(int x, int y) { MoveWindow(Point(x, y)); }
92944d93782SGreg Clayton   void Resize(int w, int h) { ::wresize(m_window, h, w); }
930b9c1b51eSKate Stone   void Resize(const Size &size) {
931b9c1b51eSKate Stone     ::wresize(m_window, size.height, size.width);
932b9c1b51eSKate Stone   }
93344d93782SGreg Clayton   void PutChar(int ch) { ::waddch(m_window, ch); }
93444d93782SGreg Clayton   void PutCString(const char *s, int len = -1) { ::waddnstr(m_window, s, len); }
93544d93782SGreg Clayton   void Refresh() { ::wrefresh(m_window); }
936b9c1b51eSKate Stone   void DeferredRefresh() {
93744d93782SGreg Clayton     // We are using panels, so we don't need to call this...
93844d93782SGreg Clayton     //::wnoutrefresh(m_window);
93944d93782SGreg Clayton   }
940b9c1b51eSKate Stone   void SetBackground(int color_pair_idx) {
941b9c1b51eSKate Stone     ::wbkgd(m_window, COLOR_PAIR(color_pair_idx));
942b9c1b51eSKate Stone   }
94344d93782SGreg Clayton   void UnderlineOn() { AttributeOn(A_UNDERLINE); }
94444d93782SGreg Clayton   void UnderlineOff() { AttributeOff(A_UNDERLINE); }
94544d93782SGreg Clayton 
946b9c1b51eSKate Stone   void PutCStringTruncated(const char *s, int right_pad) {
94744d93782SGreg Clayton     int bytes_left = GetWidth() - GetCursorX();
948b9c1b51eSKate Stone     if (bytes_left > right_pad) {
94944d93782SGreg Clayton       bytes_left -= right_pad;
95044d93782SGreg Clayton       ::waddnstr(m_window, s, bytes_left);
95144d93782SGreg Clayton     }
95244d93782SGreg Clayton   }
95344d93782SGreg Clayton 
954b9c1b51eSKate Stone   void MoveWindow(const Point &origin) {
95544d93782SGreg Clayton     const bool moving_window = origin != GetParentOrigin();
956b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
95744d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
95844d93782SGreg Clayton       Size size = GetSize();
959b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, size.height, size.width, origin.y,
960b9c1b51eSKate Stone                      origin.x),
961b9c1b51eSKate Stone             true);
962b9c1b51eSKate Stone     } else {
96344d93782SGreg Clayton       ::mvwin(m_window, origin.y, origin.x);
96444d93782SGreg Clayton     }
96544d93782SGreg Clayton   }
96644d93782SGreg Clayton 
967b9c1b51eSKate Stone   void SetBounds(const Rect &bounds) {
96844d93782SGreg Clayton     const bool moving_window = bounds.origin != GetParentOrigin();
969b9c1b51eSKate Stone     if (m_is_subwin && moving_window) {
97044d93782SGreg Clayton       // Can't move subwindows, must delete and re-create
971b9c1b51eSKate Stone       Reset(::subwin(m_parent->m_window, bounds.size.height, bounds.size.width,
972b9c1b51eSKate Stone                      bounds.origin.y, bounds.origin.x),
973b9c1b51eSKate Stone             true);
974b9c1b51eSKate Stone     } else {
97544d93782SGreg Clayton       if (moving_window)
97644d93782SGreg Clayton         MoveWindow(bounds.origin);
97744d93782SGreg Clayton       Resize(bounds.size);
97844d93782SGreg Clayton     }
97944d93782SGreg Clayton   }
98044d93782SGreg Clayton 
981b9c1b51eSKate Stone   void Printf(const char *format, ...) __attribute__((format(printf, 2, 3))) {
98244d93782SGreg Clayton     va_list args;
98344d93782SGreg Clayton     va_start(args, format);
98444d93782SGreg Clayton     vwprintw(m_window, format, args);
98544d93782SGreg Clayton     va_end(args);
98644d93782SGreg Clayton   }
98744d93782SGreg Clayton 
988b9c1b51eSKate Stone   void Touch() {
98944d93782SGreg Clayton     ::touchwin(m_window);
99044d93782SGreg Clayton     if (m_parent)
99144d93782SGreg Clayton       m_parent->Touch();
99244d93782SGreg Clayton   }
99344d93782SGreg Clayton 
994b9c1b51eSKate Stone   WindowSP CreateSubWindow(const char *name, const Rect &bounds,
995b9c1b51eSKate Stone                            bool make_active) {
99644d93782SGreg Clayton     WindowSP subwindow_sp;
997b9c1b51eSKate Stone     if (m_window) {
998*796ac80bSJonas Devlieghere       subwindow_sp = std::make_shared<Window>(
999*796ac80bSJonas Devlieghere           name,
1000*796ac80bSJonas Devlieghere           ::subwin(m_window, bounds.size.height, bounds.size.width,
1001b9c1b51eSKate Stone                          bounds.origin.y, bounds.origin.x),
1002*796ac80bSJonas Devlieghere           true);
100344d93782SGreg Clayton       subwindow_sp->m_is_subwin = true;
1004b9c1b51eSKate Stone     } else {
1005*796ac80bSJonas Devlieghere       subwindow_sp = std::make_shared<Window>(
1006*796ac80bSJonas Devlieghere           name,
1007*796ac80bSJonas Devlieghere           ::newwin(bounds.size.height, bounds.size.width, bounds.origin.y,
1008*796ac80bSJonas Devlieghere                    bounds.origin.x),
1009*796ac80bSJonas Devlieghere           true);
101044d93782SGreg Clayton       subwindow_sp->m_is_subwin = false;
101144d93782SGreg Clayton     }
101244d93782SGreg Clayton     subwindow_sp->m_parent = this;
1013b9c1b51eSKate Stone     if (make_active) {
101444d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
101544d93782SGreg Clayton       m_curr_active_window_idx = m_subwindows.size();
101644d93782SGreg Clayton     }
101744d93782SGreg Clayton     m_subwindows.push_back(subwindow_sp);
101844d93782SGreg Clayton     ::top_panel(subwindow_sp->m_panel);
101944d93782SGreg Clayton     m_needs_update = true;
102044d93782SGreg Clayton     return subwindow_sp;
102144d93782SGreg Clayton   }
102244d93782SGreg Clayton 
1023b9c1b51eSKate Stone   bool RemoveSubWindow(Window *window) {
102444d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
102544d93782SGreg Clayton     size_t i = 0;
1026b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
1027b9c1b51eSKate Stone       if ((*pos).get() == window) {
102844d93782SGreg Clayton         if (m_prev_active_window_idx == i)
102944d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1030b9c1b51eSKate Stone         else if (m_prev_active_window_idx != UINT32_MAX &&
1031b9c1b51eSKate Stone                  m_prev_active_window_idx > i)
103244d93782SGreg Clayton           --m_prev_active_window_idx;
103344d93782SGreg Clayton 
103444d93782SGreg Clayton         if (m_curr_active_window_idx == i)
103544d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
1036b9c1b51eSKate Stone         else if (m_curr_active_window_idx != UINT32_MAX &&
1037b9c1b51eSKate Stone                  m_curr_active_window_idx > i)
103844d93782SGreg Clayton           --m_curr_active_window_idx;
103944d93782SGreg Clayton         window->Erase();
104044d93782SGreg Clayton         m_subwindows.erase(pos);
104144d93782SGreg Clayton         m_needs_update = true;
104244d93782SGreg Clayton         if (m_parent)
104344d93782SGreg Clayton           m_parent->Touch();
104444d93782SGreg Clayton         else
104544d93782SGreg Clayton           ::touchwin(stdscr);
104644d93782SGreg Clayton         return true;
104744d93782SGreg Clayton       }
104844d93782SGreg Clayton     }
104944d93782SGreg Clayton     return false;
105044d93782SGreg Clayton   }
105144d93782SGreg Clayton 
1052b9c1b51eSKate Stone   WindowSP FindSubWindow(const char *name) {
105344d93782SGreg Clayton     Windows::iterator pos, end = m_subwindows.end();
105444d93782SGreg Clayton     size_t i = 0;
1055b9c1b51eSKate Stone     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
10568d20cfdfSJonas Devlieghere       if ((*pos)->m_name == name)
105744d93782SGreg Clayton         return *pos;
105844d93782SGreg Clayton     }
105944d93782SGreg Clayton     return WindowSP();
106044d93782SGreg Clayton   }
106144d93782SGreg Clayton 
1062b9c1b51eSKate Stone   void RemoveSubWindows() {
106344d93782SGreg Clayton     m_curr_active_window_idx = UINT32_MAX;
106444d93782SGreg Clayton     m_prev_active_window_idx = UINT32_MAX;
106544d93782SGreg Clayton     for (Windows::iterator pos = m_subwindows.begin();
1066b9c1b51eSKate Stone          pos != m_subwindows.end(); pos = m_subwindows.erase(pos)) {
106744d93782SGreg Clayton       (*pos)->Erase();
106844d93782SGreg Clayton     }
106944d93782SGreg Clayton     if (m_parent)
107044d93782SGreg Clayton       m_parent->Touch();
107144d93782SGreg Clayton     else
107244d93782SGreg Clayton       ::touchwin(stdscr);
107344d93782SGreg Clayton   }
107444d93782SGreg Clayton 
1075b9c1b51eSKate Stone   WINDOW *get() { return m_window; }
107644d93782SGreg Clayton 
1077b9c1b51eSKate Stone   operator WINDOW *() { return m_window; }
107844d93782SGreg Clayton 
107944d93782SGreg Clayton   //----------------------------------------------------------------------
108044d93782SGreg Clayton   // Window drawing utilities
108144d93782SGreg Clayton   //----------------------------------------------------------------------
1082b9c1b51eSKate Stone   void DrawTitleBox(const char *title, const char *bottom_message = nullptr) {
108344d93782SGreg Clayton     attr_t attr = 0;
108444d93782SGreg Clayton     if (IsActive())
108544d93782SGreg Clayton       attr = A_BOLD | COLOR_PAIR(2);
108644d93782SGreg Clayton     else
108744d93782SGreg Clayton       attr = 0;
108844d93782SGreg Clayton     if (attr)
108944d93782SGreg Clayton       AttributeOn(attr);
109044d93782SGreg Clayton 
109144d93782SGreg Clayton     Box();
109244d93782SGreg Clayton     MoveCursor(3, 0);
109344d93782SGreg Clayton 
1094b9c1b51eSKate Stone     if (title && title[0]) {
109544d93782SGreg Clayton       PutChar('<');
109644d93782SGreg Clayton       PutCString(title);
109744d93782SGreg Clayton       PutChar('>');
109844d93782SGreg Clayton     }
109944d93782SGreg Clayton 
1100b9c1b51eSKate Stone     if (bottom_message && bottom_message[0]) {
110144d93782SGreg Clayton       int bottom_message_length = strlen(bottom_message);
110244d93782SGreg Clayton       int x = GetWidth() - 3 - (bottom_message_length + 2);
110344d93782SGreg Clayton 
1104b9c1b51eSKate Stone       if (x > 0) {
110544d93782SGreg Clayton         MoveCursor(x, GetHeight() - 1);
110644d93782SGreg Clayton         PutChar('[');
110744d93782SGreg Clayton         PutCString(bottom_message);
110844d93782SGreg Clayton         PutChar(']');
1109b9c1b51eSKate Stone       } else {
111044d93782SGreg Clayton         MoveCursor(1, GetHeight() - 1);
111144d93782SGreg Clayton         PutChar('[');
111244d93782SGreg Clayton         PutCStringTruncated(bottom_message, 1);
111344d93782SGreg Clayton       }
111444d93782SGreg Clayton     }
111544d93782SGreg Clayton     if (attr)
111644d93782SGreg Clayton       AttributeOff(attr);
111744d93782SGreg Clayton   }
111844d93782SGreg Clayton 
1119b9c1b51eSKate Stone   virtual void Draw(bool force) {
112044d93782SGreg Clayton     if (m_delegate_sp && m_delegate_sp->WindowDelegateDraw(*this, force))
112144d93782SGreg Clayton       return;
112244d93782SGreg Clayton 
112344d93782SGreg Clayton     for (auto &subwindow_sp : m_subwindows)
112444d93782SGreg Clayton       subwindow_sp->Draw(force);
112544d93782SGreg Clayton   }
112644d93782SGreg Clayton 
1127b9c1b51eSKate Stone   bool CreateHelpSubwindow() {
1128b9c1b51eSKate Stone     if (m_delegate_sp) {
112944d93782SGreg Clayton       const char *text = m_delegate_sp->WindowDelegateGetHelpText();
113044d93782SGreg Clayton       KeyHelp *key_help = m_delegate_sp->WindowDelegateGetKeyHelp();
1131b9c1b51eSKate Stone       if ((text && text[0]) || key_help) {
1132e1a6074bSBenjamin Kramer         std::unique_ptr<HelpDialogDelegate> help_delegate_ap(
1133b9c1b51eSKate Stone             new HelpDialogDelegate(text, key_help));
113444d93782SGreg Clayton         const size_t num_lines = help_delegate_ap->GetNumLines();
113544d93782SGreg Clayton         const size_t max_length = help_delegate_ap->GetMaxLineLength();
113644d93782SGreg Clayton         Rect bounds = GetBounds();
113744d93782SGreg Clayton         bounds.Inset(1, 1);
1138b9c1b51eSKate Stone         if (max_length + 4 < static_cast<size_t>(bounds.size.width)) {
113944d93782SGreg Clayton           bounds.origin.x += (bounds.size.width - max_length + 4) / 2;
114044d93782SGreg Clayton           bounds.size.width = max_length + 4;
1141b9c1b51eSKate Stone         } else {
1142b9c1b51eSKate Stone           if (bounds.size.width > 100) {
114344d93782SGreg Clayton             const int inset_w = bounds.size.width / 4;
114444d93782SGreg Clayton             bounds.origin.x += inset_w;
114544d93782SGreg Clayton             bounds.size.width -= 2 * inset_w;
114644d93782SGreg Clayton           }
114744d93782SGreg Clayton         }
114844d93782SGreg Clayton 
1149b9c1b51eSKate Stone         if (num_lines + 2 < static_cast<size_t>(bounds.size.height)) {
115044d93782SGreg Clayton           bounds.origin.y += (bounds.size.height - num_lines + 2) / 2;
115144d93782SGreg Clayton           bounds.size.height = num_lines + 2;
1152b9c1b51eSKate Stone         } else {
1153b9c1b51eSKate Stone           if (bounds.size.height > 100) {
115444d93782SGreg Clayton             const int inset_h = bounds.size.height / 4;
115544d93782SGreg Clayton             bounds.origin.y += inset_h;
115644d93782SGreg Clayton             bounds.size.height -= 2 * inset_h;
115744d93782SGreg Clayton           }
115844d93782SGreg Clayton         }
11595fdb09bbSGreg Clayton         WindowSP help_window_sp;
11605fdb09bbSGreg Clayton         Window *parent_window = GetParent();
11615fdb09bbSGreg Clayton         if (parent_window)
11625fdb09bbSGreg Clayton           help_window_sp = parent_window->CreateSubWindow("Help", bounds, true);
11635fdb09bbSGreg Clayton         else
11645fdb09bbSGreg Clayton           help_window_sp = CreateSubWindow("Help", bounds, true);
1165b9c1b51eSKate Stone         help_window_sp->SetDelegate(
1166b9c1b51eSKate Stone             WindowDelegateSP(help_delegate_ap.release()));
116744d93782SGreg Clayton         return true;
116844d93782SGreg Clayton       }
116944d93782SGreg Clayton     }
117044d93782SGreg Clayton     return false;
117144d93782SGreg Clayton   }
117244d93782SGreg Clayton 
1173b9c1b51eSKate Stone   virtual HandleCharResult HandleChar(int key) {
117444d93782SGreg Clayton     // Always check the active window first
117544d93782SGreg Clayton     HandleCharResult result = eKeyNotHandled;
117644d93782SGreg Clayton     WindowSP active_window_sp = GetActiveWindow();
1177b9c1b51eSKate Stone     if (active_window_sp) {
117844d93782SGreg Clayton       result = active_window_sp->HandleChar(key);
117944d93782SGreg Clayton       if (result != eKeyNotHandled)
118044d93782SGreg Clayton         return result;
118144d93782SGreg Clayton     }
118244d93782SGreg Clayton 
1183b9c1b51eSKate Stone     if (m_delegate_sp) {
118444d93782SGreg Clayton       result = m_delegate_sp->WindowDelegateHandleChar(*this, key);
118544d93782SGreg Clayton       if (result != eKeyNotHandled)
118644d93782SGreg Clayton         return result;
118744d93782SGreg Clayton     }
118844d93782SGreg Clayton 
118905097246SAdrian Prantl     // Then check for any windows that want any keys that weren't handled. This
119005097246SAdrian Prantl     // is typically only for a menubar. Make a copy of the subwindows in case
119105097246SAdrian Prantl     // any HandleChar() functions muck with the subwindows. If we don't do
119205097246SAdrian Prantl     // this, we can crash when iterating over the subwindows.
119344d93782SGreg Clayton     Windows subwindows(m_subwindows);
1194b9c1b51eSKate Stone     for (auto subwindow_sp : subwindows) {
1195b9c1b51eSKate Stone       if (!subwindow_sp->m_can_activate) {
119644d93782SGreg Clayton         HandleCharResult result = subwindow_sp->HandleChar(key);
119744d93782SGreg Clayton         if (result != eKeyNotHandled)
119844d93782SGreg Clayton           return result;
119944d93782SGreg Clayton       }
120044d93782SGreg Clayton     }
120144d93782SGreg Clayton 
120244d93782SGreg Clayton     return eKeyNotHandled;
120344d93782SGreg Clayton   }
120444d93782SGreg Clayton 
1205b9c1b51eSKate Stone   bool SetActiveWindow(Window *window) {
120644d93782SGreg Clayton     const size_t num_subwindows = m_subwindows.size();
1207b9c1b51eSKate Stone     for (size_t i = 0; i < num_subwindows; ++i) {
1208b9c1b51eSKate Stone       if (m_subwindows[i].get() == window) {
120944d93782SGreg Clayton         m_prev_active_window_idx = m_curr_active_window_idx;
121044d93782SGreg Clayton         ::top_panel(window->m_panel);
121144d93782SGreg Clayton         m_curr_active_window_idx = i;
121244d93782SGreg Clayton         return true;
121344d93782SGreg Clayton       }
121444d93782SGreg Clayton     }
121544d93782SGreg Clayton     return false;
121644d93782SGreg Clayton   }
121744d93782SGreg Clayton 
1218b9c1b51eSKate Stone   WindowSP GetActiveWindow() {
1219b9c1b51eSKate Stone     if (!m_subwindows.empty()) {
1220b9c1b51eSKate Stone       if (m_curr_active_window_idx >= m_subwindows.size()) {
1221b9c1b51eSKate Stone         if (m_prev_active_window_idx < m_subwindows.size()) {
122244d93782SGreg Clayton           m_curr_active_window_idx = m_prev_active_window_idx;
122344d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
1224b9c1b51eSKate Stone         } else if (IsActive()) {
122544d93782SGreg Clayton           m_prev_active_window_idx = UINT32_MAX;
122644d93782SGreg Clayton           m_curr_active_window_idx = UINT32_MAX;
122744d93782SGreg Clayton 
122844d93782SGreg Clayton           // Find first window that wants to be active if this window is active
122944d93782SGreg Clayton           const size_t num_subwindows = m_subwindows.size();
1230b9c1b51eSKate Stone           for (size_t i = 0; i < num_subwindows; ++i) {
1231b9c1b51eSKate Stone             if (m_subwindows[i]->GetCanBeActive()) {
123244d93782SGreg Clayton               m_curr_active_window_idx = i;
123344d93782SGreg Clayton               break;
123444d93782SGreg Clayton             }
123544d93782SGreg Clayton           }
123644d93782SGreg Clayton         }
123744d93782SGreg Clayton       }
123844d93782SGreg Clayton 
123944d93782SGreg Clayton       if (m_curr_active_window_idx < m_subwindows.size())
124044d93782SGreg Clayton         return m_subwindows[m_curr_active_window_idx];
124144d93782SGreg Clayton     }
124244d93782SGreg Clayton     return WindowSP();
124344d93782SGreg Clayton   }
124444d93782SGreg Clayton 
1245b9c1b51eSKate Stone   bool GetCanBeActive() const { return m_can_activate; }
124644d93782SGreg Clayton 
1247b9c1b51eSKate Stone   void SetCanBeActive(bool b) { m_can_activate = b; }
124844d93782SGreg Clayton 
1249b9c1b51eSKate Stone   const WindowDelegateSP &GetDelegate() const { return m_delegate_sp; }
125044d93782SGreg Clayton 
1251b9c1b51eSKate Stone   void SetDelegate(const WindowDelegateSP &delegate_sp) {
125244d93782SGreg Clayton     m_delegate_sp = delegate_sp;
125344d93782SGreg Clayton   }
125444d93782SGreg Clayton 
1255b9c1b51eSKate Stone   Window *GetParent() const { return m_parent; }
125644d93782SGreg Clayton 
1257b9c1b51eSKate Stone   bool IsActive() const {
125844d93782SGreg Clayton     if (m_parent)
125944d93782SGreg Clayton       return m_parent->GetActiveWindow().get() == this;
126044d93782SGreg Clayton     else
126144d93782SGreg Clayton       return true; // Top level window is always active
126244d93782SGreg Clayton   }
126344d93782SGreg Clayton 
1264b9c1b51eSKate Stone   void SelectNextWindowAsActive() {
126544d93782SGreg Clayton     // Move active focus to next window
126644d93782SGreg Clayton     const size_t num_subwindows = m_subwindows.size();
1267b9c1b51eSKate Stone     if (m_curr_active_window_idx == UINT32_MAX) {
126844d93782SGreg Clayton       uint32_t idx = 0;
1269b9c1b51eSKate Stone       for (auto subwindow_sp : m_subwindows) {
1270b9c1b51eSKate Stone         if (subwindow_sp->GetCanBeActive()) {
127144d93782SGreg Clayton           m_curr_active_window_idx = idx;
127244d93782SGreg Clayton           break;
127344d93782SGreg Clayton         }
127444d93782SGreg Clayton         ++idx;
127544d93782SGreg Clayton       }
1276b9c1b51eSKate Stone     } else if (m_curr_active_window_idx + 1 < num_subwindows) {
127744d93782SGreg Clayton       bool handled = false;
127844d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1279b9c1b51eSKate Stone       for (size_t idx = m_curr_active_window_idx + 1; idx < num_subwindows;
1280b9c1b51eSKate Stone            ++idx) {
1281b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
128244d93782SGreg Clayton           m_curr_active_window_idx = idx;
128344d93782SGreg Clayton           handled = true;
128444d93782SGreg Clayton           break;
128544d93782SGreg Clayton         }
128644d93782SGreg Clayton       }
1287b9c1b51eSKate Stone       if (!handled) {
1288b9c1b51eSKate Stone         for (size_t idx = 0; idx <= m_prev_active_window_idx; ++idx) {
1289b9c1b51eSKate Stone           if (m_subwindows[idx]->GetCanBeActive()) {
129044d93782SGreg Clayton             m_curr_active_window_idx = idx;
129144d93782SGreg Clayton             break;
129244d93782SGreg Clayton           }
129344d93782SGreg Clayton         }
129444d93782SGreg Clayton       }
1295b9c1b51eSKate Stone     } else {
129644d93782SGreg Clayton       m_prev_active_window_idx = m_curr_active_window_idx;
1297b9c1b51eSKate Stone       for (size_t idx = 0; idx < num_subwindows; ++idx) {
1298b9c1b51eSKate Stone         if (m_subwindows[idx]->GetCanBeActive()) {
129944d93782SGreg Clayton           m_curr_active_window_idx = idx;
130044d93782SGreg Clayton           break;
130144d93782SGreg Clayton         }
130244d93782SGreg Clayton       }
130344d93782SGreg Clayton     }
130444d93782SGreg Clayton   }
130544d93782SGreg Clayton 
1306b9c1b51eSKate Stone   const char *GetName() const { return m_name.c_str(); }
1307315b6884SEugene Zelenko 
130844d93782SGreg Clayton protected:
130944d93782SGreg Clayton   std::string m_name;
131044d93782SGreg Clayton   WINDOW *m_window;
131144d93782SGreg Clayton   PANEL *m_panel;
131244d93782SGreg Clayton   Window *m_parent;
131344d93782SGreg Clayton   Windows m_subwindows;
131444d93782SGreg Clayton   WindowDelegateSP m_delegate_sp;
131544d93782SGreg Clayton   uint32_t m_curr_active_window_idx;
131644d93782SGreg Clayton   uint32_t m_prev_active_window_idx;
131744d93782SGreg Clayton   bool m_delete;
131844d93782SGreg Clayton   bool m_needs_update;
131944d93782SGreg Clayton   bool m_can_activate;
132044d93782SGreg Clayton   bool m_is_subwin;
132144d93782SGreg Clayton 
132244d93782SGreg Clayton private:
132344d93782SGreg Clayton   DISALLOW_COPY_AND_ASSIGN(Window);
132444d93782SGreg Clayton };
132544d93782SGreg Clayton 
1326b9c1b51eSKate Stone class MenuDelegate {
132744d93782SGreg Clayton public:
1328315b6884SEugene Zelenko   virtual ~MenuDelegate() = default;
132944d93782SGreg Clayton 
1330b9c1b51eSKate Stone   virtual MenuActionResult MenuDelegateAction(Menu &menu) = 0;
133144d93782SGreg Clayton };
133244d93782SGreg Clayton 
1333b9c1b51eSKate Stone class Menu : public WindowDelegate {
133444d93782SGreg Clayton public:
1335b9c1b51eSKate Stone   enum class Type { Invalid, Bar, Item, Separator };
133644d93782SGreg Clayton 
133744d93782SGreg Clayton   // Menubar or separator constructor
133844d93782SGreg Clayton   Menu(Type type);
133944d93782SGreg Clayton 
134044d93782SGreg Clayton   // Menuitem constructor
1341b9c1b51eSKate Stone   Menu(const char *name, const char *key_name, int key_value,
134244d93782SGreg Clayton        uint64_t identifier);
134344d93782SGreg Clayton 
1344315b6884SEugene Zelenko   ~Menu() override = default;
134544d93782SGreg Clayton 
1346b9c1b51eSKate Stone   const MenuDelegateSP &GetDelegate() const { return m_delegate_sp; }
134744d93782SGreg Clayton 
1348b9c1b51eSKate Stone   void SetDelegate(const MenuDelegateSP &delegate_sp) {
134944d93782SGreg Clayton     m_delegate_sp = delegate_sp;
135044d93782SGreg Clayton   }
135144d93782SGreg Clayton 
1352b9c1b51eSKate Stone   void RecalculateNameLengths();
135344d93782SGreg Clayton 
1354b9c1b51eSKate Stone   void AddSubmenu(const MenuSP &menu_sp);
135544d93782SGreg Clayton 
1356b9c1b51eSKate Stone   int DrawAndRunMenu(Window &window);
135744d93782SGreg Clayton 
1358b9c1b51eSKate Stone   void DrawMenuTitle(Window &window, bool highlight);
135944d93782SGreg Clayton 
1360b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override;
136144d93782SGreg Clayton 
1362b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
136344d93782SGreg Clayton 
1364b9c1b51eSKate Stone   MenuActionResult ActionPrivate(Menu &menu) {
136544d93782SGreg Clayton     MenuActionResult result = MenuActionResult::NotHandled;
1366b9c1b51eSKate Stone     if (m_delegate_sp) {
136744d93782SGreg Clayton       result = m_delegate_sp->MenuDelegateAction(menu);
136844d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
136944d93782SGreg Clayton         return result;
1370b9c1b51eSKate Stone     } else if (m_parent) {
137144d93782SGreg Clayton       result = m_parent->ActionPrivate(menu);
137244d93782SGreg Clayton       if (result != MenuActionResult::NotHandled)
137344d93782SGreg Clayton         return result;
137444d93782SGreg Clayton     }
137544d93782SGreg Clayton     return m_canned_result;
137644d93782SGreg Clayton   }
137744d93782SGreg Clayton 
1378b9c1b51eSKate Stone   MenuActionResult Action() {
137905097246SAdrian Prantl     // Call the recursive action so it can try to handle it with the menu
138005097246SAdrian Prantl     // delegate, and if not, try our parent menu
138144d93782SGreg Clayton     return ActionPrivate(*this);
138244d93782SGreg Clayton   }
138344d93782SGreg Clayton 
1384b9c1b51eSKate Stone   void SetCannedResult(MenuActionResult result) { m_canned_result = result; }
138544d93782SGreg Clayton 
1386b9c1b51eSKate Stone   Menus &GetSubmenus() { return m_submenus; }
138744d93782SGreg Clayton 
1388b9c1b51eSKate Stone   const Menus &GetSubmenus() const { return m_submenus; }
138944d93782SGreg Clayton 
1390b9c1b51eSKate Stone   int GetSelectedSubmenuIndex() const { return m_selected; }
139144d93782SGreg Clayton 
1392b9c1b51eSKate Stone   void SetSelectedSubmenuIndex(int idx) { m_selected = idx; }
139344d93782SGreg Clayton 
1394b9c1b51eSKate Stone   Type GetType() const { return m_type; }
139544d93782SGreg Clayton 
1396b9c1b51eSKate Stone   int GetStartingColumn() const { return m_start_col; }
139744d93782SGreg Clayton 
1398b9c1b51eSKate Stone   void SetStartingColumn(int col) { m_start_col = col; }
139944d93782SGreg Clayton 
1400b9c1b51eSKate Stone   int GetKeyValue() const { return m_key_value; }
140144d93782SGreg Clayton 
1402b9c1b51eSKate Stone   void SetKeyValue(int key_value) { m_key_value = key_value; }
140344d93782SGreg Clayton 
1404b9c1b51eSKate Stone   std::string &GetName() { return m_name; }
140544d93782SGreg Clayton 
1406b9c1b51eSKate Stone   std::string &GetKeyName() { return m_key_name; }
140744d93782SGreg Clayton 
1408b9c1b51eSKate Stone   int GetDrawWidth() const {
140944d93782SGreg Clayton     return m_max_submenu_name_length + m_max_submenu_key_name_length + 8;
141044d93782SGreg Clayton   }
141144d93782SGreg Clayton 
1412b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
141344d93782SGreg Clayton 
1414b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
141544d93782SGreg Clayton 
141644d93782SGreg Clayton protected:
141744d93782SGreg Clayton   std::string m_name;
141844d93782SGreg Clayton   std::string m_key_name;
141944d93782SGreg Clayton   uint64_t m_identifier;
142044d93782SGreg Clayton   Type m_type;
142144d93782SGreg Clayton   int m_key_value;
142244d93782SGreg Clayton   int m_start_col;
142344d93782SGreg Clayton   int m_max_submenu_name_length;
142444d93782SGreg Clayton   int m_max_submenu_key_name_length;
142544d93782SGreg Clayton   int m_selected;
142644d93782SGreg Clayton   Menu *m_parent;
142744d93782SGreg Clayton   Menus m_submenus;
142844d93782SGreg Clayton   WindowSP m_menu_window_sp;
142944d93782SGreg Clayton   MenuActionResult m_canned_result;
143044d93782SGreg Clayton   MenuDelegateSP m_delegate_sp;
143144d93782SGreg Clayton };
143244d93782SGreg Clayton 
143344d93782SGreg Clayton // Menubar or separator constructor
1434b9c1b51eSKate Stone Menu::Menu(Type type)
1435b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(0), m_type(type), m_key_value(0),
1436b9c1b51eSKate Stone       m_start_col(0), m_max_submenu_name_length(0),
1437b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1438b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1439b9c1b51eSKate Stone       m_delegate_sp() {}
144044d93782SGreg Clayton 
144144d93782SGreg Clayton // Menuitem constructor
1442b9c1b51eSKate Stone Menu::Menu(const char *name, const char *key_name, int key_value,
1443b9c1b51eSKate Stone            uint64_t identifier)
1444b9c1b51eSKate Stone     : m_name(), m_key_name(), m_identifier(identifier), m_type(Type::Invalid),
1445b9c1b51eSKate Stone       m_key_value(key_value), m_start_col(0), m_max_submenu_name_length(0),
1446b9c1b51eSKate Stone       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
1447b9c1b51eSKate Stone       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
1448b9c1b51eSKate Stone       m_delegate_sp() {
1449b9c1b51eSKate Stone   if (name && name[0]) {
145044d93782SGreg Clayton     m_name = name;
145144d93782SGreg Clayton     m_type = Type::Item;
145244d93782SGreg Clayton     if (key_name && key_name[0])
145344d93782SGreg Clayton       m_key_name = key_name;
1454b9c1b51eSKate Stone   } else {
145544d93782SGreg Clayton     m_type = Type::Separator;
145644d93782SGreg Clayton   }
145744d93782SGreg Clayton }
145844d93782SGreg Clayton 
1459b9c1b51eSKate Stone void Menu::RecalculateNameLengths() {
146044d93782SGreg Clayton   m_max_submenu_name_length = 0;
146144d93782SGreg Clayton   m_max_submenu_key_name_length = 0;
146244d93782SGreg Clayton   Menus &submenus = GetSubmenus();
146344d93782SGreg Clayton   const size_t num_submenus = submenus.size();
1464b9c1b51eSKate Stone   for (size_t i = 0; i < num_submenus; ++i) {
146544d93782SGreg Clayton     Menu *submenu = submenus[i].get();
14663985c8c6SSaleem Abdulrasool     if (static_cast<size_t>(m_max_submenu_name_length) < submenu->m_name.size())
146744d93782SGreg Clayton       m_max_submenu_name_length = submenu->m_name.size();
1468b9c1b51eSKate Stone     if (static_cast<size_t>(m_max_submenu_key_name_length) <
1469b9c1b51eSKate Stone         submenu->m_key_name.size())
147044d93782SGreg Clayton       m_max_submenu_key_name_length = submenu->m_key_name.size();
147144d93782SGreg Clayton   }
147244d93782SGreg Clayton }
147344d93782SGreg Clayton 
1474b9c1b51eSKate Stone void Menu::AddSubmenu(const MenuSP &menu_sp) {
147544d93782SGreg Clayton   menu_sp->m_parent = this;
14763985c8c6SSaleem Abdulrasool   if (static_cast<size_t>(m_max_submenu_name_length) < menu_sp->m_name.size())
147744d93782SGreg Clayton     m_max_submenu_name_length = menu_sp->m_name.size();
1478b9c1b51eSKate Stone   if (static_cast<size_t>(m_max_submenu_key_name_length) <
1479b9c1b51eSKate Stone       menu_sp->m_key_name.size())
148044d93782SGreg Clayton     m_max_submenu_key_name_length = menu_sp->m_key_name.size();
148144d93782SGreg Clayton   m_submenus.push_back(menu_sp);
148244d93782SGreg Clayton }
148344d93782SGreg Clayton 
1484b9c1b51eSKate Stone void Menu::DrawMenuTitle(Window &window, bool highlight) {
1485b9c1b51eSKate Stone   if (m_type == Type::Separator) {
148644d93782SGreg Clayton     window.MoveCursor(0, window.GetCursorY());
148744d93782SGreg Clayton     window.PutChar(ACS_LTEE);
148844d93782SGreg Clayton     int width = window.GetWidth();
1489b9c1b51eSKate Stone     if (width > 2) {
149044d93782SGreg Clayton       width -= 2;
14913985c8c6SSaleem Abdulrasool       for (int i = 0; i < width; ++i)
149244d93782SGreg Clayton         window.PutChar(ACS_HLINE);
149344d93782SGreg Clayton     }
149444d93782SGreg Clayton     window.PutChar(ACS_RTEE);
1495b9c1b51eSKate Stone   } else {
149644d93782SGreg Clayton     const int shortcut_key = m_key_value;
149744d93782SGreg Clayton     bool underlined_shortcut = false;
149844d93782SGreg Clayton     const attr_t hilgight_attr = A_REVERSE;
149944d93782SGreg Clayton     if (highlight)
150044d93782SGreg Clayton       window.AttributeOn(hilgight_attr);
1501b9c1b51eSKate Stone     if (isprint(shortcut_key)) {
150244d93782SGreg Clayton       size_t lower_pos = m_name.find(tolower(shortcut_key));
150344d93782SGreg Clayton       size_t upper_pos = m_name.find(toupper(shortcut_key));
150444d93782SGreg Clayton       const char *name = m_name.c_str();
150544d93782SGreg Clayton       size_t pos = std::min<size_t>(lower_pos, upper_pos);
1506b9c1b51eSKate Stone       if (pos != std::string::npos) {
150744d93782SGreg Clayton         underlined_shortcut = true;
1508b9c1b51eSKate Stone         if (pos > 0) {
150944d93782SGreg Clayton           window.PutCString(name, pos);
151044d93782SGreg Clayton           name += pos;
151144d93782SGreg Clayton         }
151244d93782SGreg Clayton         const attr_t shortcut_attr = A_UNDERLINE | A_BOLD;
151344d93782SGreg Clayton         window.AttributeOn(shortcut_attr);
151444d93782SGreg Clayton         window.PutChar(name[0]);
151544d93782SGreg Clayton         window.AttributeOff(shortcut_attr);
151644d93782SGreg Clayton         name++;
151744d93782SGreg Clayton         if (name[0])
151844d93782SGreg Clayton           window.PutCString(name);
151944d93782SGreg Clayton       }
152044d93782SGreg Clayton     }
152144d93782SGreg Clayton 
1522b9c1b51eSKate Stone     if (!underlined_shortcut) {
152344d93782SGreg Clayton       window.PutCString(m_name.c_str());
152444d93782SGreg Clayton     }
152544d93782SGreg Clayton 
152644d93782SGreg Clayton     if (highlight)
152744d93782SGreg Clayton       window.AttributeOff(hilgight_attr);
152844d93782SGreg Clayton 
1529b9c1b51eSKate Stone     if (m_key_name.empty()) {
1530b9c1b51eSKate Stone       if (!underlined_shortcut && isprint(m_key_value)) {
153144d93782SGreg Clayton         window.AttributeOn(COLOR_PAIR(3));
153244d93782SGreg Clayton         window.Printf(" (%c)", m_key_value);
153344d93782SGreg Clayton         window.AttributeOff(COLOR_PAIR(3));
153444d93782SGreg Clayton       }
1535b9c1b51eSKate Stone     } else {
153644d93782SGreg Clayton       window.AttributeOn(COLOR_PAIR(3));
153744d93782SGreg Clayton       window.Printf(" (%s)", m_key_name.c_str());
153844d93782SGreg Clayton       window.AttributeOff(COLOR_PAIR(3));
153944d93782SGreg Clayton     }
154044d93782SGreg Clayton   }
154144d93782SGreg Clayton }
154244d93782SGreg Clayton 
1543b9c1b51eSKate Stone bool Menu::WindowDelegateDraw(Window &window, bool force) {
154444d93782SGreg Clayton   Menus &submenus = GetSubmenus();
154544d93782SGreg Clayton   const size_t num_submenus = submenus.size();
154644d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
154744d93782SGreg Clayton   Menu::Type menu_type = GetType();
1548b9c1b51eSKate Stone   switch (menu_type) {
1549b9c1b51eSKate Stone   case Menu::Type::Bar: {
155044d93782SGreg Clayton     window.SetBackground(2);
155144d93782SGreg Clayton     window.MoveCursor(0, 0);
1552b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
155344d93782SGreg Clayton       Menu *menu = submenus[i].get();
155444d93782SGreg Clayton       if (i > 0)
155544d93782SGreg Clayton         window.PutChar(' ');
155644d93782SGreg Clayton       menu->SetStartingColumn(window.GetCursorX());
155744d93782SGreg Clayton       window.PutCString("| ");
155844d93782SGreg Clayton       menu->DrawMenuTitle(window, false);
155944d93782SGreg Clayton     }
156044d93782SGreg Clayton     window.PutCString(" |");
156144d93782SGreg Clayton     window.DeferredRefresh();
1562b9c1b51eSKate Stone   } break;
156344d93782SGreg Clayton 
1564b9c1b51eSKate Stone   case Menu::Type::Item: {
156544d93782SGreg Clayton     int y = 1;
156644d93782SGreg Clayton     int x = 3;
156744d93782SGreg Clayton     // Draw the menu
156844d93782SGreg Clayton     int cursor_x = 0;
156944d93782SGreg Clayton     int cursor_y = 0;
157044d93782SGreg Clayton     window.Erase();
157144d93782SGreg Clayton     window.SetBackground(2);
157244d93782SGreg Clayton     window.Box();
1573b9c1b51eSKate Stone     for (size_t i = 0; i < num_submenus; ++i) {
1574b9c1b51eSKate Stone       const bool is_selected = (i == static_cast<size_t>(selected_idx));
157544d93782SGreg Clayton       window.MoveCursor(x, y + i);
1576b9c1b51eSKate Stone       if (is_selected) {
157744d93782SGreg Clayton         // Remember where we want the cursor to be
157844d93782SGreg Clayton         cursor_x = x - 1;
157944d93782SGreg Clayton         cursor_y = y + i;
158044d93782SGreg Clayton       }
158144d93782SGreg Clayton       submenus[i]->DrawMenuTitle(window, is_selected);
158244d93782SGreg Clayton     }
158344d93782SGreg Clayton     window.MoveCursor(cursor_x, cursor_y);
158444d93782SGreg Clayton     window.DeferredRefresh();
1585b9c1b51eSKate Stone   } break;
158644d93782SGreg Clayton 
158744d93782SGreg Clayton   default:
158844d93782SGreg Clayton   case Menu::Type::Separator:
158944d93782SGreg Clayton     break;
159044d93782SGreg Clayton   }
159144d93782SGreg Clayton   return true; // Drawing handled...
159244d93782SGreg Clayton }
159344d93782SGreg Clayton 
1594b9c1b51eSKate Stone HandleCharResult Menu::WindowDelegateHandleChar(Window &window, int key) {
159544d93782SGreg Clayton   HandleCharResult result = eKeyNotHandled;
159644d93782SGreg Clayton 
159744d93782SGreg Clayton   Menus &submenus = GetSubmenus();
159844d93782SGreg Clayton   const size_t num_submenus = submenus.size();
159944d93782SGreg Clayton   const int selected_idx = GetSelectedSubmenuIndex();
160044d93782SGreg Clayton   Menu::Type menu_type = GetType();
1601b9c1b51eSKate Stone   if (menu_type == Menu::Type::Bar) {
160244d93782SGreg Clayton     MenuSP run_menu_sp;
1603b9c1b51eSKate Stone     switch (key) {
160444d93782SGreg Clayton     case KEY_DOWN:
160544d93782SGreg Clayton     case KEY_UP:
160644d93782SGreg Clayton       // Show last menu or first menu
16073985c8c6SSaleem Abdulrasool       if (selected_idx < static_cast<int>(num_submenus))
160844d93782SGreg Clayton         run_menu_sp = submenus[selected_idx];
160944d93782SGreg Clayton       else if (!submenus.empty())
161044d93782SGreg Clayton         run_menu_sp = submenus.front();
161144d93782SGreg Clayton       result = eKeyHandled;
161244d93782SGreg Clayton       break;
161344d93782SGreg Clayton 
161444d93782SGreg Clayton     case KEY_RIGHT:
161544d93782SGreg Clayton       ++m_selected;
16163985c8c6SSaleem Abdulrasool       if (m_selected >= static_cast<int>(num_submenus))
161744d93782SGreg Clayton         m_selected = 0;
16183985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
161944d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
162044d93782SGreg Clayton       else if (!submenus.empty())
162144d93782SGreg Clayton         run_menu_sp = submenus.front();
162244d93782SGreg Clayton       result = eKeyHandled;
162344d93782SGreg Clayton       break;
162444d93782SGreg Clayton 
162544d93782SGreg Clayton     case KEY_LEFT:
162644d93782SGreg Clayton       --m_selected;
162744d93782SGreg Clayton       if (m_selected < 0)
162844d93782SGreg Clayton         m_selected = num_submenus - 1;
16293985c8c6SSaleem Abdulrasool       if (m_selected < static_cast<int>(num_submenus))
163044d93782SGreg Clayton         run_menu_sp = submenus[m_selected];
163144d93782SGreg Clayton       else if (!submenus.empty())
163244d93782SGreg Clayton         run_menu_sp = submenus.front();
163344d93782SGreg Clayton       result = eKeyHandled;
163444d93782SGreg Clayton       break;
163544d93782SGreg Clayton 
163644d93782SGreg Clayton     default:
1637b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
1638b9c1b51eSKate Stone         if (submenus[i]->GetKeyValue() == key) {
163944d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
164044d93782SGreg Clayton           run_menu_sp = submenus[i];
164144d93782SGreg Clayton           result = eKeyHandled;
164244d93782SGreg Clayton           break;
164344d93782SGreg Clayton         }
164444d93782SGreg Clayton       }
164544d93782SGreg Clayton       break;
164644d93782SGreg Clayton     }
164744d93782SGreg Clayton 
1648b9c1b51eSKate Stone     if (run_menu_sp) {
164905097246SAdrian Prantl       // Run the action on this menu in case we need to populate the menu with
165005097246SAdrian Prantl       // dynamic content and also in case check marks, and any other menu
165105097246SAdrian Prantl       // decorations need to be calculated
165244d93782SGreg Clayton       if (run_menu_sp->Action() == MenuActionResult::Quit)
165344d93782SGreg Clayton         return eQuitApplication;
165444d93782SGreg Clayton 
165544d93782SGreg Clayton       Rect menu_bounds;
165644d93782SGreg Clayton       menu_bounds.origin.x = run_menu_sp->GetStartingColumn();
165744d93782SGreg Clayton       menu_bounds.origin.y = 1;
165844d93782SGreg Clayton       menu_bounds.size.width = run_menu_sp->GetDrawWidth();
165944d93782SGreg Clayton       menu_bounds.size.height = run_menu_sp->GetSubmenus().size() + 2;
166044d93782SGreg Clayton       if (m_menu_window_sp)
166144d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(m_menu_window_sp.get());
166244d93782SGreg Clayton 
1663b9c1b51eSKate Stone       m_menu_window_sp = window.GetParent()->CreateSubWindow(
1664b9c1b51eSKate Stone           run_menu_sp->GetName().c_str(), menu_bounds, true);
166544d93782SGreg Clayton       m_menu_window_sp->SetDelegate(run_menu_sp);
166644d93782SGreg Clayton     }
1667b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Item) {
1668b9c1b51eSKate Stone     switch (key) {
166944d93782SGreg Clayton     case KEY_DOWN:
1670b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
167144d93782SGreg Clayton         const int start_select = m_selected;
1672b9c1b51eSKate Stone         while (++m_selected != start_select) {
16733985c8c6SSaleem Abdulrasool           if (static_cast<size_t>(m_selected) >= num_submenus)
167444d93782SGreg Clayton             m_selected = 0;
167544d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
167644d93782SGreg Clayton             continue;
167744d93782SGreg Clayton           else
167844d93782SGreg Clayton             break;
167944d93782SGreg Clayton         }
168044d93782SGreg Clayton         return eKeyHandled;
168144d93782SGreg Clayton       }
168244d93782SGreg Clayton       break;
168344d93782SGreg Clayton 
168444d93782SGreg Clayton     case KEY_UP:
1685b9c1b51eSKate Stone       if (m_submenus.size() > 1) {
168644d93782SGreg Clayton         const int start_select = m_selected;
1687b9c1b51eSKate Stone         while (--m_selected != start_select) {
16883985c8c6SSaleem Abdulrasool           if (m_selected < static_cast<int>(0))
168944d93782SGreg Clayton             m_selected = num_submenus - 1;
169044d93782SGreg Clayton           if (m_submenus[m_selected]->GetType() == Type::Separator)
169144d93782SGreg Clayton             continue;
169244d93782SGreg Clayton           else
169344d93782SGreg Clayton             break;
169444d93782SGreg Clayton         }
169544d93782SGreg Clayton         return eKeyHandled;
169644d93782SGreg Clayton       }
169744d93782SGreg Clayton       break;
169844d93782SGreg Clayton 
169944d93782SGreg Clayton     case KEY_RETURN:
1700b9c1b51eSKate Stone       if (static_cast<size_t>(selected_idx) < num_submenus) {
170144d93782SGreg Clayton         if (submenus[selected_idx]->Action() == MenuActionResult::Quit)
170244d93782SGreg Clayton           return eQuitApplication;
170344d93782SGreg Clayton         window.GetParent()->RemoveSubWindow(&window);
170444d93782SGreg Clayton         return eKeyHandled;
170544d93782SGreg Clayton       }
170644d93782SGreg Clayton       break;
170744d93782SGreg Clayton 
1708b9c1b51eSKate Stone     case KEY_ESCAPE: // Beware: pressing escape key has 1 to 2 second delay in
1709b9c1b51eSKate Stone                      // case other chars are entered for escaped sequences
171044d93782SGreg Clayton       window.GetParent()->RemoveSubWindow(&window);
171144d93782SGreg Clayton       return eKeyHandled;
171244d93782SGreg Clayton 
171344d93782SGreg Clayton     default:
1714b9c1b51eSKate Stone       for (size_t i = 0; i < num_submenus; ++i) {
171544d93782SGreg Clayton         Menu *menu = submenus[i].get();
1716b9c1b51eSKate Stone         if (menu->GetKeyValue() == key) {
171744d93782SGreg Clayton           SetSelectedSubmenuIndex(i);
171844d93782SGreg Clayton           window.GetParent()->RemoveSubWindow(&window);
171944d93782SGreg Clayton           if (menu->Action() == MenuActionResult::Quit)
172044d93782SGreg Clayton             return eQuitApplication;
172144d93782SGreg Clayton           return eKeyHandled;
172244d93782SGreg Clayton         }
172344d93782SGreg Clayton       }
172444d93782SGreg Clayton       break;
172544d93782SGreg Clayton     }
1726b9c1b51eSKate Stone   } else if (menu_type == Menu::Type::Separator) {
172744d93782SGreg Clayton   }
172844d93782SGreg Clayton   return result;
172944d93782SGreg Clayton }
173044d93782SGreg Clayton 
1731b9c1b51eSKate Stone class Application {
173244d93782SGreg Clayton public:
1733b9c1b51eSKate Stone   Application(FILE *in, FILE *out)
1734b9c1b51eSKate Stone       : m_window_sp(), m_screen(nullptr), m_in(in), m_out(out) {}
173544d93782SGreg Clayton 
1736b9c1b51eSKate Stone   ~Application() {
173744d93782SGreg Clayton     m_window_delegates.clear();
173844d93782SGreg Clayton     m_window_sp.reset();
1739b9c1b51eSKate Stone     if (m_screen) {
174044d93782SGreg Clayton       ::delscreen(m_screen);
1741c5dac77aSEugene Zelenko       m_screen = nullptr;
174244d93782SGreg Clayton     }
174344d93782SGreg Clayton   }
174444d93782SGreg Clayton 
1745b9c1b51eSKate Stone   void Initialize() {
174644d93782SGreg Clayton     ::setlocale(LC_ALL, "");
174744d93782SGreg Clayton     ::setlocale(LC_CTYPE, "");
1748c5dac77aSEugene Zelenko     m_screen = ::newterm(nullptr, m_out, m_in);
174944d93782SGreg Clayton     ::start_color();
175044d93782SGreg Clayton     ::curs_set(0);
175144d93782SGreg Clayton     ::noecho();
175244d93782SGreg Clayton     ::keypad(stdscr, TRUE);
175344d93782SGreg Clayton   }
175444d93782SGreg Clayton 
1755b9c1b51eSKate Stone   void Terminate() { ::endwin(); }
175644d93782SGreg Clayton 
1757b9c1b51eSKate Stone   void Run(Debugger &debugger) {
175844d93782SGreg Clayton     bool done = false;
175944d93782SGreg Clayton     int delay_in_tenths_of_a_second = 1;
176044d93782SGreg Clayton 
176105097246SAdrian Prantl     // Alas the threading model in curses is a bit lame so we need to resort to
176205097246SAdrian Prantl     // polling every 0.5 seconds. We could poll for stdin ourselves and then
176305097246SAdrian Prantl     // pass the keys down but then we need to translate all of the escape
176405097246SAdrian Prantl     // sequences ourselves. So we resort to polling for input because we need
176505097246SAdrian Prantl     // to receive async process events while in this loop.
176644d93782SGreg Clayton 
1767b9c1b51eSKate Stone     halfdelay(delay_in_tenths_of_a_second); // Poll using some number of tenths
1768b9c1b51eSKate Stone                                             // of seconds seconds when calling
1769b9c1b51eSKate Stone                                             // Window::GetChar()
177044d93782SGreg Clayton 
1771b9c1b51eSKate Stone     ListenerSP listener_sp(
1772b9c1b51eSKate Stone         Listener::MakeListener("lldb.IOHandler.curses.Application"));
177344d93782SGreg Clayton     ConstString broadcaster_class_target(Target::GetStaticBroadcasterClass());
177444d93782SGreg Clayton     ConstString broadcaster_class_process(Process::GetStaticBroadcasterClass());
177544d93782SGreg Clayton     ConstString broadcaster_class_thread(Thread::GetStaticBroadcasterClass());
177644d93782SGreg Clayton     debugger.EnableForwardEvents(listener_sp);
177744d93782SGreg Clayton 
177844d93782SGreg Clayton     bool update = true;
177944d93782SGreg Clayton #if defined(__APPLE__)
178044d93782SGreg Clayton     std::deque<int> escape_chars;
178144d93782SGreg Clayton #endif
178244d93782SGreg Clayton 
1783b9c1b51eSKate Stone     while (!done) {
1784b9c1b51eSKate Stone       if (update) {
178544d93782SGreg Clayton         m_window_sp->Draw(false);
178605097246SAdrian Prantl         // All windows should be calling Window::DeferredRefresh() instead of
178705097246SAdrian Prantl         // Window::Refresh() so we can do a single update and avoid any screen
178805097246SAdrian Prantl         // blinking
178944d93782SGreg Clayton         update_panels();
179044d93782SGreg Clayton 
1791b9c1b51eSKate Stone         // Cursor hiding isn't working on MacOSX, so hide it in the top left
1792b9c1b51eSKate Stone         // corner
179344d93782SGreg Clayton         m_window_sp->MoveCursor(0, 0);
179444d93782SGreg Clayton 
179544d93782SGreg Clayton         doupdate();
179644d93782SGreg Clayton         update = false;
179744d93782SGreg Clayton       }
179844d93782SGreg Clayton 
179944d93782SGreg Clayton #if defined(__APPLE__)
180005097246SAdrian Prantl       // Terminal.app doesn't map its function keys correctly, F1-F4 default
180105097246SAdrian Prantl       // to: \033OP, \033OQ, \033OR, \033OS, so lets take care of this here if
1802b9c1b51eSKate Stone       // possible
180344d93782SGreg Clayton       int ch;
180444d93782SGreg Clayton       if (escape_chars.empty())
180544d93782SGreg Clayton         ch = m_window_sp->GetChar();
1806b9c1b51eSKate Stone       else {
180744d93782SGreg Clayton         ch = escape_chars.front();
180844d93782SGreg Clayton         escape_chars.pop_front();
180944d93782SGreg Clayton       }
1810b9c1b51eSKate Stone       if (ch == KEY_ESCAPE) {
181144d93782SGreg Clayton         int ch2 = m_window_sp->GetChar();
1812b9c1b51eSKate Stone         if (ch2 == 'O') {
181344d93782SGreg Clayton           int ch3 = m_window_sp->GetChar();
1814b9c1b51eSKate Stone           switch (ch3) {
1815b9c1b51eSKate Stone           case 'P':
1816b9c1b51eSKate Stone             ch = KEY_F(1);
1817b9c1b51eSKate Stone             break;
1818b9c1b51eSKate Stone           case 'Q':
1819b9c1b51eSKate Stone             ch = KEY_F(2);
1820b9c1b51eSKate Stone             break;
1821b9c1b51eSKate Stone           case 'R':
1822b9c1b51eSKate Stone             ch = KEY_F(3);
1823b9c1b51eSKate Stone             break;
1824b9c1b51eSKate Stone           case 'S':
1825b9c1b51eSKate Stone             ch = KEY_F(4);
1826b9c1b51eSKate Stone             break;
182744d93782SGreg Clayton           default:
182844d93782SGreg Clayton             escape_chars.push_back(ch2);
182944d93782SGreg Clayton             if (ch3 != -1)
183044d93782SGreg Clayton               escape_chars.push_back(ch3);
183144d93782SGreg Clayton             break;
183244d93782SGreg Clayton           }
1833b9c1b51eSKate Stone         } else if (ch2 != -1)
183444d93782SGreg Clayton           escape_chars.push_back(ch2);
183544d93782SGreg Clayton       }
183644d93782SGreg Clayton #else
183744d93782SGreg Clayton       int ch = m_window_sp->GetChar();
183844d93782SGreg Clayton 
183944d93782SGreg Clayton #endif
1840b9c1b51eSKate Stone       if (ch == -1) {
1841b9c1b51eSKate Stone         if (feof(m_in) || ferror(m_in)) {
184244d93782SGreg Clayton           done = true;
1843b9c1b51eSKate Stone         } else {
184444d93782SGreg Clayton           // Just a timeout from using halfdelay(), check for events
184544d93782SGreg Clayton           EventSP event_sp;
1846b9c1b51eSKate Stone           while (listener_sp->PeekAtNextEvent()) {
1847d35031e1SPavel Labath             listener_sp->GetEvent(event_sp, std::chrono::seconds(0));
184844d93782SGreg Clayton 
1849b9c1b51eSKate Stone             if (event_sp) {
185044d93782SGreg Clayton               Broadcaster *broadcaster = event_sp->GetBroadcaster();
1851b9c1b51eSKate Stone               if (broadcaster) {
185244d93782SGreg Clayton                 // uint32_t event_type = event_sp->GetType();
1853b9c1b51eSKate Stone                 ConstString broadcaster_class(
1854b9c1b51eSKate Stone                     broadcaster->GetBroadcasterClass());
1855b9c1b51eSKate Stone                 if (broadcaster_class == broadcaster_class_process) {
1856b9c1b51eSKate Stone                   debugger.GetCommandInterpreter().UpdateExecutionContext(
1857b9c1b51eSKate Stone                       nullptr);
185844d93782SGreg Clayton                   update = true;
185944d93782SGreg Clayton                   continue; // Don't get any key, just update our view
186044d93782SGreg Clayton                 }
186144d93782SGreg Clayton               }
186244d93782SGreg Clayton             }
186344d93782SGreg Clayton           }
186444d93782SGreg Clayton         }
1865b9c1b51eSKate Stone       } else {
186644d93782SGreg Clayton         HandleCharResult key_result = m_window_sp->HandleChar(ch);
1867b9c1b51eSKate Stone         switch (key_result) {
186844d93782SGreg Clayton         case eKeyHandled:
1869c5dac77aSEugene Zelenko           debugger.GetCommandInterpreter().UpdateExecutionContext(nullptr);
187044d93782SGreg Clayton           update = true;
187144d93782SGreg Clayton           break;
187244d93782SGreg Clayton         case eKeyNotHandled:
187344d93782SGreg Clayton           break;
187444d93782SGreg Clayton         case eQuitApplication:
187544d93782SGreg Clayton           done = true;
187644d93782SGreg Clayton           break;
187744d93782SGreg Clayton         }
187844d93782SGreg Clayton       }
187944d93782SGreg Clayton     }
188044d93782SGreg Clayton 
188144d93782SGreg Clayton     debugger.CancelForwardEvents(listener_sp);
188244d93782SGreg Clayton   }
188344d93782SGreg Clayton 
1884b9c1b51eSKate Stone   WindowSP &GetMainWindow() {
188544d93782SGreg Clayton     if (!m_window_sp)
1886*796ac80bSJonas Devlieghere       m_window_sp = std::make_shared<Window>("main", stdscr, false);
188744d93782SGreg Clayton     return m_window_sp;
188844d93782SGreg Clayton   }
188944d93782SGreg Clayton 
1890b9c1b51eSKate Stone   WindowDelegates &GetWindowDelegates() { return m_window_delegates; }
189144d93782SGreg Clayton 
189244d93782SGreg Clayton protected:
189344d93782SGreg Clayton   WindowSP m_window_sp;
189444d93782SGreg Clayton   WindowDelegates m_window_delegates;
189544d93782SGreg Clayton   SCREEN *m_screen;
189644d93782SGreg Clayton   FILE *m_in;
189744d93782SGreg Clayton   FILE *m_out;
189844d93782SGreg Clayton };
189944d93782SGreg Clayton 
190044d93782SGreg Clayton } // namespace curses
190144d93782SGreg Clayton 
190244d93782SGreg Clayton using namespace curses;
190344d93782SGreg Clayton 
1904b9c1b51eSKate Stone struct Row {
19058369b28dSGreg Clayton   ValueObjectManager value;
190644d93782SGreg Clayton   Row *parent;
19078369b28dSGreg Clayton   // The process stop ID when the children were calculated.
19088369b28dSGreg Clayton   uint32_t children_stop_id;
190944d93782SGreg Clayton   int row_idx;
191044d93782SGreg Clayton   int x;
191144d93782SGreg Clayton   int y;
191244d93782SGreg Clayton   bool might_have_children;
191344d93782SGreg Clayton   bool expanded;
191444d93782SGreg Clayton   bool calculated_children;
191544d93782SGreg Clayton   std::vector<Row> children;
191644d93782SGreg Clayton 
1917b9c1b51eSKate Stone   Row(const ValueObjectSP &v, Row *p)
19188369b28dSGreg Clayton       : value(v, lldb::eDynamicDontRunTarget, true), parent(p), row_idx(0),
19198369b28dSGreg Clayton         x(1), y(1), might_have_children(v ? v->MightHaveChildren() : false),
1920b9c1b51eSKate Stone         expanded(false), calculated_children(false), children() {}
192144d93782SGreg Clayton 
1922b9c1b51eSKate Stone   size_t GetDepth() const {
192344d93782SGreg Clayton     if (parent)
192444d93782SGreg Clayton       return 1 + parent->GetDepth();
192544d93782SGreg Clayton     return 0;
192644d93782SGreg Clayton   }
192744d93782SGreg Clayton 
1928b9c1b51eSKate Stone   void Expand() {
192944d93782SGreg Clayton     expanded = true;
19308369b28dSGreg Clayton   }
19318369b28dSGreg Clayton 
19328369b28dSGreg Clayton   std::vector<Row> &GetChildren() {
19338369b28dSGreg Clayton     ProcessSP process_sp = value.GetProcessSP();
19348369b28dSGreg Clayton     auto stop_id = process_sp->GetStopID();
19358369b28dSGreg Clayton     if (process_sp && stop_id != children_stop_id) {
19368369b28dSGreg Clayton       children_stop_id = stop_id;
19378369b28dSGreg Clayton       calculated_children = false;
19388369b28dSGreg Clayton     }
1939b9c1b51eSKate Stone     if (!calculated_children) {
19408369b28dSGreg Clayton       children.clear();
194144d93782SGreg Clayton       calculated_children = true;
19428369b28dSGreg Clayton       ValueObjectSP valobj = value.GetSP();
1943b9c1b51eSKate Stone       if (valobj) {
194444d93782SGreg Clayton         const size_t num_children = valobj->GetNumChildren();
1945b9c1b51eSKate Stone         for (size_t i = 0; i < num_children; ++i) {
194644d93782SGreg Clayton           children.push_back(Row(valobj->GetChildAtIndex(i, true), this));
194744d93782SGreg Clayton         }
194844d93782SGreg Clayton       }
194944d93782SGreg Clayton     }
19508369b28dSGreg Clayton     return children;
195144d93782SGreg Clayton   }
195244d93782SGreg Clayton 
19538369b28dSGreg Clayton   void Unexpand() {
19548369b28dSGreg Clayton     expanded = false;
19558369b28dSGreg Clayton     calculated_children = false;
19568369b28dSGreg Clayton     children.clear();
19578369b28dSGreg Clayton   }
195844d93782SGreg Clayton 
1959b9c1b51eSKate Stone   void DrawTree(Window &window) {
196044d93782SGreg Clayton     if (parent)
196144d93782SGreg Clayton       parent->DrawTreeForChild(window, this, 0);
196244d93782SGreg Clayton 
1963b9c1b51eSKate Stone     if (might_have_children) {
196405097246SAdrian Prantl       // It we can get UTF8 characters to work we should try to use the
196505097246SAdrian Prantl       // "symbol" UTF8 string below
196644d93782SGreg Clayton       //            const char *symbol = "";
196744d93782SGreg Clayton       //            if (row.expanded)
196844d93782SGreg Clayton       //                symbol = "\xe2\x96\xbd ";
196944d93782SGreg Clayton       //            else
197044d93782SGreg Clayton       //                symbol = "\xe2\x96\xb7 ";
197144d93782SGreg Clayton       //            window.PutCString (symbol);
197244d93782SGreg Clayton 
197305097246SAdrian Prantl       // The ACS_DARROW and ACS_RARROW don't look very nice they are just a 'v'
197405097246SAdrian Prantl       // or '>' character...
197544d93782SGreg Clayton       //            if (expanded)
197644d93782SGreg Clayton       //                window.PutChar (ACS_DARROW);
197744d93782SGreg Clayton       //            else
197844d93782SGreg Clayton       //                window.PutChar (ACS_RARROW);
197905097246SAdrian Prantl       // Since we can't find any good looking right arrow/down arrow symbols,
198005097246SAdrian Prantl       // just use a diamond...
198144d93782SGreg Clayton       window.PutChar(ACS_DIAMOND);
198244d93782SGreg Clayton       window.PutChar(ACS_HLINE);
198344d93782SGreg Clayton     }
198444d93782SGreg Clayton   }
198544d93782SGreg Clayton 
1986b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, Row *child, uint32_t reverse_depth) {
198744d93782SGreg Clayton     if (parent)
198844d93782SGreg Clayton       parent->DrawTreeForChild(window, this, reverse_depth + 1);
198944d93782SGreg Clayton 
19908369b28dSGreg Clayton     if (&GetChildren().back() == child) {
199144d93782SGreg Clayton       // Last child
1992b9c1b51eSKate Stone       if (reverse_depth == 0) {
199344d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
199444d93782SGreg Clayton         window.PutChar(ACS_HLINE);
1995b9c1b51eSKate Stone       } else {
199644d93782SGreg Clayton         window.PutChar(' ');
199744d93782SGreg Clayton         window.PutChar(' ');
199844d93782SGreg Clayton       }
1999b9c1b51eSKate Stone     } else {
2000b9c1b51eSKate Stone       if (reverse_depth == 0) {
200144d93782SGreg Clayton         window.PutChar(ACS_LTEE);
200244d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2003b9c1b51eSKate Stone       } else {
200444d93782SGreg Clayton         window.PutChar(ACS_VLINE);
200544d93782SGreg Clayton         window.PutChar(' ');
200644d93782SGreg Clayton       }
200744d93782SGreg Clayton     }
200844d93782SGreg Clayton   }
200944d93782SGreg Clayton };
201044d93782SGreg Clayton 
2011b9c1b51eSKate Stone struct DisplayOptions {
201244d93782SGreg Clayton   bool show_types;
201344d93782SGreg Clayton };
201444d93782SGreg Clayton 
201544d93782SGreg Clayton class TreeItem;
201644d93782SGreg Clayton 
2017b9c1b51eSKate Stone class TreeDelegate {
201844d93782SGreg Clayton public:
2019c5dac77aSEugene Zelenko   TreeDelegate() = default;
2020315b6884SEugene Zelenko   virtual ~TreeDelegate() = default;
2021315b6884SEugene Zelenko 
202244d93782SGreg Clayton   virtual void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) = 0;
202344d93782SGreg Clayton   virtual void TreeDelegateGenerateChildren(TreeItem &item) = 0;
2024b9c1b51eSKate Stone   virtual bool TreeDelegateItemSelected(
2025b9c1b51eSKate Stone       TreeItem &item) = 0; // Return true if we need to update views
202644d93782SGreg Clayton };
2027315b6884SEugene Zelenko 
202844d93782SGreg Clayton typedef std::shared_ptr<TreeDelegate> TreeDelegateSP;
202944d93782SGreg Clayton 
2030b9c1b51eSKate Stone class TreeItem {
203144d93782SGreg Clayton public:
2032b9c1b51eSKate Stone   TreeItem(TreeItem *parent, TreeDelegate &delegate, bool might_have_children)
2033b9c1b51eSKate Stone       : m_parent(parent), m_delegate(delegate), m_user_data(nullptr),
2034b9c1b51eSKate Stone         m_identifier(0), m_row_idx(-1), m_children(),
2035b9c1b51eSKate Stone         m_might_have_children(might_have_children), m_is_expanded(false) {}
203644d93782SGreg Clayton 
2037b9c1b51eSKate Stone   TreeItem &operator=(const TreeItem &rhs) {
2038b9c1b51eSKate Stone     if (this != &rhs) {
203944d93782SGreg Clayton       m_parent = rhs.m_parent;
204044d93782SGreg Clayton       m_delegate = rhs.m_delegate;
2041ec990867SGreg Clayton       m_user_data = rhs.m_user_data;
204244d93782SGreg Clayton       m_identifier = rhs.m_identifier;
204344d93782SGreg Clayton       m_row_idx = rhs.m_row_idx;
204444d93782SGreg Clayton       m_children = rhs.m_children;
204544d93782SGreg Clayton       m_might_have_children = rhs.m_might_have_children;
204644d93782SGreg Clayton       m_is_expanded = rhs.m_is_expanded;
204744d93782SGreg Clayton     }
204844d93782SGreg Clayton     return *this;
204944d93782SGreg Clayton   }
205044d93782SGreg Clayton 
2051b9c1b51eSKate Stone   size_t GetDepth() const {
205244d93782SGreg Clayton     if (m_parent)
205344d93782SGreg Clayton       return 1 + m_parent->GetDepth();
205444d93782SGreg Clayton     return 0;
205544d93782SGreg Clayton   }
205644d93782SGreg Clayton 
2057b9c1b51eSKate Stone   int GetRowIndex() const { return m_row_idx; }
205844d93782SGreg Clayton 
2059b9c1b51eSKate Stone   void ClearChildren() { m_children.clear(); }
206044d93782SGreg Clayton 
2061b9c1b51eSKate Stone   void Resize(size_t n, const TreeItem &t) { m_children.resize(n, t); }
206244d93782SGreg Clayton 
2063b9c1b51eSKate Stone   TreeItem &operator[](size_t i) { return m_children[i]; }
206444d93782SGreg Clayton 
2065b9c1b51eSKate Stone   void SetRowIndex(int row_idx) { m_row_idx = row_idx; }
206644d93782SGreg Clayton 
2067b9c1b51eSKate Stone   size_t GetNumChildren() {
206844d93782SGreg Clayton     m_delegate.TreeDelegateGenerateChildren(*this);
206944d93782SGreg Clayton     return m_children.size();
207044d93782SGreg Clayton   }
207144d93782SGreg Clayton 
2072b9c1b51eSKate Stone   void ItemWasSelected() { m_delegate.TreeDelegateItemSelected(*this); }
2073315b6884SEugene Zelenko 
2074b9c1b51eSKate Stone   void CalculateRowIndexes(int &row_idx) {
207544d93782SGreg Clayton     SetRowIndex(row_idx);
207644d93782SGreg Clayton     ++row_idx;
207744d93782SGreg Clayton 
2078ec990867SGreg Clayton     const bool expanded = IsExpanded();
2079ec990867SGreg Clayton 
208005097246SAdrian Prantl     // The root item must calculate its children, or we must calculate the
208105097246SAdrian Prantl     // number of children if the item is expanded
2082c5dac77aSEugene Zelenko     if (m_parent == nullptr || expanded)
208344d93782SGreg Clayton       GetNumChildren();
208444d93782SGreg Clayton 
2085b9c1b51eSKate Stone     for (auto &item : m_children) {
208644d93782SGreg Clayton       if (expanded)
208744d93782SGreg Clayton         item.CalculateRowIndexes(row_idx);
208844d93782SGreg Clayton       else
208944d93782SGreg Clayton         item.SetRowIndex(-1);
209044d93782SGreg Clayton     }
209144d93782SGreg Clayton   }
209244d93782SGreg Clayton 
2093b9c1b51eSKate Stone   TreeItem *GetParent() { return m_parent; }
209444d93782SGreg Clayton 
2095b9c1b51eSKate Stone   bool IsExpanded() const { return m_is_expanded; }
209644d93782SGreg Clayton 
2097b9c1b51eSKate Stone   void Expand() { m_is_expanded = true; }
209844d93782SGreg Clayton 
2099b9c1b51eSKate Stone   void Unexpand() { m_is_expanded = false; }
210044d93782SGreg Clayton 
2101b9c1b51eSKate Stone   bool Draw(Window &window, const int first_visible_row,
2102b9c1b51eSKate Stone             const uint32_t selected_row_idx, int &row_idx, int &num_rows_left) {
210344d93782SGreg Clayton     if (num_rows_left <= 0)
210444d93782SGreg Clayton       return false;
210544d93782SGreg Clayton 
2106b9c1b51eSKate Stone     if (m_row_idx >= first_visible_row) {
210744d93782SGreg Clayton       window.MoveCursor(2, row_idx + 1);
210844d93782SGreg Clayton 
210944d93782SGreg Clayton       if (m_parent)
211044d93782SGreg Clayton         m_parent->DrawTreeForChild(window, this, 0);
211144d93782SGreg Clayton 
2112b9c1b51eSKate Stone       if (m_might_have_children) {
2113b9c1b51eSKate Stone         // It we can get UTF8 characters to work we should try to use the
211405097246SAdrian Prantl         // "symbol" UTF8 string below
211544d93782SGreg Clayton         //            const char *symbol = "";
211644d93782SGreg Clayton         //            if (row.expanded)
211744d93782SGreg Clayton         //                symbol = "\xe2\x96\xbd ";
211844d93782SGreg Clayton         //            else
211944d93782SGreg Clayton         //                symbol = "\xe2\x96\xb7 ";
212044d93782SGreg Clayton         //            window.PutCString (symbol);
212144d93782SGreg Clayton 
212244d93782SGreg Clayton         // The ACS_DARROW and ACS_RARROW don't look very nice they are just a
212344d93782SGreg Clayton         // 'v' or '>' character...
212444d93782SGreg Clayton         //            if (expanded)
212544d93782SGreg Clayton         //                window.PutChar (ACS_DARROW);
212644d93782SGreg Clayton         //            else
212744d93782SGreg Clayton         //                window.PutChar (ACS_RARROW);
212805097246SAdrian Prantl         // Since we can't find any good looking right arrow/down arrow symbols,
212905097246SAdrian Prantl         // just use a diamond...
213044d93782SGreg Clayton         window.PutChar(ACS_DIAMOND);
213144d93782SGreg Clayton         window.PutChar(ACS_HLINE);
213244d93782SGreg Clayton       }
2133b9c1b51eSKate Stone       bool highlight = (selected_row_idx == static_cast<size_t>(m_row_idx)) &&
2134b9c1b51eSKate Stone                        window.IsActive();
213544d93782SGreg Clayton 
213644d93782SGreg Clayton       if (highlight)
213744d93782SGreg Clayton         window.AttributeOn(A_REVERSE);
213844d93782SGreg Clayton 
213944d93782SGreg Clayton       m_delegate.TreeDelegateDrawTreeItem(*this, window);
214044d93782SGreg Clayton 
214144d93782SGreg Clayton       if (highlight)
214244d93782SGreg Clayton         window.AttributeOff(A_REVERSE);
214344d93782SGreg Clayton       ++row_idx;
214444d93782SGreg Clayton       --num_rows_left;
214544d93782SGreg Clayton     }
214644d93782SGreg Clayton 
214744d93782SGreg Clayton     if (num_rows_left <= 0)
214844d93782SGreg Clayton       return false; // We are done drawing...
214944d93782SGreg Clayton 
2150b9c1b51eSKate Stone     if (IsExpanded()) {
2151b9c1b51eSKate Stone       for (auto &item : m_children) {
215205097246SAdrian Prantl         // If we displayed all the rows and item.Draw() returns false we are
215305097246SAdrian Prantl         // done drawing and can exit this for loop
2154b9c1b51eSKate Stone         if (!item.Draw(window, first_visible_row, selected_row_idx, row_idx,
2155b9c1b51eSKate Stone                        num_rows_left))
215644d93782SGreg Clayton           break;
215744d93782SGreg Clayton       }
215844d93782SGreg Clayton     }
215944d93782SGreg Clayton     return num_rows_left >= 0; // Return true if not done drawing yet
216044d93782SGreg Clayton   }
216144d93782SGreg Clayton 
2162b9c1b51eSKate Stone   void DrawTreeForChild(Window &window, TreeItem *child,
2163b9c1b51eSKate Stone                         uint32_t reverse_depth) {
216444d93782SGreg Clayton     if (m_parent)
216544d93782SGreg Clayton       m_parent->DrawTreeForChild(window, this, reverse_depth + 1);
216644d93782SGreg Clayton 
2167b9c1b51eSKate Stone     if (&m_children.back() == child) {
216844d93782SGreg Clayton       // Last child
2169b9c1b51eSKate Stone       if (reverse_depth == 0) {
217044d93782SGreg Clayton         window.PutChar(ACS_LLCORNER);
217144d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2172b9c1b51eSKate Stone       } else {
217344d93782SGreg Clayton         window.PutChar(' ');
217444d93782SGreg Clayton         window.PutChar(' ');
217544d93782SGreg Clayton       }
2176b9c1b51eSKate Stone     } else {
2177b9c1b51eSKate Stone       if (reverse_depth == 0) {
217844d93782SGreg Clayton         window.PutChar(ACS_LTEE);
217944d93782SGreg Clayton         window.PutChar(ACS_HLINE);
2180b9c1b51eSKate Stone       } else {
218144d93782SGreg Clayton         window.PutChar(ACS_VLINE);
218244d93782SGreg Clayton         window.PutChar(' ');
218344d93782SGreg Clayton       }
218444d93782SGreg Clayton     }
218544d93782SGreg Clayton   }
218644d93782SGreg Clayton 
2187b9c1b51eSKate Stone   TreeItem *GetItemForRowIndex(uint32_t row_idx) {
21883985c8c6SSaleem Abdulrasool     if (static_cast<uint32_t>(m_row_idx) == row_idx)
218944d93782SGreg Clayton       return this;
219044d93782SGreg Clayton     if (m_children.empty())
2191c5dac77aSEugene Zelenko       return nullptr;
2192b9c1b51eSKate Stone     if (IsExpanded()) {
2193b9c1b51eSKate Stone       for (auto &item : m_children) {
219444d93782SGreg Clayton         TreeItem *selected_item_ptr = item.GetItemForRowIndex(row_idx);
219544d93782SGreg Clayton         if (selected_item_ptr)
219644d93782SGreg Clayton           return selected_item_ptr;
219744d93782SGreg Clayton       }
219844d93782SGreg Clayton     }
2199c5dac77aSEugene Zelenko     return nullptr;
220044d93782SGreg Clayton   }
220144d93782SGreg Clayton 
2202b9c1b51eSKate Stone   void *GetUserData() const { return m_user_data; }
2203ec990867SGreg Clayton 
2204b9c1b51eSKate Stone   void SetUserData(void *user_data) { m_user_data = user_data; }
2205ec990867SGreg Clayton 
2206b9c1b51eSKate Stone   uint64_t GetIdentifier() const { return m_identifier; }
220744d93782SGreg Clayton 
2208b9c1b51eSKate Stone   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
220944d93782SGreg Clayton 
2210b9c1b51eSKate Stone   void SetMightHaveChildren(bool b) { m_might_have_children = b; }
2211ec990867SGreg Clayton 
221244d93782SGreg Clayton protected:
221344d93782SGreg Clayton   TreeItem *m_parent;
221444d93782SGreg Clayton   TreeDelegate &m_delegate;
2215ec990867SGreg Clayton   void *m_user_data;
221644d93782SGreg Clayton   uint64_t m_identifier;
2217b9c1b51eSKate Stone   int m_row_idx; // Zero based visible row index, -1 if not visible or for the
2218b9c1b51eSKate Stone                  // root item
221944d93782SGreg Clayton   std::vector<TreeItem> m_children;
222044d93782SGreg Clayton   bool m_might_have_children;
222144d93782SGreg Clayton   bool m_is_expanded;
222244d93782SGreg Clayton };
222344d93782SGreg Clayton 
2224b9c1b51eSKate Stone class TreeWindowDelegate : public WindowDelegate {
222544d93782SGreg Clayton public:
2226b9c1b51eSKate Stone   TreeWindowDelegate(Debugger &debugger, const TreeDelegateSP &delegate_sp)
2227b9c1b51eSKate Stone       : m_debugger(debugger), m_delegate_sp(delegate_sp),
2228b9c1b51eSKate Stone         m_root(nullptr, *delegate_sp, true), m_selected_item(nullptr),
2229b9c1b51eSKate Stone         m_num_rows(0), m_selected_row_idx(0), m_first_visible_row(0),
2230b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
223144d93782SGreg Clayton 
2232b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
223344d93782SGreg Clayton 
2234b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
2235b9c1b51eSKate Stone     ExecutionContext exe_ctx(
2236b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
223744d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
223844d93782SGreg Clayton 
223944d93782SGreg Clayton     bool display_content = false;
2240b9c1b51eSKate Stone     if (process) {
224144d93782SGreg Clayton       StateType state = process->GetState();
2242b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
224344d93782SGreg Clayton         // We are stopped, so it is ok to
224444d93782SGreg Clayton         display_content = true;
2245b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
224644d93782SGreg Clayton         return true; // Don't do any updating when we are running
224744d93782SGreg Clayton       }
224844d93782SGreg Clayton     }
224944d93782SGreg Clayton 
225044d93782SGreg Clayton     m_min_x = 2;
225144d93782SGreg Clayton     m_min_y = 1;
225244d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
225344d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
225444d93782SGreg Clayton 
225544d93782SGreg Clayton     window.Erase();
225644d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
225744d93782SGreg Clayton 
2258b9c1b51eSKate Stone     if (display_content) {
225944d93782SGreg Clayton       const int num_visible_rows = NumVisibleRows();
226044d93782SGreg Clayton       m_num_rows = 0;
226144d93782SGreg Clayton       m_root.CalculateRowIndexes(m_num_rows);
226244d93782SGreg Clayton 
226305097246SAdrian Prantl       // If we unexpanded while having something selected our total number of
226405097246SAdrian Prantl       // rows is less than the num visible rows, then make sure we show all the
226505097246SAdrian Prantl       // rows by setting the first visible row accordingly.
226644d93782SGreg Clayton       if (m_first_visible_row > 0 && m_num_rows < num_visible_rows)
226744d93782SGreg Clayton         m_first_visible_row = 0;
226844d93782SGreg Clayton 
226944d93782SGreg Clayton       // Make sure the selected row is always visible
227044d93782SGreg Clayton       if (m_selected_row_idx < m_first_visible_row)
227144d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx;
227244d93782SGreg Clayton       else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
227344d93782SGreg Clayton         m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
227444d93782SGreg Clayton 
227544d93782SGreg Clayton       int row_idx = 0;
227644d93782SGreg Clayton       int num_rows_left = num_visible_rows;
2277b9c1b51eSKate Stone       m_root.Draw(window, m_first_visible_row, m_selected_row_idx, row_idx,
2278b9c1b51eSKate Stone                   num_rows_left);
227944d93782SGreg Clayton       // Get the selected row
228044d93782SGreg Clayton       m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
2281b9c1b51eSKate Stone     } else {
2282c5dac77aSEugene Zelenko       m_selected_item = nullptr;
228344d93782SGreg Clayton     }
228444d93782SGreg Clayton 
228544d93782SGreg Clayton     window.DeferredRefresh();
228644d93782SGreg Clayton 
228744d93782SGreg Clayton     return true; // Drawing handled
228844d93782SGreg Clayton   }
228944d93782SGreg Clayton 
2290b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
229144d93782SGreg Clayton     return "Thread window keyboard shortcuts:";
229244d93782SGreg Clayton   }
229344d93782SGreg Clayton 
2294b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
229544d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
229644d93782SGreg Clayton         {KEY_UP, "Select previous item"},
229744d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
229844d93782SGreg Clayton         {KEY_RIGHT, "Expand the selected item"},
2299b9c1b51eSKate Stone         {KEY_LEFT,
2300b9c1b51eSKate Stone          "Unexpand the selected item or select parent if not expanded"},
230144d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
230244d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
230344d93782SGreg Clayton         {'h', "Show help dialog"},
230444d93782SGreg Clayton         {' ', "Toggle item expansion"},
230544d93782SGreg Clayton         {',', "Page up"},
230644d93782SGreg Clayton         {'.', "Page down"},
2307b9c1b51eSKate Stone         {'\0', nullptr}};
230844d93782SGreg Clayton     return g_source_view_key_help;
230944d93782SGreg Clayton   }
231044d93782SGreg Clayton 
2311b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2312b9c1b51eSKate Stone     switch (c) {
231344d93782SGreg Clayton     case ',':
231444d93782SGreg Clayton     case KEY_PPAGE:
231544d93782SGreg Clayton       // Page up key
2316b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
231744d93782SGreg Clayton         if (m_first_visible_row > m_max_y)
231844d93782SGreg Clayton           m_first_visible_row -= m_max_y;
231944d93782SGreg Clayton         else
232044d93782SGreg Clayton           m_first_visible_row = 0;
232144d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
232244d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
232344d93782SGreg Clayton         if (m_selected_item)
232444d93782SGreg Clayton           m_selected_item->ItemWasSelected();
232544d93782SGreg Clayton       }
232644d93782SGreg Clayton       return eKeyHandled;
232744d93782SGreg Clayton 
232844d93782SGreg Clayton     case '.':
232944d93782SGreg Clayton     case KEY_NPAGE:
233044d93782SGreg Clayton       // Page down key
2331b9c1b51eSKate Stone       if (m_num_rows > m_max_y) {
2332b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
233344d93782SGreg Clayton           m_first_visible_row += m_max_y;
233444d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
233544d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
233644d93782SGreg Clayton           if (m_selected_item)
233744d93782SGreg Clayton             m_selected_item->ItemWasSelected();
233844d93782SGreg Clayton         }
233944d93782SGreg Clayton       }
234044d93782SGreg Clayton       return eKeyHandled;
234144d93782SGreg Clayton 
234244d93782SGreg Clayton     case KEY_UP:
2343b9c1b51eSKate Stone       if (m_selected_row_idx > 0) {
234444d93782SGreg Clayton         --m_selected_row_idx;
234544d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
234644d93782SGreg Clayton         if (m_selected_item)
234744d93782SGreg Clayton           m_selected_item->ItemWasSelected();
234844d93782SGreg Clayton       }
234944d93782SGreg Clayton       return eKeyHandled;
2350315b6884SEugene Zelenko 
235144d93782SGreg Clayton     case KEY_DOWN:
2352b9c1b51eSKate Stone       if (m_selected_row_idx + 1 < m_num_rows) {
235344d93782SGreg Clayton         ++m_selected_row_idx;
235444d93782SGreg Clayton         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
235544d93782SGreg Clayton         if (m_selected_item)
235644d93782SGreg Clayton           m_selected_item->ItemWasSelected();
235744d93782SGreg Clayton       }
235844d93782SGreg Clayton       return eKeyHandled;
235944d93782SGreg Clayton 
236044d93782SGreg Clayton     case KEY_RIGHT:
2361b9c1b51eSKate Stone       if (m_selected_item) {
236244d93782SGreg Clayton         if (!m_selected_item->IsExpanded())
236344d93782SGreg Clayton           m_selected_item->Expand();
236444d93782SGreg Clayton       }
236544d93782SGreg Clayton       return eKeyHandled;
236644d93782SGreg Clayton 
236744d93782SGreg Clayton     case KEY_LEFT:
2368b9c1b51eSKate Stone       if (m_selected_item) {
236944d93782SGreg Clayton         if (m_selected_item->IsExpanded())
237044d93782SGreg Clayton           m_selected_item->Unexpand();
2371b9c1b51eSKate Stone         else if (m_selected_item->GetParent()) {
237244d93782SGreg Clayton           m_selected_row_idx = m_selected_item->GetParent()->GetRowIndex();
237344d93782SGreg Clayton           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
237444d93782SGreg Clayton           if (m_selected_item)
237544d93782SGreg Clayton             m_selected_item->ItemWasSelected();
237644d93782SGreg Clayton         }
237744d93782SGreg Clayton       }
237844d93782SGreg Clayton       return eKeyHandled;
237944d93782SGreg Clayton 
238044d93782SGreg Clayton     case ' ':
238144d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2382b9c1b51eSKate Stone       if (m_selected_item) {
238344d93782SGreg Clayton         if (m_selected_item->IsExpanded())
238444d93782SGreg Clayton           m_selected_item->Unexpand();
238544d93782SGreg Clayton         else
238644d93782SGreg Clayton           m_selected_item->Expand();
238744d93782SGreg Clayton       }
238844d93782SGreg Clayton       return eKeyHandled;
238944d93782SGreg Clayton 
239044d93782SGreg Clayton     case 'h':
239144d93782SGreg Clayton       window.CreateHelpSubwindow();
239244d93782SGreg Clayton       return eKeyHandled;
239344d93782SGreg Clayton 
239444d93782SGreg Clayton     default:
239544d93782SGreg Clayton       break;
239644d93782SGreg Clayton     }
239744d93782SGreg Clayton     return eKeyNotHandled;
239844d93782SGreg Clayton   }
239944d93782SGreg Clayton 
240044d93782SGreg Clayton protected:
240144d93782SGreg Clayton   Debugger &m_debugger;
240244d93782SGreg Clayton   TreeDelegateSP m_delegate_sp;
240344d93782SGreg Clayton   TreeItem m_root;
240444d93782SGreg Clayton   TreeItem *m_selected_item;
240544d93782SGreg Clayton   int m_num_rows;
240644d93782SGreg Clayton   int m_selected_row_idx;
240744d93782SGreg Clayton   int m_first_visible_row;
240844d93782SGreg Clayton   int m_min_x;
240944d93782SGreg Clayton   int m_min_y;
241044d93782SGreg Clayton   int m_max_x;
241144d93782SGreg Clayton   int m_max_y;
241244d93782SGreg Clayton };
241344d93782SGreg Clayton 
2414b9c1b51eSKate Stone class FrameTreeDelegate : public TreeDelegate {
241544d93782SGreg Clayton public:
2416b9c1b51eSKate Stone   FrameTreeDelegate() : TreeDelegate() {
2417b9c1b51eSKate Stone     FormatEntity::Parse(
2418b9c1b51eSKate Stone         "frame #${frame.index}: {${function.name}${function.pc-offset}}}",
2419554f68d3SGreg Clayton         m_format);
242044d93782SGreg Clayton   }
242144d93782SGreg Clayton 
2422315b6884SEugene Zelenko   ~FrameTreeDelegate() override = default;
242344d93782SGreg Clayton 
2424b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2425ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2426b9c1b51eSKate Stone     if (thread) {
242744d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2428ec990867SGreg Clayton       StackFrameSP frame_sp = thread->GetStackFrameAtIndex(frame_idx);
2429b9c1b51eSKate Stone       if (frame_sp) {
243044d93782SGreg Clayton         StreamString strm;
2431b9c1b51eSKate Stone         const SymbolContext &sc =
2432b9c1b51eSKate Stone             frame_sp->GetSymbolContext(eSymbolContextEverything);
243344d93782SGreg Clayton         ExecutionContext exe_ctx(frame_sp);
2434b9c1b51eSKate Stone         if (FormatEntity::Format(m_format, strm, &sc, &exe_ctx, nullptr,
2435b9c1b51eSKate Stone                                  nullptr, false, false)) {
243644d93782SGreg Clayton           int right_pad = 1;
2437c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
243844d93782SGreg Clayton         }
243944d93782SGreg Clayton       }
244044d93782SGreg Clayton     }
244144d93782SGreg Clayton   }
2442315b6884SEugene Zelenko 
2443b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
244444d93782SGreg Clayton     // No children for frames yet...
244544d93782SGreg Clayton   }
244644d93782SGreg Clayton 
2447b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2448ec990867SGreg Clayton     Thread *thread = (Thread *)item.GetUserData();
2449b9c1b51eSKate Stone     if (thread) {
2450b9c1b51eSKate Stone       thread->GetProcess()->GetThreadList().SetSelectedThreadByID(
2451b9c1b51eSKate Stone           thread->GetID());
245244d93782SGreg Clayton       const uint64_t frame_idx = item.GetIdentifier();
2453ec990867SGreg Clayton       thread->SetSelectedFrameByIndex(frame_idx);
245444d93782SGreg Clayton       return true;
245544d93782SGreg Clayton     }
245644d93782SGreg Clayton     return false;
245744d93782SGreg Clayton   }
2458315b6884SEugene Zelenko 
2459554f68d3SGreg Clayton protected:
2460554f68d3SGreg Clayton   FormatEntity::Entry m_format;
246144d93782SGreg Clayton };
246244d93782SGreg Clayton 
2463b9c1b51eSKate Stone class ThreadTreeDelegate : public TreeDelegate {
246444d93782SGreg Clayton public:
2465b9c1b51eSKate Stone   ThreadTreeDelegate(Debugger &debugger)
2466b9c1b51eSKate Stone       : TreeDelegate(), m_debugger(debugger), m_tid(LLDB_INVALID_THREAD_ID),
2467b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2468b9c1b51eSKate Stone     FormatEntity::Parse("thread #${thread.index}: tid = ${thread.id}{, stop "
2469b9c1b51eSKate Stone                         "reason = ${thread.stop-reason}}",
2470554f68d3SGreg Clayton                         m_format);
247144d93782SGreg Clayton   }
247244d93782SGreg Clayton 
2473315b6884SEugene Zelenko   ~ThreadTreeDelegate() override = default;
247444d93782SGreg Clayton 
2475b9c1b51eSKate Stone   ProcessSP GetProcess() {
2476b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2477b9c1b51eSKate Stone         .GetExecutionContext()
2478b9c1b51eSKate Stone         .GetProcessSP();
2479ec990867SGreg Clayton   }
2480ec990867SGreg Clayton 
2481b9c1b51eSKate Stone   ThreadSP GetThread(const TreeItem &item) {
2482ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2483ec990867SGreg Clayton     if (process_sp)
2484ec990867SGreg Clayton       return process_sp->GetThreadList().FindThreadByID(item.GetIdentifier());
2485ec990867SGreg Clayton     return ThreadSP();
2486ec990867SGreg Clayton   }
2487ec990867SGreg Clayton 
2488b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2489ec990867SGreg Clayton     ThreadSP thread_sp = GetThread(item);
2490b9c1b51eSKate Stone     if (thread_sp) {
249144d93782SGreg Clayton       StreamString strm;
249244d93782SGreg Clayton       ExecutionContext exe_ctx(thread_sp);
2493b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2494b9c1b51eSKate Stone                                nullptr, false, false)) {
249544d93782SGreg Clayton         int right_pad = 1;
2496c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
249744d93782SGreg Clayton       }
249844d93782SGreg Clayton     }
249944d93782SGreg Clayton   }
2500315b6884SEugene Zelenko 
2501b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2502ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2503b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
250444d93782SGreg Clayton       StateType state = process_sp->GetState();
2505b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2506ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2507b9c1b51eSKate Stone         if (thread_sp) {
2508b9c1b51eSKate Stone           if (m_stop_id == process_sp->GetStopID() &&
2509b9c1b51eSKate Stone               thread_sp->GetID() == m_tid)
251044d93782SGreg Clayton             return; // Children are already up to date
2511b9c1b51eSKate Stone           if (!m_frame_delegate_sp) {
251244d93782SGreg Clayton             // Always expand the thread item the first time we show it
2513*796ac80bSJonas Devlieghere             m_frame_delegate_sp = std::make_shared<FrameTreeDelegate>();
251444d93782SGreg Clayton           }
251544d93782SGreg Clayton 
251644d93782SGreg Clayton           m_stop_id = process_sp->GetStopID();
251744d93782SGreg Clayton           m_tid = thread_sp->GetID();
251844d93782SGreg Clayton 
251944d93782SGreg Clayton           TreeItem t(&item, *m_frame_delegate_sp, false);
252044d93782SGreg Clayton           size_t num_frames = thread_sp->GetStackFrameCount();
252144d93782SGreg Clayton           item.Resize(num_frames, t);
2522b9c1b51eSKate Stone           for (size_t i = 0; i < num_frames; ++i) {
2523ec990867SGreg Clayton             item[i].SetUserData(thread_sp.get());
252444d93782SGreg Clayton             item[i].SetIdentifier(i);
252544d93782SGreg Clayton           }
252644d93782SGreg Clayton         }
252744d93782SGreg Clayton         return;
252844d93782SGreg Clayton       }
252944d93782SGreg Clayton     }
253044d93782SGreg Clayton     item.ClearChildren();
253144d93782SGreg Clayton   }
253244d93782SGreg Clayton 
2533b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override {
2534ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2535b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2536ec990867SGreg Clayton       StateType state = process_sp->GetState();
2537b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2538ec990867SGreg Clayton         ThreadSP thread_sp = GetThread(item);
2539b9c1b51eSKate Stone         if (thread_sp) {
254044d93782SGreg Clayton           ThreadList &thread_list = thread_sp->GetProcess()->GetThreadList();
2541bb19a13cSSaleem Abdulrasool           std::lock_guard<std::recursive_mutex> guard(thread_list.GetMutex());
254244d93782SGreg Clayton           ThreadSP selected_thread_sp = thread_list.GetSelectedThread();
2543b9c1b51eSKate Stone           if (selected_thread_sp->GetID() != thread_sp->GetID()) {
254444d93782SGreg Clayton             thread_list.SetSelectedThreadByID(thread_sp->GetID());
254544d93782SGreg Clayton             return true;
254644d93782SGreg Clayton           }
254744d93782SGreg Clayton         }
2548ec990867SGreg Clayton       }
2549ec990867SGreg Clayton     }
255044d93782SGreg Clayton     return false;
255144d93782SGreg Clayton   }
255244d93782SGreg Clayton 
255344d93782SGreg Clayton protected:
255444d93782SGreg Clayton   Debugger &m_debugger;
255544d93782SGreg Clayton   std::shared_ptr<FrameTreeDelegate> m_frame_delegate_sp;
255644d93782SGreg Clayton   lldb::user_id_t m_tid;
255744d93782SGreg Clayton   uint32_t m_stop_id;
2558554f68d3SGreg Clayton   FormatEntity::Entry m_format;
255944d93782SGreg Clayton };
256044d93782SGreg Clayton 
2561b9c1b51eSKate Stone class ThreadsTreeDelegate : public TreeDelegate {
2562ec990867SGreg Clayton public:
2563b9c1b51eSKate Stone   ThreadsTreeDelegate(Debugger &debugger)
2564b9c1b51eSKate Stone       : TreeDelegate(), m_thread_delegate_sp(), m_debugger(debugger),
2565b9c1b51eSKate Stone         m_stop_id(UINT32_MAX) {
2566554f68d3SGreg Clayton     FormatEntity::Parse("process ${process.id}{, name = ${process.name}}",
2567554f68d3SGreg Clayton                         m_format);
2568ec990867SGreg Clayton   }
2569ec990867SGreg Clayton 
2570315b6884SEugene Zelenko   ~ThreadsTreeDelegate() override = default;
2571ec990867SGreg Clayton 
2572b9c1b51eSKate Stone   ProcessSP GetProcess() {
2573b9c1b51eSKate Stone     return m_debugger.GetCommandInterpreter()
2574b9c1b51eSKate Stone         .GetExecutionContext()
2575b9c1b51eSKate Stone         .GetProcessSP();
2576ec990867SGreg Clayton   }
2577ec990867SGreg Clayton 
2578b9c1b51eSKate Stone   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
2579ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2580b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2581ec990867SGreg Clayton       StreamString strm;
2582ec990867SGreg Clayton       ExecutionContext exe_ctx(process_sp);
2583b9c1b51eSKate Stone       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
2584b9c1b51eSKate Stone                                nullptr, false, false)) {
2585ec990867SGreg Clayton         int right_pad = 1;
2586c156427dSZachary Turner         window.PutCStringTruncated(strm.GetString().str().c_str(), right_pad);
2587ec990867SGreg Clayton       }
2588ec990867SGreg Clayton     }
2589ec990867SGreg Clayton   }
2590ec990867SGreg Clayton 
2591b9c1b51eSKate Stone   void TreeDelegateGenerateChildren(TreeItem &item) override {
2592ec990867SGreg Clayton     ProcessSP process_sp = GetProcess();
2593b9c1b51eSKate Stone     if (process_sp && process_sp->IsAlive()) {
2594ec990867SGreg Clayton       StateType state = process_sp->GetState();
2595b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
2596ec990867SGreg Clayton         const uint32_t stop_id = process_sp->GetStopID();
2597ec990867SGreg Clayton         if (m_stop_id == stop_id)
2598ec990867SGreg Clayton           return; // Children are already up to date
2599ec990867SGreg Clayton 
2600ec990867SGreg Clayton         m_stop_id = stop_id;
2601ec990867SGreg Clayton 
2602b9c1b51eSKate Stone         if (!m_thread_delegate_sp) {
2603ec990867SGreg Clayton           // Always expand the thread item the first time we show it
2604ec990867SGreg Clayton           // item.Expand();
2605*796ac80bSJonas Devlieghere           m_thread_delegate_sp =
2606*796ac80bSJonas Devlieghere               std::make_shared<ThreadTreeDelegate>(m_debugger);
2607ec990867SGreg Clayton         }
2608ec990867SGreg Clayton 
2609ec990867SGreg Clayton         TreeItem t(&item, *m_thread_delegate_sp, false);
2610ec990867SGreg Clayton         ThreadList &threads = process_sp->GetThreadList();
2611bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
2612ec990867SGreg Clayton         size_t num_threads = threads.GetSize();
2613ec990867SGreg Clayton         item.Resize(num_threads, t);
2614b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
2615ec990867SGreg Clayton           item[i].SetIdentifier(threads.GetThreadAtIndex(i)->GetID());
2616ec990867SGreg Clayton           item[i].SetMightHaveChildren(true);
2617ec990867SGreg Clayton         }
2618ec990867SGreg Clayton         return;
2619ec990867SGreg Clayton       }
2620ec990867SGreg Clayton     }
2621ec990867SGreg Clayton     item.ClearChildren();
2622ec990867SGreg Clayton   }
2623ec990867SGreg Clayton 
2624b9c1b51eSKate Stone   bool TreeDelegateItemSelected(TreeItem &item) override { return false; }
2625ec990867SGreg Clayton 
2626ec990867SGreg Clayton protected:
2627ec990867SGreg Clayton   std::shared_ptr<ThreadTreeDelegate> m_thread_delegate_sp;
2628ec990867SGreg Clayton   Debugger &m_debugger;
2629ec990867SGreg Clayton   uint32_t m_stop_id;
2630554f68d3SGreg Clayton   FormatEntity::Entry m_format;
2631ec990867SGreg Clayton };
2632ec990867SGreg Clayton 
2633b9c1b51eSKate Stone class ValueObjectListDelegate : public WindowDelegate {
263444d93782SGreg Clayton public:
2635b9c1b51eSKate Stone   ValueObjectListDelegate()
26368369b28dSGreg Clayton       : m_rows(), m_selected_row(nullptr),
2637b9c1b51eSKate Stone         m_selected_row_idx(0), m_first_visible_row(0), m_num_rows(0),
2638b9c1b51eSKate Stone         m_max_x(0), m_max_y(0) {}
263944d93782SGreg Clayton 
2640b9c1b51eSKate Stone   ValueObjectListDelegate(ValueObjectList &valobj_list)
26418369b28dSGreg Clayton       : m_rows(), m_selected_row(nullptr),
2642b9c1b51eSKate Stone         m_selected_row_idx(0), m_first_visible_row(0), m_num_rows(0),
2643b9c1b51eSKate Stone         m_max_x(0), m_max_y(0) {
264444d93782SGreg Clayton     SetValues(valobj_list);
264544d93782SGreg Clayton   }
264644d93782SGreg Clayton 
2647315b6884SEugene Zelenko   ~ValueObjectListDelegate() override = default;
264844d93782SGreg Clayton 
2649b9c1b51eSKate Stone   void SetValues(ValueObjectList &valobj_list) {
2650c5dac77aSEugene Zelenko     m_selected_row = nullptr;
265144d93782SGreg Clayton     m_selected_row_idx = 0;
265244d93782SGreg Clayton     m_first_visible_row = 0;
265344d93782SGreg Clayton     m_num_rows = 0;
265444d93782SGreg Clayton     m_rows.clear();
26558369b28dSGreg Clayton     for (auto &valobj_sp : valobj_list.GetObjects())
26568369b28dSGreg Clayton       m_rows.push_back(Row(valobj_sp, nullptr));
265744d93782SGreg Clayton   }
265844d93782SGreg Clayton 
2659b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
266044d93782SGreg Clayton     m_num_rows = 0;
266144d93782SGreg Clayton     m_min_x = 2;
266244d93782SGreg Clayton     m_min_y = 1;
266344d93782SGreg Clayton     m_max_x = window.GetWidth() - 1;
266444d93782SGreg Clayton     m_max_y = window.GetHeight() - 1;
266544d93782SGreg Clayton 
266644d93782SGreg Clayton     window.Erase();
266744d93782SGreg Clayton     window.DrawTitleBox(window.GetName());
266844d93782SGreg Clayton 
266944d93782SGreg Clayton     const int num_visible_rows = NumVisibleRows();
267044d93782SGreg Clayton     const int num_rows = CalculateTotalNumberRows(m_rows);
267144d93782SGreg Clayton 
267205097246SAdrian Prantl     // If we unexpanded while having something selected our total number of
267305097246SAdrian Prantl     // rows is less than the num visible rows, then make sure we show all the
267405097246SAdrian Prantl     // rows by setting the first visible row accordingly.
267544d93782SGreg Clayton     if (m_first_visible_row > 0 && num_rows < num_visible_rows)
267644d93782SGreg Clayton       m_first_visible_row = 0;
267744d93782SGreg Clayton 
267844d93782SGreg Clayton     // Make sure the selected row is always visible
267944d93782SGreg Clayton     if (m_selected_row_idx < m_first_visible_row)
268044d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx;
268144d93782SGreg Clayton     else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
268244d93782SGreg Clayton       m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
268344d93782SGreg Clayton 
268444d93782SGreg Clayton     DisplayRows(window, m_rows, g_options);
268544d93782SGreg Clayton 
268644d93782SGreg Clayton     window.DeferredRefresh();
268744d93782SGreg Clayton 
268844d93782SGreg Clayton     // Get the selected row
268944d93782SGreg Clayton     m_selected_row = GetRowForRowIndex(m_selected_row_idx);
269005097246SAdrian Prantl     // Keep the cursor on the selected row so the highlight and the cursor are
269105097246SAdrian Prantl     // always on the same line
269244d93782SGreg Clayton     if (m_selected_row)
2693b9c1b51eSKate Stone       window.MoveCursor(m_selected_row->x, m_selected_row->y);
269444d93782SGreg Clayton 
269544d93782SGreg Clayton     return true; // Drawing handled
269644d93782SGreg Clayton   }
269744d93782SGreg Clayton 
2698b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
269944d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
270044d93782SGreg Clayton         {KEY_UP, "Select previous item"},
270144d93782SGreg Clayton         {KEY_DOWN, "Select next item"},
270244d93782SGreg Clayton         {KEY_RIGHT, "Expand selected item"},
270344d93782SGreg Clayton         {KEY_LEFT, "Unexpand selected item or select parent if not expanded"},
270444d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
270544d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
270644d93782SGreg Clayton         {'A', "Format as annotated address"},
270744d93782SGreg Clayton         {'b', "Format as binary"},
270844d93782SGreg Clayton         {'B', "Format as hex bytes with ASCII"},
270944d93782SGreg Clayton         {'c', "Format as character"},
271044d93782SGreg Clayton         {'d', "Format as a signed integer"},
271144d93782SGreg Clayton         {'D', "Format selected value using the default format for the type"},
271244d93782SGreg Clayton         {'f', "Format as float"},
271344d93782SGreg Clayton         {'h', "Show help dialog"},
271444d93782SGreg Clayton         {'i', "Format as instructions"},
271544d93782SGreg Clayton         {'o', "Format as octal"},
271644d93782SGreg Clayton         {'p', "Format as pointer"},
271744d93782SGreg Clayton         {'s', "Format as C string"},
271844d93782SGreg Clayton         {'t', "Toggle showing/hiding type names"},
271944d93782SGreg Clayton         {'u', "Format as an unsigned integer"},
272044d93782SGreg Clayton         {'x', "Format as hex"},
272144d93782SGreg Clayton         {'X', "Format as uppercase hex"},
272244d93782SGreg Clayton         {' ', "Toggle item expansion"},
272344d93782SGreg Clayton         {',', "Page up"},
272444d93782SGreg Clayton         {'.', "Page down"},
2725b9c1b51eSKate Stone         {'\0', nullptr}};
272644d93782SGreg Clayton     return g_source_view_key_help;
272744d93782SGreg Clayton   }
272844d93782SGreg Clayton 
2729b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
2730b9c1b51eSKate Stone     switch (c) {
273144d93782SGreg Clayton     case 'x':
273244d93782SGreg Clayton     case 'X':
273344d93782SGreg Clayton     case 'o':
273444d93782SGreg Clayton     case 's':
273544d93782SGreg Clayton     case 'u':
273644d93782SGreg Clayton     case 'd':
273744d93782SGreg Clayton     case 'D':
273844d93782SGreg Clayton     case 'i':
273944d93782SGreg Clayton     case 'A':
274044d93782SGreg Clayton     case 'p':
274144d93782SGreg Clayton     case 'c':
274244d93782SGreg Clayton     case 'b':
274344d93782SGreg Clayton     case 'B':
274444d93782SGreg Clayton     case 'f':
274544d93782SGreg Clayton       // Change the format for the currently selected item
27468369b28dSGreg Clayton       if (m_selected_row) {
27478369b28dSGreg Clayton         auto valobj_sp = m_selected_row->value.GetSP();
27488369b28dSGreg Clayton         if (valobj_sp)
27498369b28dSGreg Clayton           valobj_sp->SetFormat(FormatForChar(c));
27508369b28dSGreg Clayton       }
275144d93782SGreg Clayton       return eKeyHandled;
275244d93782SGreg Clayton 
275344d93782SGreg Clayton     case 't':
275444d93782SGreg Clayton       // Toggle showing type names
275544d93782SGreg Clayton       g_options.show_types = !g_options.show_types;
275644d93782SGreg Clayton       return eKeyHandled;
275744d93782SGreg Clayton 
275844d93782SGreg Clayton     case ',':
275944d93782SGreg Clayton     case KEY_PPAGE:
276044d93782SGreg Clayton       // Page up key
2761b9c1b51eSKate Stone       if (m_first_visible_row > 0) {
27623985c8c6SSaleem Abdulrasool         if (static_cast<int>(m_first_visible_row) > m_max_y)
276344d93782SGreg Clayton           m_first_visible_row -= m_max_y;
276444d93782SGreg Clayton         else
276544d93782SGreg Clayton           m_first_visible_row = 0;
276644d93782SGreg Clayton         m_selected_row_idx = m_first_visible_row;
276744d93782SGreg Clayton       }
276844d93782SGreg Clayton       return eKeyHandled;
276944d93782SGreg Clayton 
277044d93782SGreg Clayton     case '.':
277144d93782SGreg Clayton     case KEY_NPAGE:
277244d93782SGreg Clayton       // Page down key
2773b9c1b51eSKate Stone       if (m_num_rows > static_cast<size_t>(m_max_y)) {
2774b9c1b51eSKate Stone         if (m_first_visible_row + m_max_y < m_num_rows) {
277544d93782SGreg Clayton           m_first_visible_row += m_max_y;
277644d93782SGreg Clayton           m_selected_row_idx = m_first_visible_row;
277744d93782SGreg Clayton         }
277844d93782SGreg Clayton       }
277944d93782SGreg Clayton       return eKeyHandled;
278044d93782SGreg Clayton 
278144d93782SGreg Clayton     case KEY_UP:
278244d93782SGreg Clayton       if (m_selected_row_idx > 0)
278344d93782SGreg Clayton         --m_selected_row_idx;
278444d93782SGreg Clayton       return eKeyHandled;
2785315b6884SEugene Zelenko 
278644d93782SGreg Clayton     case KEY_DOWN:
278744d93782SGreg Clayton       if (m_selected_row_idx + 1 < m_num_rows)
278844d93782SGreg Clayton         ++m_selected_row_idx;
278944d93782SGreg Clayton       return eKeyHandled;
279044d93782SGreg Clayton 
279144d93782SGreg Clayton     case KEY_RIGHT:
2792b9c1b51eSKate Stone       if (m_selected_row) {
279344d93782SGreg Clayton         if (!m_selected_row->expanded)
279444d93782SGreg Clayton           m_selected_row->Expand();
279544d93782SGreg Clayton       }
279644d93782SGreg Clayton       return eKeyHandled;
279744d93782SGreg Clayton 
279844d93782SGreg Clayton     case KEY_LEFT:
2799b9c1b51eSKate Stone       if (m_selected_row) {
280044d93782SGreg Clayton         if (m_selected_row->expanded)
280144d93782SGreg Clayton           m_selected_row->Unexpand();
280244d93782SGreg Clayton         else if (m_selected_row->parent)
280344d93782SGreg Clayton           m_selected_row_idx = m_selected_row->parent->row_idx;
280444d93782SGreg Clayton       }
280544d93782SGreg Clayton       return eKeyHandled;
280644d93782SGreg Clayton 
280744d93782SGreg Clayton     case ' ':
280844d93782SGreg Clayton       // Toggle expansion state when SPACE is pressed
2809b9c1b51eSKate Stone       if (m_selected_row) {
281044d93782SGreg Clayton         if (m_selected_row->expanded)
281144d93782SGreg Clayton           m_selected_row->Unexpand();
281244d93782SGreg Clayton         else
281344d93782SGreg Clayton           m_selected_row->Expand();
281444d93782SGreg Clayton       }
281544d93782SGreg Clayton       return eKeyHandled;
281644d93782SGreg Clayton 
281744d93782SGreg Clayton     case 'h':
281844d93782SGreg Clayton       window.CreateHelpSubwindow();
281944d93782SGreg Clayton       return eKeyHandled;
282044d93782SGreg Clayton 
282144d93782SGreg Clayton     default:
282244d93782SGreg Clayton       break;
282344d93782SGreg Clayton     }
282444d93782SGreg Clayton     return eKeyNotHandled;
282544d93782SGreg Clayton   }
282644d93782SGreg Clayton 
282744d93782SGreg Clayton protected:
282844d93782SGreg Clayton   std::vector<Row> m_rows;
282944d93782SGreg Clayton   Row *m_selected_row;
283044d93782SGreg Clayton   uint32_t m_selected_row_idx;
283144d93782SGreg Clayton   uint32_t m_first_visible_row;
283244d93782SGreg Clayton   uint32_t m_num_rows;
283344d93782SGreg Clayton   int m_min_x;
283444d93782SGreg Clayton   int m_min_y;
283544d93782SGreg Clayton   int m_max_x;
283644d93782SGreg Clayton   int m_max_y;
283744d93782SGreg Clayton 
2838b9c1b51eSKate Stone   static Format FormatForChar(int c) {
2839b9c1b51eSKate Stone     switch (c) {
2840b9c1b51eSKate Stone     case 'x':
2841b9c1b51eSKate Stone       return eFormatHex;
2842b9c1b51eSKate Stone     case 'X':
2843b9c1b51eSKate Stone       return eFormatHexUppercase;
2844b9c1b51eSKate Stone     case 'o':
2845b9c1b51eSKate Stone       return eFormatOctal;
2846b9c1b51eSKate Stone     case 's':
2847b9c1b51eSKate Stone       return eFormatCString;
2848b9c1b51eSKate Stone     case 'u':
2849b9c1b51eSKate Stone       return eFormatUnsigned;
2850b9c1b51eSKate Stone     case 'd':
2851b9c1b51eSKate Stone       return eFormatDecimal;
2852b9c1b51eSKate Stone     case 'D':
2853b9c1b51eSKate Stone       return eFormatDefault;
2854b9c1b51eSKate Stone     case 'i':
2855b9c1b51eSKate Stone       return eFormatInstruction;
2856b9c1b51eSKate Stone     case 'A':
2857b9c1b51eSKate Stone       return eFormatAddressInfo;
2858b9c1b51eSKate Stone     case 'p':
2859b9c1b51eSKate Stone       return eFormatPointer;
2860b9c1b51eSKate Stone     case 'c':
2861b9c1b51eSKate Stone       return eFormatChar;
2862b9c1b51eSKate Stone     case 'b':
2863b9c1b51eSKate Stone       return eFormatBinary;
2864b9c1b51eSKate Stone     case 'B':
2865b9c1b51eSKate Stone       return eFormatBytesWithASCII;
2866b9c1b51eSKate Stone     case 'f':
2867b9c1b51eSKate Stone       return eFormatFloat;
286844d93782SGreg Clayton     }
286944d93782SGreg Clayton     return eFormatDefault;
287044d93782SGreg Clayton   }
287144d93782SGreg Clayton 
2872b9c1b51eSKate Stone   bool DisplayRowObject(Window &window, Row &row, DisplayOptions &options,
2873b9c1b51eSKate Stone                         bool highlight, bool last_child) {
28748369b28dSGreg Clayton     ValueObject *valobj = row.value.GetSP().get();
287544d93782SGreg Clayton 
2876c5dac77aSEugene Zelenko     if (valobj == nullptr)
287744d93782SGreg Clayton       return false;
287844d93782SGreg Clayton 
2879b9c1b51eSKate Stone     const char *type_name =
2880b9c1b51eSKate Stone         options.show_types ? valobj->GetTypeName().GetCString() : nullptr;
288144d93782SGreg Clayton     const char *name = valobj->GetName().GetCString();
288244d93782SGreg Clayton     const char *value = valobj->GetValueAsCString();
288344d93782SGreg Clayton     const char *summary = valobj->GetSummaryAsCString();
288444d93782SGreg Clayton 
288544d93782SGreg Clayton     window.MoveCursor(row.x, row.y);
288644d93782SGreg Clayton 
288744d93782SGreg Clayton     row.DrawTree(window);
288844d93782SGreg Clayton 
288944d93782SGreg Clayton     if (highlight)
289044d93782SGreg Clayton       window.AttributeOn(A_REVERSE);
289144d93782SGreg Clayton 
289244d93782SGreg Clayton     if (type_name && type_name[0])
289344d93782SGreg Clayton       window.Printf("(%s) ", type_name);
289444d93782SGreg Clayton 
289544d93782SGreg Clayton     if (name && name[0])
289644d93782SGreg Clayton       window.PutCString(name);
289744d93782SGreg Clayton 
289844d93782SGreg Clayton     attr_t changd_attr = 0;
289944d93782SGreg Clayton     if (valobj->GetValueDidChange())
290044d93782SGreg Clayton       changd_attr = COLOR_PAIR(5) | A_BOLD;
290144d93782SGreg Clayton 
2902b9c1b51eSKate Stone     if (value && value[0]) {
290344d93782SGreg Clayton       window.PutCString(" = ");
290444d93782SGreg Clayton       if (changd_attr)
290544d93782SGreg Clayton         window.AttributeOn(changd_attr);
290644d93782SGreg Clayton       window.PutCString(value);
290744d93782SGreg Clayton       if (changd_attr)
290844d93782SGreg Clayton         window.AttributeOff(changd_attr);
290944d93782SGreg Clayton     }
291044d93782SGreg Clayton 
2911b9c1b51eSKate Stone     if (summary && summary[0]) {
291244d93782SGreg Clayton       window.PutChar(' ');
291344d93782SGreg Clayton       if (changd_attr)
291444d93782SGreg Clayton         window.AttributeOn(changd_attr);
291544d93782SGreg Clayton       window.PutCString(summary);
291644d93782SGreg Clayton       if (changd_attr)
291744d93782SGreg Clayton         window.AttributeOff(changd_attr);
291844d93782SGreg Clayton     }
291944d93782SGreg Clayton 
292044d93782SGreg Clayton     if (highlight)
292144d93782SGreg Clayton       window.AttributeOff(A_REVERSE);
292244d93782SGreg Clayton 
292344d93782SGreg Clayton     return true;
292444d93782SGreg Clayton   }
2925315b6884SEugene Zelenko 
2926b9c1b51eSKate Stone   void DisplayRows(Window &window, std::vector<Row> &rows,
2927b9c1b51eSKate Stone                    DisplayOptions &options) {
292844d93782SGreg Clayton     // >   0x25B7
292944d93782SGreg Clayton     // \/  0x25BD
293044d93782SGreg Clayton 
293144d93782SGreg Clayton     bool window_is_active = window.IsActive();
2932b9c1b51eSKate Stone     for (auto &row : rows) {
293344d93782SGreg Clayton       const bool last_child = row.parent && &rows[rows.size() - 1] == &row;
293444d93782SGreg Clayton       // Save the row index in each Row structure
293544d93782SGreg Clayton       row.row_idx = m_num_rows;
293644d93782SGreg Clayton       if ((m_num_rows >= m_first_visible_row) &&
2937b9c1b51eSKate Stone           ((m_num_rows - m_first_visible_row) <
2938b9c1b51eSKate Stone            static_cast<size_t>(NumVisibleRows()))) {
293944d93782SGreg Clayton         row.x = m_min_x;
294044d93782SGreg Clayton         row.y = m_num_rows - m_first_visible_row + 1;
2941b9c1b51eSKate Stone         if (DisplayRowObject(window, row, options,
2942b9c1b51eSKate Stone                              window_is_active &&
2943b9c1b51eSKate Stone                                  m_num_rows == m_selected_row_idx,
2944b9c1b51eSKate Stone                              last_child)) {
294544d93782SGreg Clayton           ++m_num_rows;
2946b9c1b51eSKate Stone         } else {
294744d93782SGreg Clayton           row.x = 0;
294844d93782SGreg Clayton           row.y = 0;
294944d93782SGreg Clayton         }
2950b9c1b51eSKate Stone       } else {
295144d93782SGreg Clayton         row.x = 0;
295244d93782SGreg Clayton         row.y = 0;
295344d93782SGreg Clayton         ++m_num_rows;
295444d93782SGreg Clayton       }
295544d93782SGreg Clayton 
29568369b28dSGreg Clayton       auto &children = row.GetChildren();
29578369b28dSGreg Clayton       if (row.expanded && !children.empty()) {
29588369b28dSGreg Clayton         DisplayRows(window, children, options);
295944d93782SGreg Clayton       }
296044d93782SGreg Clayton     }
296144d93782SGreg Clayton   }
296244d93782SGreg Clayton 
29638369b28dSGreg Clayton   int CalculateTotalNumberRows(std::vector<Row> &rows) {
296444d93782SGreg Clayton     int row_count = 0;
29658369b28dSGreg Clayton     for (auto &row : rows) {
296644d93782SGreg Clayton       ++row_count;
296744d93782SGreg Clayton       if (row.expanded)
29688369b28dSGreg Clayton         row_count += CalculateTotalNumberRows(row.GetChildren());
296944d93782SGreg Clayton     }
297044d93782SGreg Clayton     return row_count;
297144d93782SGreg Clayton   }
2972315b6884SEugene Zelenko 
2973b9c1b51eSKate Stone   static Row *GetRowForRowIndexImpl(std::vector<Row> &rows, size_t &row_index) {
2974b9c1b51eSKate Stone     for (auto &row : rows) {
297544d93782SGreg Clayton       if (row_index == 0)
297644d93782SGreg Clayton         return &row;
2977b9c1b51eSKate Stone       else {
297844d93782SGreg Clayton         --row_index;
29798369b28dSGreg Clayton         auto &children = row.GetChildren();
29808369b28dSGreg Clayton         if (row.expanded && !children.empty()) {
29818369b28dSGreg Clayton           Row *result = GetRowForRowIndexImpl(children, row_index);
298244d93782SGreg Clayton           if (result)
298344d93782SGreg Clayton             return result;
298444d93782SGreg Clayton         }
298544d93782SGreg Clayton       }
298644d93782SGreg Clayton     }
2987c5dac77aSEugene Zelenko     return nullptr;
298844d93782SGreg Clayton   }
298944d93782SGreg Clayton 
2990b9c1b51eSKate Stone   Row *GetRowForRowIndex(size_t row_index) {
299144d93782SGreg Clayton     return GetRowForRowIndexImpl(m_rows, row_index);
299244d93782SGreg Clayton   }
299344d93782SGreg Clayton 
2994b9c1b51eSKate Stone   int NumVisibleRows() const { return m_max_y - m_min_y; }
299544d93782SGreg Clayton 
299644d93782SGreg Clayton   static DisplayOptions g_options;
299744d93782SGreg Clayton };
299844d93782SGreg Clayton 
2999b9c1b51eSKate Stone class FrameVariablesWindowDelegate : public ValueObjectListDelegate {
300044d93782SGreg Clayton public:
3001b9c1b51eSKate Stone   FrameVariablesWindowDelegate(Debugger &debugger)
3002b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger),
3003b9c1b51eSKate Stone         m_frame_block(nullptr) {}
300444d93782SGreg Clayton 
3005315b6884SEugene Zelenko   ~FrameVariablesWindowDelegate() override = default;
300644d93782SGreg Clayton 
3007b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
300844d93782SGreg Clayton     return "Frame variable window keyboard shortcuts:";
300944d93782SGreg Clayton   }
301044d93782SGreg Clayton 
3011b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3012b9c1b51eSKate Stone     ExecutionContext exe_ctx(
3013b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
301444d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
3015c5dac77aSEugene Zelenko     Block *frame_block = nullptr;
3016c5dac77aSEugene Zelenko     StackFrame *frame = nullptr;
301744d93782SGreg Clayton 
3018b9c1b51eSKate Stone     if (process) {
301944d93782SGreg Clayton       StateType state = process->GetState();
3020b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
302144d93782SGreg Clayton         frame = exe_ctx.GetFramePtr();
302244d93782SGreg Clayton         if (frame)
302344d93782SGreg Clayton           frame_block = frame->GetFrameBlock();
3024b9c1b51eSKate Stone       } else if (StateIsRunningState(state)) {
302544d93782SGreg Clayton         return true; // Don't do any updating when we are running
302644d93782SGreg Clayton       }
302744d93782SGreg Clayton     }
302844d93782SGreg Clayton 
302944d93782SGreg Clayton     ValueObjectList local_values;
3030b9c1b51eSKate Stone     if (frame_block) {
303144d93782SGreg Clayton       // Only update the variables if they have changed
3032b9c1b51eSKate Stone       if (m_frame_block != frame_block) {
303344d93782SGreg Clayton         m_frame_block = frame_block;
303444d93782SGreg Clayton 
303544d93782SGreg Clayton         VariableList *locals = frame->GetVariableList(true);
3036b9c1b51eSKate Stone         if (locals) {
303744d93782SGreg Clayton           const DynamicValueType use_dynamic = eDynamicDontRunTarget;
303844d93782SGreg Clayton           const size_t num_locals = locals->GetSize();
3039b9c1b51eSKate Stone           for (size_t i = 0; i < num_locals; ++i) {
3040b9c1b51eSKate Stone             ValueObjectSP value_sp = frame->GetValueObjectForFrameVariable(
3041b9c1b51eSKate Stone                 locals->GetVariableAtIndex(i), use_dynamic);
3042b9c1b51eSKate Stone             if (value_sp) {
3043eb72dc7dSGreg Clayton               ValueObjectSP synthetic_value_sp = value_sp->GetSyntheticValue();
3044eb72dc7dSGreg Clayton               if (synthetic_value_sp)
3045eb72dc7dSGreg Clayton                 local_values.Append(synthetic_value_sp);
3046eb72dc7dSGreg Clayton               else
3047eb72dc7dSGreg Clayton                 local_values.Append(value_sp);
3048eb72dc7dSGreg Clayton             }
3049eb72dc7dSGreg Clayton           }
305044d93782SGreg Clayton           // Update the values
305144d93782SGreg Clayton           SetValues(local_values);
305244d93782SGreg Clayton         }
305344d93782SGreg Clayton       }
3054b9c1b51eSKate Stone     } else {
3055c5dac77aSEugene Zelenko       m_frame_block = nullptr;
305644d93782SGreg Clayton       // Update the values with an empty list if there is no frame
305744d93782SGreg Clayton       SetValues(local_values);
305844d93782SGreg Clayton     }
305944d93782SGreg Clayton 
306044d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
306144d93782SGreg Clayton   }
306244d93782SGreg Clayton 
306344d93782SGreg Clayton protected:
306444d93782SGreg Clayton   Debugger &m_debugger;
306544d93782SGreg Clayton   Block *m_frame_block;
306644d93782SGreg Clayton };
306744d93782SGreg Clayton 
3068b9c1b51eSKate Stone class RegistersWindowDelegate : public ValueObjectListDelegate {
306944d93782SGreg Clayton public:
3070b9c1b51eSKate Stone   RegistersWindowDelegate(Debugger &debugger)
3071b9c1b51eSKate Stone       : ValueObjectListDelegate(), m_debugger(debugger) {}
307244d93782SGreg Clayton 
3073315b6884SEugene Zelenko   ~RegistersWindowDelegate() override = default;
307444d93782SGreg Clayton 
3075b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
307644d93782SGreg Clayton     return "Register window keyboard shortcuts:";
307744d93782SGreg Clayton   }
307844d93782SGreg Clayton 
3079b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3080b9c1b51eSKate Stone     ExecutionContext exe_ctx(
3081b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext());
308244d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
308344d93782SGreg Clayton 
308444d93782SGreg Clayton     ValueObjectList value_list;
3085b9c1b51eSKate Stone     if (frame) {
3086b9c1b51eSKate Stone       if (frame->GetStackID() != m_stack_id) {
308744d93782SGreg Clayton         m_stack_id = frame->GetStackID();
308844d93782SGreg Clayton         RegisterContextSP reg_ctx(frame->GetRegisterContext());
3089b9c1b51eSKate Stone         if (reg_ctx) {
309044d93782SGreg Clayton           const uint32_t num_sets = reg_ctx->GetRegisterSetCount();
3091b9c1b51eSKate Stone           for (uint32_t set_idx = 0; set_idx < num_sets; ++set_idx) {
3092b9c1b51eSKate Stone             value_list.Append(
3093b9c1b51eSKate Stone                 ValueObjectRegisterSet::Create(frame, reg_ctx, set_idx));
309444d93782SGreg Clayton           }
309544d93782SGreg Clayton         }
309644d93782SGreg Clayton         SetValues(value_list);
309744d93782SGreg Clayton       }
3098b9c1b51eSKate Stone     } else {
309944d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
310044d93782SGreg Clayton       if (process && process->IsAlive())
310144d93782SGreg Clayton         return true; // Don't do any updating if we are running
3102b9c1b51eSKate Stone       else {
310305097246SAdrian Prantl         // Update the values with an empty list if there is no process or the
310405097246SAdrian Prantl         // process isn't alive anymore
310544d93782SGreg Clayton         SetValues(value_list);
310644d93782SGreg Clayton       }
310744d93782SGreg Clayton     }
310844d93782SGreg Clayton     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
310944d93782SGreg Clayton   }
311044d93782SGreg Clayton 
311144d93782SGreg Clayton protected:
311244d93782SGreg Clayton   Debugger &m_debugger;
311344d93782SGreg Clayton   StackID m_stack_id;
311444d93782SGreg Clayton };
311544d93782SGreg Clayton 
3116b9c1b51eSKate Stone static const char *CursesKeyToCString(int ch) {
311744d93782SGreg Clayton   static char g_desc[32];
3118b9c1b51eSKate Stone   if (ch >= KEY_F0 && ch < KEY_F0 + 64) {
311944d93782SGreg Clayton     snprintf(g_desc, sizeof(g_desc), "F%u", ch - KEY_F0);
312044d93782SGreg Clayton     return g_desc;
312144d93782SGreg Clayton   }
3122b9c1b51eSKate Stone   switch (ch) {
3123b9c1b51eSKate Stone   case KEY_DOWN:
3124b9c1b51eSKate Stone     return "down";
3125b9c1b51eSKate Stone   case KEY_UP:
3126b9c1b51eSKate Stone     return "up";
3127b9c1b51eSKate Stone   case KEY_LEFT:
3128b9c1b51eSKate Stone     return "left";
3129b9c1b51eSKate Stone   case KEY_RIGHT:
3130b9c1b51eSKate Stone     return "right";
3131b9c1b51eSKate Stone   case KEY_HOME:
3132b9c1b51eSKate Stone     return "home";
3133b9c1b51eSKate Stone   case KEY_BACKSPACE:
3134b9c1b51eSKate Stone     return "backspace";
3135b9c1b51eSKate Stone   case KEY_DL:
3136b9c1b51eSKate Stone     return "delete-line";
3137b9c1b51eSKate Stone   case KEY_IL:
3138b9c1b51eSKate Stone     return "insert-line";
3139b9c1b51eSKate Stone   case KEY_DC:
3140b9c1b51eSKate Stone     return "delete-char";
3141b9c1b51eSKate Stone   case KEY_IC:
3142b9c1b51eSKate Stone     return "insert-char";
3143b9c1b51eSKate Stone   case KEY_CLEAR:
3144b9c1b51eSKate Stone     return "clear";
3145b9c1b51eSKate Stone   case KEY_EOS:
3146b9c1b51eSKate Stone     return "clear-to-eos";
3147b9c1b51eSKate Stone   case KEY_EOL:
3148b9c1b51eSKate Stone     return "clear-to-eol";
3149b9c1b51eSKate Stone   case KEY_SF:
3150b9c1b51eSKate Stone     return "scroll-forward";
3151b9c1b51eSKate Stone   case KEY_SR:
3152b9c1b51eSKate Stone     return "scroll-backward";
3153b9c1b51eSKate Stone   case KEY_NPAGE:
3154b9c1b51eSKate Stone     return "page-down";
3155b9c1b51eSKate Stone   case KEY_PPAGE:
3156b9c1b51eSKate Stone     return "page-up";
3157b9c1b51eSKate Stone   case KEY_STAB:
3158b9c1b51eSKate Stone     return "set-tab";
3159b9c1b51eSKate Stone   case KEY_CTAB:
3160b9c1b51eSKate Stone     return "clear-tab";
3161b9c1b51eSKate Stone   case KEY_CATAB:
3162b9c1b51eSKate Stone     return "clear-all-tabs";
3163b9c1b51eSKate Stone   case KEY_ENTER:
3164b9c1b51eSKate Stone     return "enter";
3165b9c1b51eSKate Stone   case KEY_PRINT:
3166b9c1b51eSKate Stone     return "print";
3167b9c1b51eSKate Stone   case KEY_LL:
3168b9c1b51eSKate Stone     return "lower-left key";
3169b9c1b51eSKate Stone   case KEY_A1:
3170b9c1b51eSKate Stone     return "upper left of keypad";
3171b9c1b51eSKate Stone   case KEY_A3:
3172b9c1b51eSKate Stone     return "upper right of keypad";
3173b9c1b51eSKate Stone   case KEY_B2:
3174b9c1b51eSKate Stone     return "center of keypad";
3175b9c1b51eSKate Stone   case KEY_C1:
3176b9c1b51eSKate Stone     return "lower left of keypad";
3177b9c1b51eSKate Stone   case KEY_C3:
3178b9c1b51eSKate Stone     return "lower right of keypad";
3179b9c1b51eSKate Stone   case KEY_BTAB:
3180b9c1b51eSKate Stone     return "back-tab key";
3181b9c1b51eSKate Stone   case KEY_BEG:
3182b9c1b51eSKate Stone     return "begin key";
3183b9c1b51eSKate Stone   case KEY_CANCEL:
3184b9c1b51eSKate Stone     return "cancel key";
3185b9c1b51eSKate Stone   case KEY_CLOSE:
3186b9c1b51eSKate Stone     return "close key";
3187b9c1b51eSKate Stone   case KEY_COMMAND:
3188b9c1b51eSKate Stone     return "command key";
3189b9c1b51eSKate Stone   case KEY_COPY:
3190b9c1b51eSKate Stone     return "copy key";
3191b9c1b51eSKate Stone   case KEY_CREATE:
3192b9c1b51eSKate Stone     return "create key";
3193b9c1b51eSKate Stone   case KEY_END:
3194b9c1b51eSKate Stone     return "end key";
3195b9c1b51eSKate Stone   case KEY_EXIT:
3196b9c1b51eSKate Stone     return "exit key";
3197b9c1b51eSKate Stone   case KEY_FIND:
3198b9c1b51eSKate Stone     return "find key";
3199b9c1b51eSKate Stone   case KEY_HELP:
3200b9c1b51eSKate Stone     return "help key";
3201b9c1b51eSKate Stone   case KEY_MARK:
3202b9c1b51eSKate Stone     return "mark key";
3203b9c1b51eSKate Stone   case KEY_MESSAGE:
3204b9c1b51eSKate Stone     return "message key";
3205b9c1b51eSKate Stone   case KEY_MOVE:
3206b9c1b51eSKate Stone     return "move key";
3207b9c1b51eSKate Stone   case KEY_NEXT:
3208b9c1b51eSKate Stone     return "next key";
3209b9c1b51eSKate Stone   case KEY_OPEN:
3210b9c1b51eSKate Stone     return "open key";
3211b9c1b51eSKate Stone   case KEY_OPTIONS:
3212b9c1b51eSKate Stone     return "options key";
3213b9c1b51eSKate Stone   case KEY_PREVIOUS:
3214b9c1b51eSKate Stone     return "previous key";
3215b9c1b51eSKate Stone   case KEY_REDO:
3216b9c1b51eSKate Stone     return "redo key";
3217b9c1b51eSKate Stone   case KEY_REFERENCE:
3218b9c1b51eSKate Stone     return "reference key";
3219b9c1b51eSKate Stone   case KEY_REFRESH:
3220b9c1b51eSKate Stone     return "refresh key";
3221b9c1b51eSKate Stone   case KEY_REPLACE:
3222b9c1b51eSKate Stone     return "replace key";
3223b9c1b51eSKate Stone   case KEY_RESTART:
3224b9c1b51eSKate Stone     return "restart key";
3225b9c1b51eSKate Stone   case KEY_RESUME:
3226b9c1b51eSKate Stone     return "resume key";
3227b9c1b51eSKate Stone   case KEY_SAVE:
3228b9c1b51eSKate Stone     return "save key";
3229b9c1b51eSKate Stone   case KEY_SBEG:
3230b9c1b51eSKate Stone     return "shifted begin key";
3231b9c1b51eSKate Stone   case KEY_SCANCEL:
3232b9c1b51eSKate Stone     return "shifted cancel key";
3233b9c1b51eSKate Stone   case KEY_SCOMMAND:
3234b9c1b51eSKate Stone     return "shifted command key";
3235b9c1b51eSKate Stone   case KEY_SCOPY:
3236b9c1b51eSKate Stone     return "shifted copy key";
3237b9c1b51eSKate Stone   case KEY_SCREATE:
3238b9c1b51eSKate Stone     return "shifted create key";
3239b9c1b51eSKate Stone   case KEY_SDC:
3240b9c1b51eSKate Stone     return "shifted delete-character key";
3241b9c1b51eSKate Stone   case KEY_SDL:
3242b9c1b51eSKate Stone     return "shifted delete-line key";
3243b9c1b51eSKate Stone   case KEY_SELECT:
3244b9c1b51eSKate Stone     return "select key";
3245b9c1b51eSKate Stone   case KEY_SEND:
3246b9c1b51eSKate Stone     return "shifted end key";
3247b9c1b51eSKate Stone   case KEY_SEOL:
3248b9c1b51eSKate Stone     return "shifted clear-to-end-of-line key";
3249b9c1b51eSKate Stone   case KEY_SEXIT:
3250b9c1b51eSKate Stone     return "shifted exit key";
3251b9c1b51eSKate Stone   case KEY_SFIND:
3252b9c1b51eSKate Stone     return "shifted find key";
3253b9c1b51eSKate Stone   case KEY_SHELP:
3254b9c1b51eSKate Stone     return "shifted help key";
3255b9c1b51eSKate Stone   case KEY_SHOME:
3256b9c1b51eSKate Stone     return "shifted home key";
3257b9c1b51eSKate Stone   case KEY_SIC:
3258b9c1b51eSKate Stone     return "shifted insert-character key";
3259b9c1b51eSKate Stone   case KEY_SLEFT:
3260b9c1b51eSKate Stone     return "shifted left-arrow key";
3261b9c1b51eSKate Stone   case KEY_SMESSAGE:
3262b9c1b51eSKate Stone     return "shifted message key";
3263b9c1b51eSKate Stone   case KEY_SMOVE:
3264b9c1b51eSKate Stone     return "shifted move key";
3265b9c1b51eSKate Stone   case KEY_SNEXT:
3266b9c1b51eSKate Stone     return "shifted next key";
3267b9c1b51eSKate Stone   case KEY_SOPTIONS:
3268b9c1b51eSKate Stone     return "shifted options key";
3269b9c1b51eSKate Stone   case KEY_SPREVIOUS:
3270b9c1b51eSKate Stone     return "shifted previous key";
3271b9c1b51eSKate Stone   case KEY_SPRINT:
3272b9c1b51eSKate Stone     return "shifted print key";
3273b9c1b51eSKate Stone   case KEY_SREDO:
3274b9c1b51eSKate Stone     return "shifted redo key";
3275b9c1b51eSKate Stone   case KEY_SREPLACE:
3276b9c1b51eSKate Stone     return "shifted replace key";
3277b9c1b51eSKate Stone   case KEY_SRIGHT:
3278b9c1b51eSKate Stone     return "shifted right-arrow key";
3279b9c1b51eSKate Stone   case KEY_SRSUME:
3280b9c1b51eSKate Stone     return "shifted resume key";
3281b9c1b51eSKate Stone   case KEY_SSAVE:
3282b9c1b51eSKate Stone     return "shifted save key";
3283b9c1b51eSKate Stone   case KEY_SSUSPEND:
3284b9c1b51eSKate Stone     return "shifted suspend key";
3285b9c1b51eSKate Stone   case KEY_SUNDO:
3286b9c1b51eSKate Stone     return "shifted undo key";
3287b9c1b51eSKate Stone   case KEY_SUSPEND:
3288b9c1b51eSKate Stone     return "suspend key";
3289b9c1b51eSKate Stone   case KEY_UNDO:
3290b9c1b51eSKate Stone     return "undo key";
3291b9c1b51eSKate Stone   case KEY_MOUSE:
3292b9c1b51eSKate Stone     return "Mouse event has occurred";
3293b9c1b51eSKate Stone   case KEY_RESIZE:
3294b9c1b51eSKate Stone     return "Terminal resize event";
329527801f4fSBruce Mitchener #ifdef KEY_EVENT
3296b9c1b51eSKate Stone   case KEY_EVENT:
3297b9c1b51eSKate Stone     return "We were interrupted by an event";
329827801f4fSBruce Mitchener #endif
3299b9c1b51eSKate Stone   case KEY_RETURN:
3300b9c1b51eSKate Stone     return "return";
3301b9c1b51eSKate Stone   case ' ':
3302b9c1b51eSKate Stone     return "space";
3303b9c1b51eSKate Stone   case '\t':
3304b9c1b51eSKate Stone     return "tab";
3305b9c1b51eSKate Stone   case KEY_ESCAPE:
3306b9c1b51eSKate Stone     return "escape";
330744d93782SGreg Clayton   default:
330844d93782SGreg Clayton     if (isprint(ch))
330944d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "%c", ch);
331044d93782SGreg Clayton     else
331144d93782SGreg Clayton       snprintf(g_desc, sizeof(g_desc), "\\x%2.2x", ch);
331244d93782SGreg Clayton     return g_desc;
331344d93782SGreg Clayton   }
3314c5dac77aSEugene Zelenko   return nullptr;
331544d93782SGreg Clayton }
331644d93782SGreg Clayton 
3317b9c1b51eSKate Stone HelpDialogDelegate::HelpDialogDelegate(const char *text,
3318b9c1b51eSKate Stone                                        KeyHelp *key_help_array)
3319b9c1b51eSKate Stone     : m_text(), m_first_visible_line(0) {
3320b9c1b51eSKate Stone   if (text && text[0]) {
332144d93782SGreg Clayton     m_text.SplitIntoLines(text);
332244d93782SGreg Clayton     m_text.AppendString("");
332344d93782SGreg Clayton   }
3324b9c1b51eSKate Stone   if (key_help_array) {
3325b9c1b51eSKate Stone     for (KeyHelp *key = key_help_array; key->ch; ++key) {
332644d93782SGreg Clayton       StreamString key_description;
3327b9c1b51eSKate Stone       key_description.Printf("%10s - %s", CursesKeyToCString(key->ch),
3328b9c1b51eSKate Stone                              key->description);
3329c156427dSZachary Turner       m_text.AppendString(key_description.GetString());
333044d93782SGreg Clayton     }
333144d93782SGreg Clayton   }
333244d93782SGreg Clayton }
333344d93782SGreg Clayton 
3334315b6884SEugene Zelenko HelpDialogDelegate::~HelpDialogDelegate() = default;
333544d93782SGreg Clayton 
3336b9c1b51eSKate Stone bool HelpDialogDelegate::WindowDelegateDraw(Window &window, bool force) {
333744d93782SGreg Clayton   window.Erase();
333844d93782SGreg Clayton   const int window_height = window.GetHeight();
333944d93782SGreg Clayton   int x = 2;
334044d93782SGreg Clayton   int y = 1;
334144d93782SGreg Clayton   const int min_y = y;
334244d93782SGreg Clayton   const int max_y = window_height - 1 - y;
33433985c8c6SSaleem Abdulrasool   const size_t num_visible_lines = max_y - min_y + 1;
334444d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
334544d93782SGreg Clayton   const char *bottom_message;
334644d93782SGreg Clayton   if (num_lines <= num_visible_lines)
334744d93782SGreg Clayton     bottom_message = "Press any key to exit";
334844d93782SGreg Clayton   else
334944d93782SGreg Clayton     bottom_message = "Use arrows to scroll, any other key to exit";
335044d93782SGreg Clayton   window.DrawTitleBox(window.GetName(), bottom_message);
3351b9c1b51eSKate Stone   while (y <= max_y) {
335244d93782SGreg Clayton     window.MoveCursor(x, y);
3353b9c1b51eSKate Stone     window.PutCStringTruncated(
3354b9c1b51eSKate Stone         m_text.GetStringAtIndex(m_first_visible_line + y - min_y), 1);
335544d93782SGreg Clayton     ++y;
335644d93782SGreg Clayton   }
335744d93782SGreg Clayton   return true;
335844d93782SGreg Clayton }
335944d93782SGreg Clayton 
3360b9c1b51eSKate Stone HandleCharResult HelpDialogDelegate::WindowDelegateHandleChar(Window &window,
3361b9c1b51eSKate Stone                                                               int key) {
336244d93782SGreg Clayton   bool done = false;
336344d93782SGreg Clayton   const size_t num_lines = m_text.GetSize();
336444d93782SGreg Clayton   const size_t num_visible_lines = window.GetHeight() - 2;
336544d93782SGreg Clayton 
3366b9c1b51eSKate Stone   if (num_lines <= num_visible_lines) {
336744d93782SGreg Clayton     done = true;
336805097246SAdrian Prantl     // If we have all lines visible and don't need scrolling, then any key
336905097246SAdrian Prantl     // press will cause us to exit
3370b9c1b51eSKate Stone   } else {
3371b9c1b51eSKate Stone     switch (key) {
337244d93782SGreg Clayton     case KEY_UP:
337344d93782SGreg Clayton       if (m_first_visible_line > 0)
337444d93782SGreg Clayton         --m_first_visible_line;
337544d93782SGreg Clayton       break;
337644d93782SGreg Clayton 
337744d93782SGreg Clayton     case KEY_DOWN:
337844d93782SGreg Clayton       if (m_first_visible_line + num_visible_lines < num_lines)
337944d93782SGreg Clayton         ++m_first_visible_line;
338044d93782SGreg Clayton       break;
338144d93782SGreg Clayton 
338244d93782SGreg Clayton     case KEY_PPAGE:
338344d93782SGreg Clayton     case ',':
3384b9c1b51eSKate Stone       if (m_first_visible_line > 0) {
33853985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_visible_lines)
338644d93782SGreg Clayton           m_first_visible_line -= num_visible_lines;
338744d93782SGreg Clayton         else
338844d93782SGreg Clayton           m_first_visible_line = 0;
338944d93782SGreg Clayton       }
339044d93782SGreg Clayton       break;
3391315b6884SEugene Zelenko 
339244d93782SGreg Clayton     case KEY_NPAGE:
339344d93782SGreg Clayton     case '.':
3394b9c1b51eSKate Stone       if (m_first_visible_line + num_visible_lines < num_lines) {
339544d93782SGreg Clayton         m_first_visible_line += num_visible_lines;
33963985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > num_lines)
339744d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
339844d93782SGreg Clayton       }
339944d93782SGreg Clayton       break;
3400315b6884SEugene Zelenko 
340144d93782SGreg Clayton     default:
340244d93782SGreg Clayton       done = true;
340344d93782SGreg Clayton       break;
340444d93782SGreg Clayton     }
340544d93782SGreg Clayton   }
340644d93782SGreg Clayton   if (done)
340744d93782SGreg Clayton     window.GetParent()->RemoveSubWindow(&window);
340844d93782SGreg Clayton   return eKeyHandled;
340944d93782SGreg Clayton }
341044d93782SGreg Clayton 
3411b9c1b51eSKate Stone class ApplicationDelegate : public WindowDelegate, public MenuDelegate {
341244d93782SGreg Clayton public:
341344d93782SGreg Clayton   enum {
341444d93782SGreg Clayton     eMenuID_LLDB = 1,
341544d93782SGreg Clayton     eMenuID_LLDBAbout,
341644d93782SGreg Clayton     eMenuID_LLDBExit,
341744d93782SGreg Clayton 
341844d93782SGreg Clayton     eMenuID_Target,
341944d93782SGreg Clayton     eMenuID_TargetCreate,
342044d93782SGreg Clayton     eMenuID_TargetDelete,
342144d93782SGreg Clayton 
342244d93782SGreg Clayton     eMenuID_Process,
342344d93782SGreg Clayton     eMenuID_ProcessAttach,
342444d93782SGreg Clayton     eMenuID_ProcessDetach,
342544d93782SGreg Clayton     eMenuID_ProcessLaunch,
342644d93782SGreg Clayton     eMenuID_ProcessContinue,
342744d93782SGreg Clayton     eMenuID_ProcessHalt,
342844d93782SGreg Clayton     eMenuID_ProcessKill,
342944d93782SGreg Clayton 
343044d93782SGreg Clayton     eMenuID_Thread,
343144d93782SGreg Clayton     eMenuID_ThreadStepIn,
343244d93782SGreg Clayton     eMenuID_ThreadStepOver,
343344d93782SGreg Clayton     eMenuID_ThreadStepOut,
343444d93782SGreg Clayton 
343544d93782SGreg Clayton     eMenuID_View,
343644d93782SGreg Clayton     eMenuID_ViewBacktrace,
343744d93782SGreg Clayton     eMenuID_ViewRegisters,
343844d93782SGreg Clayton     eMenuID_ViewSource,
343944d93782SGreg Clayton     eMenuID_ViewVariables,
344044d93782SGreg Clayton 
344144d93782SGreg Clayton     eMenuID_Help,
344244d93782SGreg Clayton     eMenuID_HelpGUIHelp
344344d93782SGreg Clayton   };
344444d93782SGreg Clayton 
3445b9c1b51eSKate Stone   ApplicationDelegate(Application &app, Debugger &debugger)
3446b9c1b51eSKate Stone       : WindowDelegate(), MenuDelegate(), m_app(app), m_debugger(debugger) {}
344744d93782SGreg Clayton 
3448315b6884SEugene Zelenko   ~ApplicationDelegate() override = default;
3449bd5ae6b4SGreg Clayton 
3450b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
345144d93782SGreg Clayton     return false; // Drawing not handled, let standard window drawing happen
345244d93782SGreg Clayton   }
345344d93782SGreg Clayton 
3454b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override {
3455b9c1b51eSKate Stone     switch (key) {
34565fdb09bbSGreg Clayton     case '\t':
345744d93782SGreg Clayton       window.SelectNextWindowAsActive();
345844d93782SGreg Clayton       return eKeyHandled;
34595fdb09bbSGreg Clayton 
34605fdb09bbSGreg Clayton     case 'h':
34615fdb09bbSGreg Clayton       window.CreateHelpSubwindow();
34625fdb09bbSGreg Clayton       return eKeyHandled;
34635fdb09bbSGreg Clayton 
34645fdb09bbSGreg Clayton     case KEY_ESCAPE:
34655fdb09bbSGreg Clayton       return eQuitApplication;
34665fdb09bbSGreg Clayton 
34675fdb09bbSGreg Clayton     default:
34685fdb09bbSGreg Clayton       break;
346944d93782SGreg Clayton     }
347044d93782SGreg Clayton     return eKeyNotHandled;
347144d93782SGreg Clayton   }
347244d93782SGreg Clayton 
3473b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
34745fdb09bbSGreg Clayton     return "Welcome to the LLDB curses GUI.\n\n"
34755fdb09bbSGreg Clayton            "Press the TAB key to change the selected view.\n"
3476b9c1b51eSKate Stone            "Each view has its own keyboard shortcuts, press 'h' to open a "
3477b9c1b51eSKate Stone            "dialog to display them.\n\n"
34785fdb09bbSGreg Clayton            "Common key bindings for all views:";
34795fdb09bbSGreg Clayton   }
34805fdb09bbSGreg Clayton 
3481b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
34825fdb09bbSGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
34835fdb09bbSGreg Clayton         {'\t', "Select next view"},
34845fdb09bbSGreg Clayton         {'h', "Show help dialog with view specific key bindings"},
34855fdb09bbSGreg Clayton         {',', "Page up"},
34865fdb09bbSGreg Clayton         {'.', "Page down"},
34875fdb09bbSGreg Clayton         {KEY_UP, "Select previous"},
34885fdb09bbSGreg Clayton         {KEY_DOWN, "Select next"},
34895fdb09bbSGreg Clayton         {KEY_LEFT, "Unexpand or select parent"},
34905fdb09bbSGreg Clayton         {KEY_RIGHT, "Expand"},
34915fdb09bbSGreg Clayton         {KEY_PPAGE, "Page up"},
34925fdb09bbSGreg Clayton         {KEY_NPAGE, "Page down"},
3493b9c1b51eSKate Stone         {'\0', nullptr}};
34945fdb09bbSGreg Clayton     return g_source_view_key_help;
34955fdb09bbSGreg Clayton   }
34965fdb09bbSGreg Clayton 
3497b9c1b51eSKate Stone   MenuActionResult MenuDelegateAction(Menu &menu) override {
3498b9c1b51eSKate Stone     switch (menu.GetIdentifier()) {
3499b9c1b51eSKate Stone     case eMenuID_ThreadStepIn: {
3500b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3501b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3502b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
350344d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3504b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3505b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
35064b4b2478SJim Ingham           exe_ctx.GetThreadRef().StepIn(true);
350744d93782SGreg Clayton       }
350844d93782SGreg Clayton     }
350944d93782SGreg Clayton       return MenuActionResult::Handled;
351044d93782SGreg Clayton 
3511b9c1b51eSKate Stone     case eMenuID_ThreadStepOut: {
3512b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3513b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3514b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
351544d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3516b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3517b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
351844d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
351944d93782SGreg Clayton       }
352044d93782SGreg Clayton     }
352144d93782SGreg Clayton       return MenuActionResult::Handled;
352244d93782SGreg Clayton 
3523b9c1b51eSKate Stone     case eMenuID_ThreadStepOver: {
3524b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3525b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3526b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope()) {
352744d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3528b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3529b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
353044d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOver(true);
353144d93782SGreg Clayton       }
353244d93782SGreg Clayton     }
353344d93782SGreg Clayton       return MenuActionResult::Handled;
353444d93782SGreg Clayton 
3535b9c1b51eSKate Stone     case eMenuID_ProcessContinue: {
3536b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3537b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3538b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
353944d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
3540b9c1b51eSKate Stone         if (process && process->IsAlive() &&
3541b9c1b51eSKate Stone             StateIsStoppedState(process->GetState(), true))
354244d93782SGreg Clayton           process->Resume();
354344d93782SGreg Clayton       }
354444d93782SGreg Clayton     }
354544d93782SGreg Clayton       return MenuActionResult::Handled;
354644d93782SGreg Clayton 
3547b9c1b51eSKate Stone     case eMenuID_ProcessKill: {
3548b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3549b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3550b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
355144d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
355244d93782SGreg Clayton         if (process && process->IsAlive())
3553ede3193bSJason Molenda           process->Destroy(false);
355444d93782SGreg Clayton       }
355544d93782SGreg Clayton     }
355644d93782SGreg Clayton       return MenuActionResult::Handled;
355744d93782SGreg Clayton 
3558b9c1b51eSKate Stone     case eMenuID_ProcessHalt: {
3559b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3560b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3561b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
356244d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
356344d93782SGreg Clayton         if (process && process->IsAlive())
356444d93782SGreg Clayton           process->Halt();
356544d93782SGreg Clayton       }
356644d93782SGreg Clayton     }
356744d93782SGreg Clayton       return MenuActionResult::Handled;
356844d93782SGreg Clayton 
3569b9c1b51eSKate Stone     case eMenuID_ProcessDetach: {
3570b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3571b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
3572b9c1b51eSKate Stone       if (exe_ctx.HasProcessScope()) {
357344d93782SGreg Clayton         Process *process = exe_ctx.GetProcessPtr();
357444d93782SGreg Clayton         if (process && process->IsAlive())
357544d93782SGreg Clayton           process->Detach(false);
357644d93782SGreg Clayton       }
357744d93782SGreg Clayton     }
357844d93782SGreg Clayton       return MenuActionResult::Handled;
357944d93782SGreg Clayton 
3580b9c1b51eSKate Stone     case eMenuID_Process: {
3581b9c1b51eSKate Stone       // Populate the menu with all of the threads if the process is stopped
358205097246SAdrian Prantl       // when the Process menu gets selected and is about to display its
358305097246SAdrian Prantl       // submenu.
358444d93782SGreg Clayton       Menus &submenus = menu.GetSubmenus();
3585b9c1b51eSKate Stone       ExecutionContext exe_ctx =
3586b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
358744d93782SGreg Clayton       Process *process = exe_ctx.GetProcessPtr();
3588b9c1b51eSKate Stone       if (process && process->IsAlive() &&
3589b9c1b51eSKate Stone           StateIsStoppedState(process->GetState(), true)) {
359044d93782SGreg Clayton         if (submenus.size() == 7)
359144d93782SGreg Clayton           menu.AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
359244d93782SGreg Clayton         else if (submenus.size() > 8)
359344d93782SGreg Clayton           submenus.erase(submenus.begin() + 8, submenus.end());
359444d93782SGreg Clayton 
359544d93782SGreg Clayton         ThreadList &threads = process->GetThreadList();
3596bb19a13cSSaleem Abdulrasool         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
359744d93782SGreg Clayton         size_t num_threads = threads.GetSize();
3598b9c1b51eSKate Stone         for (size_t i = 0; i < num_threads; ++i) {
359944d93782SGreg Clayton           ThreadSP thread_sp = threads.GetThreadAtIndex(i);
360044d93782SGreg Clayton           char menu_char = '\0';
360144d93782SGreg Clayton           if (i < 9)
360244d93782SGreg Clayton             menu_char = '1' + i;
360344d93782SGreg Clayton           StreamString thread_menu_title;
360444d93782SGreg Clayton           thread_menu_title.Printf("Thread %u", thread_sp->GetIndexID());
360544d93782SGreg Clayton           const char *thread_name = thread_sp->GetName();
360644d93782SGreg Clayton           if (thread_name && thread_name[0])
360744d93782SGreg Clayton             thread_menu_title.Printf(" %s", thread_name);
3608b9c1b51eSKate Stone           else {
360944d93782SGreg Clayton             const char *queue_name = thread_sp->GetQueueName();
361044d93782SGreg Clayton             if (queue_name && queue_name[0])
361144d93782SGreg Clayton               thread_menu_title.Printf(" %s", queue_name);
361244d93782SGreg Clayton           }
3613b9c1b51eSKate Stone           menu.AddSubmenu(
3614c156427dSZachary Turner               MenuSP(new Menu(thread_menu_title.GetString().str().c_str(),
3615c156427dSZachary Turner                               nullptr, menu_char, thread_sp->GetID())));
361644d93782SGreg Clayton         }
3617b9c1b51eSKate Stone       } else if (submenus.size() > 7) {
361805097246SAdrian Prantl         // Remove the separator and any other thread submenu items that were
361905097246SAdrian Prantl         // previously added
362044d93782SGreg Clayton         submenus.erase(submenus.begin() + 7, submenus.end());
362144d93782SGreg Clayton       }
3622b9c1b51eSKate Stone       // Since we are adding and removing items we need to recalculate the name
3623b9c1b51eSKate Stone       // lengths
362444d93782SGreg Clayton       menu.RecalculateNameLengths();
362544d93782SGreg Clayton     }
362644d93782SGreg Clayton       return MenuActionResult::Handled;
362744d93782SGreg Clayton 
3628b9c1b51eSKate Stone     case eMenuID_ViewVariables: {
362944d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
363044d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
363144d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
363244d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
363344d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
363444d93782SGreg Clayton 
3635b9c1b51eSKate Stone       if (variables_window_sp) {
363644d93782SGreg Clayton         const Rect variables_bounds = variables_window_sp->GetBounds();
363744d93782SGreg Clayton 
363844d93782SGreg Clayton         main_window_sp->RemoveSubWindow(variables_window_sp.get());
363944d93782SGreg Clayton 
3640b9c1b51eSKate Stone         if (registers_window_sp) {
3641b9c1b51eSKate Stone           // We have a registers window, so give all the area back to the
3642b9c1b51eSKate Stone           // registers window
364344d93782SGreg Clayton           Rect registers_bounds = variables_bounds;
364444d93782SGreg Clayton           registers_bounds.size.width = source_bounds.size.width;
364544d93782SGreg Clayton           registers_window_sp->SetBounds(registers_bounds);
3646b9c1b51eSKate Stone         } else {
364705097246SAdrian Prantl           // We have no registers window showing so give the bottom area back
364805097246SAdrian Prantl           // to the source view
364944d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3650b9c1b51eSKate Stone                                    source_bounds.size.height +
3651b9c1b51eSKate Stone                                        variables_bounds.size.height);
365244d93782SGreg Clayton         }
3653b9c1b51eSKate Stone       } else {
365444d93782SGreg Clayton         Rect new_variables_rect;
3655b9c1b51eSKate Stone         if (registers_window_sp) {
365644d93782SGreg Clayton           // We have a registers window so split the area of the registers
365744d93782SGreg Clayton           // window into two columns where the left hand side will be the
365844d93782SGreg Clayton           // variables and the right hand side will be the registers
365944d93782SGreg Clayton           const Rect variables_bounds = registers_window_sp->GetBounds();
366044d93782SGreg Clayton           Rect new_registers_rect;
3661b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_variables_rect,
3662b9c1b51eSKate Stone                                                    new_registers_rect);
366344d93782SGreg Clayton           registers_window_sp->SetBounds(new_registers_rect);
3664b9c1b51eSKate Stone         } else {
366544d93782SGreg Clayton           // No variables window, grab the bottom part of the source window
366644d93782SGreg Clayton           Rect new_source_rect;
3667b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3668b9c1b51eSKate Stone                                                   new_variables_rect);
366944d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
367044d93782SGreg Clayton         }
3671b9c1b51eSKate Stone         WindowSP new_window_sp = main_window_sp->CreateSubWindow(
3672b9c1b51eSKate Stone             "Variables", new_variables_rect, false);
3673b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3674b9c1b51eSKate Stone             WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
367544d93782SGreg Clayton       }
367644d93782SGreg Clayton       touchwin(stdscr);
367744d93782SGreg Clayton     }
367844d93782SGreg Clayton       return MenuActionResult::Handled;
367944d93782SGreg Clayton 
3680b9c1b51eSKate Stone     case eMenuID_ViewRegisters: {
368144d93782SGreg Clayton       WindowSP main_window_sp = m_app.GetMainWindow();
368244d93782SGreg Clayton       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
368344d93782SGreg Clayton       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
368444d93782SGreg Clayton       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
368544d93782SGreg Clayton       const Rect source_bounds = source_window_sp->GetBounds();
368644d93782SGreg Clayton 
3687b9c1b51eSKate Stone       if (registers_window_sp) {
3688b9c1b51eSKate Stone         if (variables_window_sp) {
368944d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
369044d93782SGreg Clayton 
3691b9c1b51eSKate Stone           // We have a variables window, so give all the area back to the
3692b9c1b51eSKate Stone           // variables window
3693b9c1b51eSKate Stone           variables_window_sp->Resize(variables_bounds.size.width +
3694b9c1b51eSKate Stone                                           registers_window_sp->GetWidth(),
369544d93782SGreg Clayton                                       variables_bounds.size.height);
3696b9c1b51eSKate Stone         } else {
369705097246SAdrian Prantl           // We have no variables window showing so give the bottom area back
369805097246SAdrian Prantl           // to the source view
369944d93782SGreg Clayton           source_window_sp->Resize(source_bounds.size.width,
3700b9c1b51eSKate Stone                                    source_bounds.size.height +
3701b9c1b51eSKate Stone                                        registers_window_sp->GetHeight());
370244d93782SGreg Clayton         }
370344d93782SGreg Clayton         main_window_sp->RemoveSubWindow(registers_window_sp.get());
3704b9c1b51eSKate Stone       } else {
370544d93782SGreg Clayton         Rect new_regs_rect;
3706b9c1b51eSKate Stone         if (variables_window_sp) {
370705097246SAdrian Prantl           // We have a variables window, split it into two columns where the
370805097246SAdrian Prantl           // left hand side will be the variables and the right hand side will
370905097246SAdrian Prantl           // be the registers
371044d93782SGreg Clayton           const Rect variables_bounds = variables_window_sp->GetBounds();
371144d93782SGreg Clayton           Rect new_vars_rect;
3712b9c1b51eSKate Stone           variables_bounds.VerticalSplitPercentage(0.50, new_vars_rect,
3713b9c1b51eSKate Stone                                                    new_regs_rect);
371444d93782SGreg Clayton           variables_window_sp->SetBounds(new_vars_rect);
3715b9c1b51eSKate Stone         } else {
371644d93782SGreg Clayton           // No registers window, grab the bottom part of the source window
371744d93782SGreg Clayton           Rect new_source_rect;
3718b9c1b51eSKate Stone           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
3719b9c1b51eSKate Stone                                                   new_regs_rect);
372044d93782SGreg Clayton           source_window_sp->SetBounds(new_source_rect);
372144d93782SGreg Clayton         }
3722b9c1b51eSKate Stone         WindowSP new_window_sp =
3723b9c1b51eSKate Stone             main_window_sp->CreateSubWindow("Registers", new_regs_rect, false);
3724b9c1b51eSKate Stone         new_window_sp->SetDelegate(
3725b9c1b51eSKate Stone             WindowDelegateSP(new RegistersWindowDelegate(m_debugger)));
372644d93782SGreg Clayton       }
372744d93782SGreg Clayton       touchwin(stdscr);
372844d93782SGreg Clayton     }
372944d93782SGreg Clayton       return MenuActionResult::Handled;
373044d93782SGreg Clayton 
373144d93782SGreg Clayton     case eMenuID_HelpGUIHelp:
37325fdb09bbSGreg Clayton       m_app.GetMainWindow()->CreateHelpSubwindow();
373344d93782SGreg Clayton       return MenuActionResult::Handled;
373444d93782SGreg Clayton 
373544d93782SGreg Clayton     default:
373644d93782SGreg Clayton       break;
373744d93782SGreg Clayton     }
373844d93782SGreg Clayton 
373944d93782SGreg Clayton     return MenuActionResult::NotHandled;
374044d93782SGreg Clayton   }
3741b9c1b51eSKate Stone 
374244d93782SGreg Clayton protected:
374344d93782SGreg Clayton   Application &m_app;
374444d93782SGreg Clayton   Debugger &m_debugger;
374544d93782SGreg Clayton };
374644d93782SGreg Clayton 
3747b9c1b51eSKate Stone class StatusBarWindowDelegate : public WindowDelegate {
374844d93782SGreg Clayton public:
3749b9c1b51eSKate Stone   StatusBarWindowDelegate(Debugger &debugger) : m_debugger(debugger) {
3750b9c1b51eSKate Stone     FormatEntity::Parse("Thread: ${thread.id%tid}", m_format);
375144d93782SGreg Clayton   }
375244d93782SGreg Clayton 
3753315b6884SEugene Zelenko   ~StatusBarWindowDelegate() override = default;
3754bd5ae6b4SGreg Clayton 
3755b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3756b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3757b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
375844d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
375944d93782SGreg Clayton     Thread *thread = exe_ctx.GetThreadPtr();
376044d93782SGreg Clayton     StackFrame *frame = exe_ctx.GetFramePtr();
376144d93782SGreg Clayton     window.Erase();
376244d93782SGreg Clayton     window.SetBackground(2);
376344d93782SGreg Clayton     window.MoveCursor(0, 0);
3764b9c1b51eSKate Stone     if (process) {
376544d93782SGreg Clayton       const StateType state = process->GetState();
3766b9c1b51eSKate Stone       window.Printf("Process: %5" PRIu64 " %10s", process->GetID(),
3767b9c1b51eSKate Stone                     StateAsCString(state));
376844d93782SGreg Clayton 
3769b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
37705b031ebcSEd Maste         StreamString strm;
3771b9c1b51eSKate Stone         if (thread && FormatEntity::Format(m_format, strm, nullptr, &exe_ctx,
3772b9c1b51eSKate Stone                                            nullptr, nullptr, false, false)) {
377344d93782SGreg Clayton           window.MoveCursor(40, 0);
3774c156427dSZachary Turner           window.PutCStringTruncated(strm.GetString().str().c_str(), 1);
37755b031ebcSEd Maste         }
377644d93782SGreg Clayton 
377744d93782SGreg Clayton         window.MoveCursor(60, 0);
377844d93782SGreg Clayton         if (frame)
3779b9c1b51eSKate Stone           window.Printf("Frame: %3u  PC = 0x%16.16" PRIx64,
3780b9c1b51eSKate Stone                         frame->GetFrameIndex(),
3781b9c1b51eSKate Stone                         frame->GetFrameCodeAddress().GetOpcodeLoadAddress(
3782b9c1b51eSKate Stone                             exe_ctx.GetTargetPtr()));
3783b9c1b51eSKate Stone       } else if (state == eStateExited) {
378444d93782SGreg Clayton         const char *exit_desc = process->GetExitDescription();
378544d93782SGreg Clayton         const int exit_status = process->GetExitStatus();
378644d93782SGreg Clayton         if (exit_desc && exit_desc[0])
378744d93782SGreg Clayton           window.Printf(" with status = %i (%s)", exit_status, exit_desc);
378844d93782SGreg Clayton         else
378944d93782SGreg Clayton           window.Printf(" with status = %i", exit_status);
379044d93782SGreg Clayton       }
379144d93782SGreg Clayton     }
379244d93782SGreg Clayton     window.DeferredRefresh();
379344d93782SGreg Clayton     return true;
379444d93782SGreg Clayton   }
379544d93782SGreg Clayton 
379644d93782SGreg Clayton protected:
379744d93782SGreg Clayton   Debugger &m_debugger;
3798554f68d3SGreg Clayton   FormatEntity::Entry m_format;
379944d93782SGreg Clayton };
380044d93782SGreg Clayton 
3801b9c1b51eSKate Stone class SourceFileWindowDelegate : public WindowDelegate {
380244d93782SGreg Clayton public:
3803b9c1b51eSKate Stone   SourceFileWindowDelegate(Debugger &debugger)
3804b9c1b51eSKate Stone       : WindowDelegate(), m_debugger(debugger), m_sc(), m_file_sp(),
3805b9c1b51eSKate Stone         m_disassembly_scope(nullptr), m_disassembly_sp(), m_disassembly_range(),
3806b9c1b51eSKate Stone         m_title(), m_line_width(4), m_selected_line(0), m_pc_line(0),
3807b9c1b51eSKate Stone         m_stop_id(0), m_frame_idx(UINT32_MAX), m_first_visible_line(0),
3808b9c1b51eSKate Stone         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
380944d93782SGreg Clayton 
3810315b6884SEugene Zelenko   ~SourceFileWindowDelegate() override = default;
381144d93782SGreg Clayton 
3812b9c1b51eSKate Stone   void Update(const SymbolContext &sc) { m_sc = sc; }
381344d93782SGreg Clayton 
3814b9c1b51eSKate Stone   uint32_t NumVisibleLines() const { return m_max_y - m_min_y; }
381544d93782SGreg Clayton 
3816b9c1b51eSKate Stone   const char *WindowDelegateGetHelpText() override {
381744d93782SGreg Clayton     return "Source/Disassembly window keyboard shortcuts:";
381844d93782SGreg Clayton   }
381944d93782SGreg Clayton 
3820b9c1b51eSKate Stone   KeyHelp *WindowDelegateGetKeyHelp() override {
382144d93782SGreg Clayton     static curses::KeyHelp g_source_view_key_help[] = {
382244d93782SGreg Clayton         {KEY_RETURN, "Run to selected line with one shot breakpoint"},
382344d93782SGreg Clayton         {KEY_UP, "Select previous source line"},
382444d93782SGreg Clayton         {KEY_DOWN, "Select next source line"},
382544d93782SGreg Clayton         {KEY_PPAGE, "Page up"},
382644d93782SGreg Clayton         {KEY_NPAGE, "Page down"},
382744d93782SGreg Clayton         {'b', "Set breakpoint on selected source/disassembly line"},
382844d93782SGreg Clayton         {'c', "Continue process"},
382944d93782SGreg Clayton         {'d', "Detach and resume process"},
383044d93782SGreg Clayton         {'D', "Detach with process suspended"},
383144d93782SGreg Clayton         {'h', "Show help dialog"},
383244d93782SGreg Clayton         {'k', "Kill process"},
383344d93782SGreg Clayton         {'n', "Step over (source line)"},
383444d93782SGreg Clayton         {'N', "Step over (single instruction)"},
383544d93782SGreg Clayton         {'o', "Step out"},
383644d93782SGreg Clayton         {'s', "Step in (source line)"},
383744d93782SGreg Clayton         {'S', "Step in (single instruction)"},
383844d93782SGreg Clayton         {',', "Page up"},
383944d93782SGreg Clayton         {'.', "Page down"},
3840b9c1b51eSKate Stone         {'\0', nullptr}};
384144d93782SGreg Clayton     return g_source_view_key_help;
384244d93782SGreg Clayton   }
384344d93782SGreg Clayton 
3844b9c1b51eSKate Stone   bool WindowDelegateDraw(Window &window, bool force) override {
3845b9c1b51eSKate Stone     ExecutionContext exe_ctx =
3846b9c1b51eSKate Stone         m_debugger.GetCommandInterpreter().GetExecutionContext();
384744d93782SGreg Clayton     Process *process = exe_ctx.GetProcessPtr();
3848c5dac77aSEugene Zelenko     Thread *thread = nullptr;
384944d93782SGreg Clayton 
385044d93782SGreg Clayton     bool update_location = false;
3851b9c1b51eSKate Stone     if (process) {
385244d93782SGreg Clayton       StateType state = process->GetState();
3853b9c1b51eSKate Stone       if (StateIsStoppedState(state, true)) {
385444d93782SGreg Clayton         // We are stopped, so it is ok to
385544d93782SGreg Clayton         update_location = true;
385644d93782SGreg Clayton       }
385744d93782SGreg Clayton     }
385844d93782SGreg Clayton 
385944d93782SGreg Clayton     m_min_x = 1;
3860ec990867SGreg Clayton     m_min_y = 2;
386144d93782SGreg Clayton     m_max_x = window.GetMaxX() - 1;
386244d93782SGreg Clayton     m_max_y = window.GetMaxY() - 1;
386344d93782SGreg Clayton 
386444d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
386544d93782SGreg Clayton     StackFrameSP frame_sp;
386644d93782SGreg Clayton     bool set_selected_line_to_pc = false;
386744d93782SGreg Clayton 
3868b9c1b51eSKate Stone     if (update_location) {
386944d93782SGreg Clayton       const bool process_alive = process ? process->IsAlive() : false;
387044d93782SGreg Clayton       bool thread_changed = false;
3871b9c1b51eSKate Stone       if (process_alive) {
387244d93782SGreg Clayton         thread = exe_ctx.GetThreadPtr();
3873b9c1b51eSKate Stone         if (thread) {
387444d93782SGreg Clayton           frame_sp = thread->GetSelectedFrame();
387544d93782SGreg Clayton           auto tid = thread->GetID();
387644d93782SGreg Clayton           thread_changed = tid != m_tid;
387744d93782SGreg Clayton           m_tid = tid;
3878b9c1b51eSKate Stone         } else {
3879b9c1b51eSKate Stone           if (m_tid != LLDB_INVALID_THREAD_ID) {
388044d93782SGreg Clayton             thread_changed = true;
388144d93782SGreg Clayton             m_tid = LLDB_INVALID_THREAD_ID;
388244d93782SGreg Clayton           }
388344d93782SGreg Clayton         }
388444d93782SGreg Clayton       }
388544d93782SGreg Clayton       const uint32_t stop_id = process ? process->GetStopID() : 0;
388644d93782SGreg Clayton       const bool stop_id_changed = stop_id != m_stop_id;
388744d93782SGreg Clayton       bool frame_changed = false;
388844d93782SGreg Clayton       m_stop_id = stop_id;
3889ec990867SGreg Clayton       m_title.Clear();
3890b9c1b51eSKate Stone       if (frame_sp) {
389144d93782SGreg Clayton         m_sc = frame_sp->GetSymbolContext(eSymbolContextEverything);
3892b9c1b51eSKate Stone         if (m_sc.module_sp) {
3893b9c1b51eSKate Stone           m_title.Printf(
3894b9c1b51eSKate Stone               "%s", m_sc.module_sp->GetFileSpec().GetFilename().GetCString());
3895ec990867SGreg Clayton           ConstString func_name = m_sc.GetFunctionName();
3896ec990867SGreg Clayton           if (func_name)
3897ec990867SGreg Clayton             m_title.Printf("`%s", func_name.GetCString());
3898ec990867SGreg Clayton         }
389944d93782SGreg Clayton         const uint32_t frame_idx = frame_sp->GetFrameIndex();
390044d93782SGreg Clayton         frame_changed = frame_idx != m_frame_idx;
390144d93782SGreg Clayton         m_frame_idx = frame_idx;
3902b9c1b51eSKate Stone       } else {
390344d93782SGreg Clayton         m_sc.Clear(true);
390444d93782SGreg Clayton         frame_changed = m_frame_idx != UINT32_MAX;
390544d93782SGreg Clayton         m_frame_idx = UINT32_MAX;
390644d93782SGreg Clayton       }
390744d93782SGreg Clayton 
3908b9c1b51eSKate Stone       const bool context_changed =
3909b9c1b51eSKate Stone           thread_changed || frame_changed || stop_id_changed;
391044d93782SGreg Clayton 
3911b9c1b51eSKate Stone       if (process_alive) {
3912b9c1b51eSKate Stone         if (m_sc.line_entry.IsValid()) {
391344d93782SGreg Clayton           m_pc_line = m_sc.line_entry.line;
391444d93782SGreg Clayton           if (m_pc_line != UINT32_MAX)
391544d93782SGreg Clayton             --m_pc_line; // Convert to zero based line number...
391644d93782SGreg Clayton           // Update the selected line if the stop ID changed...
391744d93782SGreg Clayton           if (context_changed)
391844d93782SGreg Clayton             m_selected_line = m_pc_line;
391944d93782SGreg Clayton 
3920b9c1b51eSKate Stone           if (m_file_sp && m_file_sp->FileSpecMatches(m_sc.line_entry.file)) {
392105097246SAdrian Prantl             // Same file, nothing to do, we should either have the lines or not
392205097246SAdrian Prantl             // (source file missing)
3923b9c1b51eSKate Stone             if (m_selected_line >= static_cast<size_t>(m_first_visible_line)) {
392444d93782SGreg Clayton               if (m_selected_line >= m_first_visible_line + num_visible_lines)
392544d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
3926b9c1b51eSKate Stone             } else {
392744d93782SGreg Clayton               if (m_selected_line > 10)
392844d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
392944d93782SGreg Clayton               else
393044d93782SGreg Clayton                 m_first_visible_line = 0;
393144d93782SGreg Clayton             }
3932b9c1b51eSKate Stone           } else {
393344d93782SGreg Clayton             // File changed, set selected line to the line with the PC
393444d93782SGreg Clayton             m_selected_line = m_pc_line;
3935b9c1b51eSKate Stone             m_file_sp =
3936b9c1b51eSKate Stone                 m_debugger.GetSourceManager().GetFile(m_sc.line_entry.file);
3937b9c1b51eSKate Stone             if (m_file_sp) {
393844d93782SGreg Clayton               const size_t num_lines = m_file_sp->GetNumLines();
393944d93782SGreg Clayton               int m_line_width = 1;
394044d93782SGreg Clayton               for (size_t n = num_lines; n >= 10; n = n / 10)
394144d93782SGreg Clayton                 ++m_line_width;
394244d93782SGreg Clayton 
3943b9c1b51eSKate Stone               snprintf(m_line_format, sizeof(m_line_format), " %%%iu ",
3944b9c1b51eSKate Stone                        m_line_width);
3945b9c1b51eSKate Stone               if (num_lines < num_visible_lines ||
3946b9c1b51eSKate Stone                   m_selected_line < num_visible_lines)
394744d93782SGreg Clayton                 m_first_visible_line = 0;
394844d93782SGreg Clayton               else
394944d93782SGreg Clayton                 m_first_visible_line = m_selected_line - 10;
395044d93782SGreg Clayton             }
395144d93782SGreg Clayton           }
3952b9c1b51eSKate Stone         } else {
395344d93782SGreg Clayton           m_file_sp.reset();
395444d93782SGreg Clayton         }
395544d93782SGreg Clayton 
3956b9c1b51eSKate Stone         if (!m_file_sp || m_file_sp->GetNumLines() == 0) {
395744d93782SGreg Clayton           // Show disassembly
395844d93782SGreg Clayton           bool prefer_file_cache = false;
3959b9c1b51eSKate Stone           if (m_sc.function) {
3960b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.function) {
396144d93782SGreg Clayton               m_disassembly_scope = m_sc.function;
3962b9c1b51eSKate Stone               m_disassembly_sp = m_sc.function->GetInstructions(
3963b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3964b9c1b51eSKate Stone               if (m_disassembly_sp) {
396544d93782SGreg Clayton                 set_selected_line_to_pc = true;
396644d93782SGreg Clayton                 m_disassembly_range = m_sc.function->GetAddressRange();
3967b9c1b51eSKate Stone               } else {
396844d93782SGreg Clayton                 m_disassembly_range.Clear();
396944d93782SGreg Clayton               }
3970b9c1b51eSKate Stone             } else {
397144d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
397244d93782SGreg Clayton             }
3973b9c1b51eSKate Stone           } else if (m_sc.symbol) {
3974b9c1b51eSKate Stone             if (m_disassembly_scope != m_sc.symbol) {
397544d93782SGreg Clayton               m_disassembly_scope = m_sc.symbol;
3976b9c1b51eSKate Stone               m_disassembly_sp = m_sc.symbol->GetInstructions(
3977b9c1b51eSKate Stone                   exe_ctx, nullptr, prefer_file_cache);
3978b9c1b51eSKate Stone               if (m_disassembly_sp) {
397944d93782SGreg Clayton                 set_selected_line_to_pc = true;
3980b9c1b51eSKate Stone                 m_disassembly_range.GetBaseAddress() =
3981b9c1b51eSKate Stone                     m_sc.symbol->GetAddress();
398244d93782SGreg Clayton                 m_disassembly_range.SetByteSize(m_sc.symbol->GetByteSize());
3983b9c1b51eSKate Stone               } else {
398444d93782SGreg Clayton                 m_disassembly_range.Clear();
398544d93782SGreg Clayton               }
3986b9c1b51eSKate Stone             } else {
398744d93782SGreg Clayton               set_selected_line_to_pc = context_changed;
398844d93782SGreg Clayton             }
398944d93782SGreg Clayton           }
399044d93782SGreg Clayton         }
3991b9c1b51eSKate Stone       } else {
399244d93782SGreg Clayton         m_pc_line = UINT32_MAX;
399344d93782SGreg Clayton       }
399444d93782SGreg Clayton     }
399544d93782SGreg Clayton 
3996ec990867SGreg Clayton     const int window_width = window.GetWidth();
399744d93782SGreg Clayton     window.Erase();
399844d93782SGreg Clayton     window.DrawTitleBox("Sources");
3999b9c1b51eSKate Stone     if (!m_title.GetString().empty()) {
4000ec990867SGreg Clayton       window.AttributeOn(A_REVERSE);
4001ec990867SGreg Clayton       window.MoveCursor(1, 1);
4002ec990867SGreg Clayton       window.PutChar(' ');
4003c156427dSZachary Turner       window.PutCStringTruncated(m_title.GetString().str().c_str(), 1);
4004ec990867SGreg Clayton       int x = window.GetCursorX();
4005b9c1b51eSKate Stone       if (x < window_width - 1) {
4006ec990867SGreg Clayton         window.Printf("%*s", window_width - x - 1, "");
4007ec990867SGreg Clayton       }
4008ec990867SGreg Clayton       window.AttributeOff(A_REVERSE);
4009ec990867SGreg Clayton     }
401044d93782SGreg Clayton 
401144d93782SGreg Clayton     Target *target = exe_ctx.GetTargetPtr();
401244d93782SGreg Clayton     const size_t num_source_lines = GetNumSourceLines();
4013b9c1b51eSKate Stone     if (num_source_lines > 0) {
401444d93782SGreg Clayton       // Display source
401544d93782SGreg Clayton       BreakpointLines bp_lines;
4016b9c1b51eSKate Stone       if (target) {
401744d93782SGreg Clayton         BreakpointList &bp_list = target->GetBreakpointList();
401844d93782SGreg Clayton         const size_t num_bps = bp_list.GetSize();
4019b9c1b51eSKate Stone         for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
402044d93782SGreg Clayton           BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
402144d93782SGreg Clayton           const size_t num_bps_locs = bp_sp->GetNumLocations();
4022b9c1b51eSKate Stone           for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
4023b9c1b51eSKate Stone             BreakpointLocationSP bp_loc_sp =
4024b9c1b51eSKate Stone                 bp_sp->GetLocationAtIndex(bp_loc_idx);
402544d93782SGreg Clayton             LineEntry bp_loc_line_entry;
4026b9c1b51eSKate Stone             if (bp_loc_sp->GetAddress().CalculateSymbolContextLineEntry(
4027b9c1b51eSKate Stone                     bp_loc_line_entry)) {
4028b9c1b51eSKate Stone               if (m_file_sp->GetFileSpec() == bp_loc_line_entry.file) {
402944d93782SGreg Clayton                 bp_lines.insert(bp_loc_line_entry.line);
403044d93782SGreg Clayton               }
403144d93782SGreg Clayton             }
403244d93782SGreg Clayton           }
403344d93782SGreg Clayton         }
403444d93782SGreg Clayton       }
403544d93782SGreg Clayton 
403644d93782SGreg Clayton       const attr_t selected_highlight_attr = A_REVERSE;
403744d93782SGreg Clayton       const attr_t pc_highlight_attr = COLOR_PAIR(1);
403844d93782SGreg Clayton 
4039b9c1b51eSKate Stone       for (size_t i = 0; i < num_visible_lines; ++i) {
404044d93782SGreg Clayton         const uint32_t curr_line = m_first_visible_line + i;
4041b9c1b51eSKate Stone         if (curr_line < num_source_lines) {
4042ec990867SGreg Clayton           const int line_y = m_min_y + i;
404344d93782SGreg Clayton           window.MoveCursor(1, line_y);
404444d93782SGreg Clayton           const bool is_pc_line = curr_line == m_pc_line;
404544d93782SGreg Clayton           const bool line_is_selected = m_selected_line == curr_line;
404644d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
404744d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
404844d93782SGreg Clayton           attr_t highlight_attr = 0;
404944d93782SGreg Clayton           attr_t bp_attr = 0;
405044d93782SGreg Clayton           if (is_pc_line)
405144d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
405244d93782SGreg Clayton           else if (line_is_selected)
405344d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
405444d93782SGreg Clayton 
405544d93782SGreg Clayton           if (bp_lines.find(curr_line + 1) != bp_lines.end())
405644d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
405744d93782SGreg Clayton 
405844d93782SGreg Clayton           if (bp_attr)
405944d93782SGreg Clayton             window.AttributeOn(bp_attr);
406044d93782SGreg Clayton 
406144d93782SGreg Clayton           window.Printf(m_line_format, curr_line + 1);
406244d93782SGreg Clayton 
406344d93782SGreg Clayton           if (bp_attr)
406444d93782SGreg Clayton             window.AttributeOff(bp_attr);
406544d93782SGreg Clayton 
406644d93782SGreg Clayton           window.PutChar(ACS_VLINE);
406744d93782SGreg Clayton           // Mark the line with the PC with a diamond
406844d93782SGreg Clayton           if (is_pc_line)
406944d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
407044d93782SGreg Clayton           else
407144d93782SGreg Clayton             window.PutChar(' ');
407244d93782SGreg Clayton 
407344d93782SGreg Clayton           if (highlight_attr)
407444d93782SGreg Clayton             window.AttributeOn(highlight_attr);
4075b9c1b51eSKate Stone           const uint32_t line_len =
4076b9c1b51eSKate Stone               m_file_sp->GetLineLength(curr_line + 1, false);
407744d93782SGreg Clayton           if (line_len > 0)
407844d93782SGreg Clayton             window.PutCString(m_file_sp->PeekLineData(curr_line + 1), line_len);
407944d93782SGreg Clayton 
4080b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4081b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
408244d93782SGreg Clayton             StopInfoSP stop_info_sp;
408344d93782SGreg Clayton             if (thread)
408444d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4085b9c1b51eSKate Stone             if (stop_info_sp) {
408644d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4087b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
408844d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4089ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
409044d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4091b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4092b9c1b51eSKate Stone                 // line_y);
4093b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4094b9c1b51eSKate Stone                               stop_description);
409544d93782SGreg Clayton               }
4096b9c1b51eSKate Stone             } else {
4097ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
409844d93782SGreg Clayton             }
409944d93782SGreg Clayton           }
410044d93782SGreg Clayton           if (highlight_attr)
410144d93782SGreg Clayton             window.AttributeOff(highlight_attr);
4102b9c1b51eSKate Stone         } else {
410344d93782SGreg Clayton           break;
410444d93782SGreg Clayton         }
410544d93782SGreg Clayton       }
4106b9c1b51eSKate Stone     } else {
410744d93782SGreg Clayton       size_t num_disassembly_lines = GetNumDisassemblyLines();
4108b9c1b51eSKate Stone       if (num_disassembly_lines > 0) {
410944d93782SGreg Clayton         // Display disassembly
411044d93782SGreg Clayton         BreakpointAddrs bp_file_addrs;
411144d93782SGreg Clayton         Target *target = exe_ctx.GetTargetPtr();
4112b9c1b51eSKate Stone         if (target) {
411344d93782SGreg Clayton           BreakpointList &bp_list = target->GetBreakpointList();
411444d93782SGreg Clayton           const size_t num_bps = bp_list.GetSize();
4115b9c1b51eSKate Stone           for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
411644d93782SGreg Clayton             BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
411744d93782SGreg Clayton             const size_t num_bps_locs = bp_sp->GetNumLocations();
4118b9c1b51eSKate Stone             for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs;
4119b9c1b51eSKate Stone                  ++bp_loc_idx) {
4120b9c1b51eSKate Stone               BreakpointLocationSP bp_loc_sp =
4121b9c1b51eSKate Stone                   bp_sp->GetLocationAtIndex(bp_loc_idx);
412244d93782SGreg Clayton               LineEntry bp_loc_line_entry;
4123b9c1b51eSKate Stone               const lldb::addr_t file_addr =
4124b9c1b51eSKate Stone                   bp_loc_sp->GetAddress().GetFileAddress();
4125b9c1b51eSKate Stone               if (file_addr != LLDB_INVALID_ADDRESS) {
412644d93782SGreg Clayton                 if (m_disassembly_range.ContainsFileAddress(file_addr))
412744d93782SGreg Clayton                   bp_file_addrs.insert(file_addr);
412844d93782SGreg Clayton               }
412944d93782SGreg Clayton             }
413044d93782SGreg Clayton           }
413144d93782SGreg Clayton         }
413244d93782SGreg Clayton 
413344d93782SGreg Clayton         const attr_t selected_highlight_attr = A_REVERSE;
413444d93782SGreg Clayton         const attr_t pc_highlight_attr = COLOR_PAIR(1);
413544d93782SGreg Clayton 
413644d93782SGreg Clayton         StreamString strm;
413744d93782SGreg Clayton 
413844d93782SGreg Clayton         InstructionList &insts = m_disassembly_sp->GetInstructionList();
413944d93782SGreg Clayton         Address pc_address;
414044d93782SGreg Clayton 
414144d93782SGreg Clayton         if (frame_sp)
414244d93782SGreg Clayton           pc_address = frame_sp->GetFrameCodeAddress();
4143b9c1b51eSKate Stone         const uint32_t pc_idx =
4144b9c1b51eSKate Stone             pc_address.IsValid()
4145b9c1b51eSKate Stone                 ? insts.GetIndexOfInstructionAtAddress(pc_address)
4146b9c1b51eSKate Stone                 : UINT32_MAX;
4147b9c1b51eSKate Stone         if (set_selected_line_to_pc) {
414844d93782SGreg Clayton           m_selected_line = pc_idx;
414944d93782SGreg Clayton         }
415044d93782SGreg Clayton 
415144d93782SGreg Clayton         const uint32_t non_visible_pc_offset = (num_visible_lines / 5);
41523985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) >= num_disassembly_lines)
415344d93782SGreg Clayton           m_first_visible_line = 0;
415444d93782SGreg Clayton 
4155b9c1b51eSKate Stone         if (pc_idx < num_disassembly_lines) {
41563985c8c6SSaleem Abdulrasool           if (pc_idx < static_cast<uint32_t>(m_first_visible_line) ||
415744d93782SGreg Clayton               pc_idx >= m_first_visible_line + num_visible_lines)
415844d93782SGreg Clayton             m_first_visible_line = pc_idx - non_visible_pc_offset;
415944d93782SGreg Clayton         }
416044d93782SGreg Clayton 
4161b9c1b51eSKate Stone         for (size_t i = 0; i < num_visible_lines; ++i) {
416244d93782SGreg Clayton           const uint32_t inst_idx = m_first_visible_line + i;
416344d93782SGreg Clayton           Instruction *inst = insts.GetInstructionAtIndex(inst_idx).get();
416444d93782SGreg Clayton           if (!inst)
416544d93782SGreg Clayton             break;
416644d93782SGreg Clayton 
4167ec990867SGreg Clayton           const int line_y = m_min_y + i;
4168ec990867SGreg Clayton           window.MoveCursor(1, line_y);
416944d93782SGreg Clayton           const bool is_pc_line = frame_sp && inst_idx == pc_idx;
417044d93782SGreg Clayton           const bool line_is_selected = m_selected_line == inst_idx;
417144d93782SGreg Clayton           // Highlight the line as the PC line first, then if the selected line
417244d93782SGreg Clayton           // isn't the same as the PC line, highlight it differently
417344d93782SGreg Clayton           attr_t highlight_attr = 0;
417444d93782SGreg Clayton           attr_t bp_attr = 0;
417544d93782SGreg Clayton           if (is_pc_line)
417644d93782SGreg Clayton             highlight_attr = pc_highlight_attr;
417744d93782SGreg Clayton           else if (line_is_selected)
417844d93782SGreg Clayton             highlight_attr = selected_highlight_attr;
417944d93782SGreg Clayton 
4180b9c1b51eSKate Stone           if (bp_file_addrs.find(inst->GetAddress().GetFileAddress()) !=
4181b9c1b51eSKate Stone               bp_file_addrs.end())
418244d93782SGreg Clayton             bp_attr = COLOR_PAIR(2);
418344d93782SGreg Clayton 
418444d93782SGreg Clayton           if (bp_attr)
418544d93782SGreg Clayton             window.AttributeOn(bp_attr);
418644d93782SGreg Clayton 
4187324a1036SSaleem Abdulrasool           window.Printf(" 0x%16.16llx ",
4188b9c1b51eSKate Stone                         static_cast<unsigned long long>(
4189b9c1b51eSKate Stone                             inst->GetAddress().GetLoadAddress(target)));
419044d93782SGreg Clayton 
419144d93782SGreg Clayton           if (bp_attr)
419244d93782SGreg Clayton             window.AttributeOff(bp_attr);
419344d93782SGreg Clayton 
419444d93782SGreg Clayton           window.PutChar(ACS_VLINE);
419544d93782SGreg Clayton           // Mark the line with the PC with a diamond
419644d93782SGreg Clayton           if (is_pc_line)
419744d93782SGreg Clayton             window.PutChar(ACS_DIAMOND);
419844d93782SGreg Clayton           else
419944d93782SGreg Clayton             window.PutChar(' ');
420044d93782SGreg Clayton 
420144d93782SGreg Clayton           if (highlight_attr)
420244d93782SGreg Clayton             window.AttributeOn(highlight_attr);
420344d93782SGreg Clayton 
420444d93782SGreg Clayton           const char *mnemonic = inst->GetMnemonic(&exe_ctx);
420544d93782SGreg Clayton           const char *operands = inst->GetOperands(&exe_ctx);
420644d93782SGreg Clayton           const char *comment = inst->GetComment(&exe_ctx);
420744d93782SGreg Clayton 
4208c5dac77aSEugene Zelenko           if (mnemonic != nullptr && mnemonic[0] == '\0')
4209c5dac77aSEugene Zelenko             mnemonic = nullptr;
4210c5dac77aSEugene Zelenko           if (operands != nullptr && operands[0] == '\0')
4211c5dac77aSEugene Zelenko             operands = nullptr;
4212c5dac77aSEugene Zelenko           if (comment != nullptr && comment[0] == '\0')
4213c5dac77aSEugene Zelenko             comment = nullptr;
421444d93782SGreg Clayton 
421544d93782SGreg Clayton           strm.Clear();
421644d93782SGreg Clayton 
4217c5dac77aSEugene Zelenko           if (mnemonic != nullptr && operands != nullptr && comment != nullptr)
421844d93782SGreg Clayton             strm.Printf("%-8s %-25s ; %s", mnemonic, operands, comment);
4219c5dac77aSEugene Zelenko           else if (mnemonic != nullptr && operands != nullptr)
422044d93782SGreg Clayton             strm.Printf("%-8s %s", mnemonic, operands);
4221c5dac77aSEugene Zelenko           else if (mnemonic != nullptr)
422244d93782SGreg Clayton             strm.Printf("%s", mnemonic);
422344d93782SGreg Clayton 
422444d93782SGreg Clayton           int right_pad = 1;
4225c156427dSZachary Turner           window.PutCStringTruncated(strm.GetData(), right_pad);
422644d93782SGreg Clayton 
4227b9c1b51eSKate Stone           if (is_pc_line && frame_sp &&
4228b9c1b51eSKate Stone               frame_sp->GetConcreteFrameIndex() == 0) {
422944d93782SGreg Clayton             StopInfoSP stop_info_sp;
423044d93782SGreg Clayton             if (thread)
423144d93782SGreg Clayton               stop_info_sp = thread->GetStopInfo();
4232b9c1b51eSKate Stone             if (stop_info_sp) {
423344d93782SGreg Clayton               const char *stop_description = stop_info_sp->GetDescription();
4234b9c1b51eSKate Stone               if (stop_description && stop_description[0]) {
423544d93782SGreg Clayton                 size_t stop_description_len = strlen(stop_description);
4236ec990867SGreg Clayton                 int desc_x = window_width - stop_description_len - 16;
423744d93782SGreg Clayton                 window.Printf("%*s", desc_x - window.GetCursorX(), "");
4238b9c1b51eSKate Stone                 // window.MoveCursor(window_width - stop_description_len - 15,
4239b9c1b51eSKate Stone                 // line_y);
4240b9c1b51eSKate Stone                 window.Printf("<<< Thread %u: %s ", thread->GetIndexID(),
4241b9c1b51eSKate Stone                               stop_description);
424244d93782SGreg Clayton               }
4243b9c1b51eSKate Stone             } else {
4244ec990867SGreg Clayton               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
424544d93782SGreg Clayton             }
424644d93782SGreg Clayton           }
424744d93782SGreg Clayton           if (highlight_attr)
424844d93782SGreg Clayton             window.AttributeOff(highlight_attr);
424944d93782SGreg Clayton         }
425044d93782SGreg Clayton       }
425144d93782SGreg Clayton     }
425244d93782SGreg Clayton     window.DeferredRefresh();
425344d93782SGreg Clayton     return true; // Drawing handled
425444d93782SGreg Clayton   }
425544d93782SGreg Clayton 
4256b9c1b51eSKate Stone   size_t GetNumLines() {
425744d93782SGreg Clayton     size_t num_lines = GetNumSourceLines();
425844d93782SGreg Clayton     if (num_lines == 0)
425944d93782SGreg Clayton       num_lines = GetNumDisassemblyLines();
426044d93782SGreg Clayton     return num_lines;
426144d93782SGreg Clayton   }
426244d93782SGreg Clayton 
4263b9c1b51eSKate Stone   size_t GetNumSourceLines() const {
426444d93782SGreg Clayton     if (m_file_sp)
426544d93782SGreg Clayton       return m_file_sp->GetNumLines();
426644d93782SGreg Clayton     return 0;
426744d93782SGreg Clayton   }
4268315b6884SEugene Zelenko 
4269b9c1b51eSKate Stone   size_t GetNumDisassemblyLines() const {
427044d93782SGreg Clayton     if (m_disassembly_sp)
427144d93782SGreg Clayton       return m_disassembly_sp->GetInstructionList().GetSize();
427244d93782SGreg Clayton     return 0;
427344d93782SGreg Clayton   }
427444d93782SGreg Clayton 
4275b9c1b51eSKate Stone   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
427644d93782SGreg Clayton     const uint32_t num_visible_lines = NumVisibleLines();
427744d93782SGreg Clayton     const size_t num_lines = GetNumLines();
427844d93782SGreg Clayton 
4279b9c1b51eSKate Stone     switch (c) {
428044d93782SGreg Clayton     case ',':
428144d93782SGreg Clayton     case KEY_PPAGE:
428244d93782SGreg Clayton       // Page up key
42833985c8c6SSaleem Abdulrasool       if (static_cast<uint32_t>(m_first_visible_line) > num_visible_lines)
428444d93782SGreg Clayton         m_first_visible_line -= num_visible_lines;
428544d93782SGreg Clayton       else
428644d93782SGreg Clayton         m_first_visible_line = 0;
428744d93782SGreg Clayton       m_selected_line = m_first_visible_line;
428844d93782SGreg Clayton       return eKeyHandled;
428944d93782SGreg Clayton 
429044d93782SGreg Clayton     case '.':
429144d93782SGreg Clayton     case KEY_NPAGE:
429244d93782SGreg Clayton       // Page down key
429344d93782SGreg Clayton       {
429444d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < num_lines)
429544d93782SGreg Clayton           m_first_visible_line += num_visible_lines;
429644d93782SGreg Clayton         else if (num_lines < num_visible_lines)
429744d93782SGreg Clayton           m_first_visible_line = 0;
429844d93782SGreg Clayton         else
429944d93782SGreg Clayton           m_first_visible_line = num_lines - num_visible_lines;
430044d93782SGreg Clayton         m_selected_line = m_first_visible_line;
430144d93782SGreg Clayton       }
430244d93782SGreg Clayton       return eKeyHandled;
430344d93782SGreg Clayton 
430444d93782SGreg Clayton     case KEY_UP:
4305b9c1b51eSKate Stone       if (m_selected_line > 0) {
430644d93782SGreg Clayton         m_selected_line--;
43073985c8c6SSaleem Abdulrasool         if (static_cast<size_t>(m_first_visible_line) > m_selected_line)
430844d93782SGreg Clayton           m_first_visible_line = m_selected_line;
430944d93782SGreg Clayton       }
431044d93782SGreg Clayton       return eKeyHandled;
431144d93782SGreg Clayton 
431244d93782SGreg Clayton     case KEY_DOWN:
4313b9c1b51eSKate Stone       if (m_selected_line + 1 < num_lines) {
431444d93782SGreg Clayton         m_selected_line++;
431544d93782SGreg Clayton         if (m_first_visible_line + num_visible_lines < m_selected_line)
431644d93782SGreg Clayton           m_first_visible_line++;
431744d93782SGreg Clayton       }
431844d93782SGreg Clayton       return eKeyHandled;
431944d93782SGreg Clayton 
432044d93782SGreg Clayton     case '\r':
432144d93782SGreg Clayton     case '\n':
432244d93782SGreg Clayton     case KEY_ENTER:
432344d93782SGreg Clayton       // Set a breakpoint and run to the line using a one shot breakpoint
4324b9c1b51eSKate Stone       if (GetNumSourceLines() > 0) {
4325b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4326b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4327b9c1b51eSKate Stone         if (exe_ctx.HasProcessScope() && exe_ctx.GetProcessRef().IsAlive()) {
4328b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4329b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
433044d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4331b9c1b51eSKate Stone               m_selected_line +
4332b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4333431b1584SAdrian Prantl               0,     // Unspecified column.
43342411167fSJim Ingham               0,     // No offset
433544d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
433644d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
433744d93782SGreg Clayton               false,               // internal
4338055ad9beSIlia K               false,               // request_hardware
4339055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
434044d93782SGreg Clayton           // Make breakpoint one shot
434144d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
434244d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
434344d93782SGreg Clayton         }
4344b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4345b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4346b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4347b9c1b51eSKate Stone                                       .get();
4348b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4349b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4350b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
435144d93782SGreg Clayton           Address addr = inst->GetAddress();
4352b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4353b9c1b51eSKate Stone               addr,   // lldb_private::Address
435444d93782SGreg Clayton               false,  // internal
435544d93782SGreg Clayton               false); // request_hardware
435644d93782SGreg Clayton           // Make breakpoint one shot
435744d93782SGreg Clayton           bp_sp->GetOptions()->SetOneShot(true);
435844d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
435944d93782SGreg Clayton         }
436044d93782SGreg Clayton       }
436144d93782SGreg Clayton       return eKeyHandled;
436244d93782SGreg Clayton 
436344d93782SGreg Clayton     case 'b': // 'b' == toggle breakpoint on currently selected line
4364b9c1b51eSKate Stone       if (m_selected_line < GetNumSourceLines()) {
4365b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4366b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4367b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
4368b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4369b9c1b51eSKate Stone               nullptr, // Don't limit the breakpoint to certain modules
437044d93782SGreg Clayton               m_file_sp->GetFileSpec(), // Source file
4371b9c1b51eSKate Stone               m_selected_line +
4372b9c1b51eSKate Stone                   1, // Source line number (m_selected_line is zero based)
4373431b1584SAdrian Prantl               0,     // No column specified.
43742411167fSJim Ingham               0,     // No offset
437544d93782SGreg Clayton               eLazyBoolCalculate,  // Check inlines using global setting
437644d93782SGreg Clayton               eLazyBoolCalculate,  // Skip prologue using global setting,
437744d93782SGreg Clayton               false,               // internal
4378055ad9beSIlia K               false,               // request_hardware
4379055ad9beSIlia K               eLazyBoolCalculate); // move_to_nearest_code
438044d93782SGreg Clayton         }
4381b9c1b51eSKate Stone       } else if (m_selected_line < GetNumDisassemblyLines()) {
4382b9c1b51eSKate Stone         const Instruction *inst = m_disassembly_sp->GetInstructionList()
4383b9c1b51eSKate Stone                                       .GetInstructionAtIndex(m_selected_line)
4384b9c1b51eSKate Stone                                       .get();
4385b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4386b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4387b9c1b51eSKate Stone         if (exe_ctx.HasTargetScope()) {
438844d93782SGreg Clayton           Address addr = inst->GetAddress();
4389b9c1b51eSKate Stone           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
4390b9c1b51eSKate Stone               addr,   // lldb_private::Address
439144d93782SGreg Clayton               false,  // internal
439244d93782SGreg Clayton               false); // request_hardware
439344d93782SGreg Clayton         }
439444d93782SGreg Clayton       }
439544d93782SGreg Clayton       return eKeyHandled;
439644d93782SGreg Clayton 
439744d93782SGreg Clayton     case 'd': // 'd' == detach and let run
439844d93782SGreg Clayton     case 'D': // 'D' == detach and keep stopped
439944d93782SGreg Clayton     {
4400b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4401b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
440244d93782SGreg Clayton       if (exe_ctx.HasProcessScope())
440344d93782SGreg Clayton         exe_ctx.GetProcessRef().Detach(c == 'D');
440444d93782SGreg Clayton     }
440544d93782SGreg Clayton       return eKeyHandled;
440644d93782SGreg Clayton 
440744d93782SGreg Clayton     case 'k':
440844d93782SGreg Clayton       // 'k' == kill
440944d93782SGreg Clayton       {
4410b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4411b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
441244d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
4413ede3193bSJason Molenda           exe_ctx.GetProcessRef().Destroy(false);
441444d93782SGreg Clayton       }
441544d93782SGreg Clayton       return eKeyHandled;
441644d93782SGreg Clayton 
441744d93782SGreg Clayton     case 'c':
441844d93782SGreg Clayton       // 'c' == continue
441944d93782SGreg Clayton       {
4420b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4421b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
442244d93782SGreg Clayton         if (exe_ctx.HasProcessScope())
442344d93782SGreg Clayton           exe_ctx.GetProcessRef().Resume();
442444d93782SGreg Clayton       }
442544d93782SGreg Clayton       return eKeyHandled;
442644d93782SGreg Clayton 
442744d93782SGreg Clayton     case 'o':
442844d93782SGreg Clayton       // 'o' == step out
442944d93782SGreg Clayton       {
4430b9c1b51eSKate Stone         ExecutionContext exe_ctx =
4431b9c1b51eSKate Stone             m_debugger.GetCommandInterpreter().GetExecutionContext();
4432b9c1b51eSKate Stone         if (exe_ctx.HasThreadScope() &&
4433b9c1b51eSKate Stone             StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
443444d93782SGreg Clayton           exe_ctx.GetThreadRef().StepOut();
443544d93782SGreg Clayton         }
443644d93782SGreg Clayton       }
443744d93782SGreg Clayton       return eKeyHandled;
4438315b6884SEugene Zelenko 
443944d93782SGreg Clayton     case 'n': // 'n' == step over
444044d93782SGreg Clayton     case 'N': // 'N' == step over instruction
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         bool source_step = (c == 'n');
444744d93782SGreg Clayton         exe_ctx.GetThreadRef().StepOver(source_step);
444844d93782SGreg Clayton       }
444944d93782SGreg Clayton     }
445044d93782SGreg Clayton       return eKeyHandled;
4451315b6884SEugene Zelenko 
445244d93782SGreg Clayton     case 's': // 's' == step into
445344d93782SGreg Clayton     case 'S': // 'S' == step into instruction
445444d93782SGreg Clayton     {
4455b9c1b51eSKate Stone       ExecutionContext exe_ctx =
4456b9c1b51eSKate Stone           m_debugger.GetCommandInterpreter().GetExecutionContext();
4457b9c1b51eSKate Stone       if (exe_ctx.HasThreadScope() &&
4458b9c1b51eSKate Stone           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
445944d93782SGreg Clayton         bool source_step = (c == 's');
44604b4b2478SJim Ingham         exe_ctx.GetThreadRef().StepIn(source_step);
446144d93782SGreg Clayton       }
446244d93782SGreg Clayton     }
446344d93782SGreg Clayton       return eKeyHandled;
446444d93782SGreg Clayton 
446544d93782SGreg Clayton     case 'h':
446644d93782SGreg Clayton       window.CreateHelpSubwindow();
446744d93782SGreg Clayton       return eKeyHandled;
446844d93782SGreg Clayton 
446944d93782SGreg Clayton     default:
447044d93782SGreg Clayton       break;
447144d93782SGreg Clayton     }
447244d93782SGreg Clayton     return eKeyNotHandled;
447344d93782SGreg Clayton   }
447444d93782SGreg Clayton 
447544d93782SGreg Clayton protected:
447644d93782SGreg Clayton   typedef std::set<uint32_t> BreakpointLines;
447744d93782SGreg Clayton   typedef std::set<lldb::addr_t> BreakpointAddrs;
447844d93782SGreg Clayton 
447944d93782SGreg Clayton   Debugger &m_debugger;
448044d93782SGreg Clayton   SymbolContext m_sc;
448144d93782SGreg Clayton   SourceManager::FileSP m_file_sp;
448244d93782SGreg Clayton   SymbolContextScope *m_disassembly_scope;
448344d93782SGreg Clayton   lldb::DisassemblerSP m_disassembly_sp;
448444d93782SGreg Clayton   AddressRange m_disassembly_range;
4485ec990867SGreg Clayton   StreamString m_title;
448644d93782SGreg Clayton   lldb::user_id_t m_tid;
448744d93782SGreg Clayton   char m_line_format[8];
448844d93782SGreg Clayton   int m_line_width;
448944d93782SGreg Clayton   uint32_t m_selected_line; // The selected line
449044d93782SGreg Clayton   uint32_t m_pc_line;       // The line with the PC
449144d93782SGreg Clayton   uint32_t m_stop_id;
449244d93782SGreg Clayton   uint32_t m_frame_idx;
449344d93782SGreg Clayton   int m_first_visible_line;
449444d93782SGreg Clayton   int m_min_x;
449544d93782SGreg Clayton   int m_min_y;
449644d93782SGreg Clayton   int m_max_x;
449744d93782SGreg Clayton   int m_max_y;
449844d93782SGreg Clayton };
449944d93782SGreg Clayton 
450044d93782SGreg Clayton DisplayOptions ValueObjectListDelegate::g_options = {true};
450144d93782SGreg Clayton 
4502b9c1b51eSKate Stone IOHandlerCursesGUI::IOHandlerCursesGUI(Debugger &debugger)
4503b9c1b51eSKate Stone     : IOHandler(debugger, IOHandler::Type::Curses) {}
450444d93782SGreg Clayton 
4505b9c1b51eSKate Stone void IOHandlerCursesGUI::Activate() {
450644d93782SGreg Clayton   IOHandler::Activate();
4507b9c1b51eSKate Stone   if (!m_app_ap) {
450844d93782SGreg Clayton     m_app_ap.reset(new Application(GetInputFILE(), GetOutputFILE()));
450944d93782SGreg Clayton 
451044d93782SGreg Clayton     // This is both a window and a menu delegate
4511b9c1b51eSKate Stone     std::shared_ptr<ApplicationDelegate> app_delegate_sp(
4512b9c1b51eSKate Stone         new ApplicationDelegate(*m_app_ap, m_debugger));
451344d93782SGreg Clayton 
4514b9c1b51eSKate Stone     MenuDelegateSP app_menu_delegate_sp =
4515b9c1b51eSKate Stone         std::static_pointer_cast<MenuDelegate>(app_delegate_sp);
4516b9c1b51eSKate Stone     MenuSP lldb_menu_sp(
4517b9c1b51eSKate Stone         new Menu("LLDB", "F1", KEY_F(1), ApplicationDelegate::eMenuID_LLDB));
4518b9c1b51eSKate Stone     MenuSP exit_menuitem_sp(
4519b9c1b51eSKate Stone         new Menu("Exit", nullptr, 'x', ApplicationDelegate::eMenuID_LLDBExit));
452044d93782SGreg Clayton     exit_menuitem_sp->SetCannedResult(MenuActionResult::Quit);
4521b9c1b51eSKate Stone     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(
4522b9c1b51eSKate Stone         "About LLDB", nullptr, 'a', ApplicationDelegate::eMenuID_LLDBAbout)));
452344d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
452444d93782SGreg Clayton     lldb_menu_sp->AddSubmenu(exit_menuitem_sp);
452544d93782SGreg Clayton 
4526b9c1b51eSKate Stone     MenuSP target_menu_sp(new Menu("Target", "F2", KEY_F(2),
4527b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Target));
4528b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4529b9c1b51eSKate Stone         "Create", nullptr, 'c', ApplicationDelegate::eMenuID_TargetCreate)));
4530b9c1b51eSKate Stone     target_menu_sp->AddSubmenu(MenuSP(new Menu(
4531b9c1b51eSKate Stone         "Delete", nullptr, 'd', ApplicationDelegate::eMenuID_TargetDelete)));
453244d93782SGreg Clayton 
4533b9c1b51eSKate Stone     MenuSP process_menu_sp(new Menu("Process", "F3", KEY_F(3),
4534b9c1b51eSKate Stone                                     ApplicationDelegate::eMenuID_Process));
4535b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4536b9c1b51eSKate Stone         "Attach", nullptr, 'a', ApplicationDelegate::eMenuID_ProcessAttach)));
4537b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4538b9c1b51eSKate Stone         "Detach", nullptr, 'd', ApplicationDelegate::eMenuID_ProcessDetach)));
4539b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4540b9c1b51eSKate Stone         "Launch", nullptr, 'l', ApplicationDelegate::eMenuID_ProcessLaunch)));
454144d93782SGreg Clayton     process_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
4542b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(
4543b9c1b51eSKate Stone         MenuSP(new Menu("Continue", nullptr, 'c',
4544b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ProcessContinue)));
4545b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4546b9c1b51eSKate Stone         "Halt", nullptr, 'h', ApplicationDelegate::eMenuID_ProcessHalt)));
4547b9c1b51eSKate Stone     process_menu_sp->AddSubmenu(MenuSP(new Menu(
4548b9c1b51eSKate Stone         "Kill", nullptr, 'k', ApplicationDelegate::eMenuID_ProcessKill)));
454944d93782SGreg Clayton 
4550b9c1b51eSKate Stone     MenuSP thread_menu_sp(new Menu("Thread", "F4", KEY_F(4),
4551b9c1b51eSKate Stone                                    ApplicationDelegate::eMenuID_Thread));
4552b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4553b9c1b51eSKate Stone         "Step In", nullptr, 'i', ApplicationDelegate::eMenuID_ThreadStepIn)));
4554b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(
4555b9c1b51eSKate Stone         MenuSP(new Menu("Step Over", nullptr, 'v',
4556b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ThreadStepOver)));
4557b9c1b51eSKate Stone     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
4558b9c1b51eSKate Stone         "Step Out", nullptr, 'o', ApplicationDelegate::eMenuID_ThreadStepOut)));
455944d93782SGreg Clayton 
4560b9c1b51eSKate Stone     MenuSP view_menu_sp(
4561b9c1b51eSKate Stone         new Menu("View", "F5", KEY_F(5), ApplicationDelegate::eMenuID_View));
4562b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4563b9c1b51eSKate Stone         MenuSP(new Menu("Backtrace", nullptr, 'b',
4564b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewBacktrace)));
4565b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4566b9c1b51eSKate Stone         MenuSP(new Menu("Registers", nullptr, 'r',
4567b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewRegisters)));
4568b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(MenuSP(new Menu(
4569b9c1b51eSKate Stone         "Source", nullptr, 's', ApplicationDelegate::eMenuID_ViewSource)));
4570b9c1b51eSKate Stone     view_menu_sp->AddSubmenu(
4571b9c1b51eSKate Stone         MenuSP(new Menu("Variables", nullptr, 'v',
4572b9c1b51eSKate Stone                         ApplicationDelegate::eMenuID_ViewVariables)));
457344d93782SGreg Clayton 
4574b9c1b51eSKate Stone     MenuSP help_menu_sp(
4575b9c1b51eSKate Stone         new Menu("Help", "F6", KEY_F(6), ApplicationDelegate::eMenuID_Help));
4576b9c1b51eSKate Stone     help_menu_sp->AddSubmenu(MenuSP(new Menu(
4577b9c1b51eSKate Stone         "GUI Help", nullptr, 'g', ApplicationDelegate::eMenuID_HelpGUIHelp)));
457844d93782SGreg Clayton 
457944d93782SGreg Clayton     m_app_ap->Initialize();
458044d93782SGreg Clayton     WindowSP &main_window_sp = m_app_ap->GetMainWindow();
458144d93782SGreg Clayton 
458244d93782SGreg Clayton     MenuSP menubar_sp(new Menu(Menu::Type::Bar));
458344d93782SGreg Clayton     menubar_sp->AddSubmenu(lldb_menu_sp);
458444d93782SGreg Clayton     menubar_sp->AddSubmenu(target_menu_sp);
458544d93782SGreg Clayton     menubar_sp->AddSubmenu(process_menu_sp);
458644d93782SGreg Clayton     menubar_sp->AddSubmenu(thread_menu_sp);
458744d93782SGreg Clayton     menubar_sp->AddSubmenu(view_menu_sp);
458844d93782SGreg Clayton     menubar_sp->AddSubmenu(help_menu_sp);
458944d93782SGreg Clayton     menubar_sp->SetDelegate(app_menu_delegate_sp);
459044d93782SGreg Clayton 
459144d93782SGreg Clayton     Rect content_bounds = main_window_sp->GetFrame();
459244d93782SGreg Clayton     Rect menubar_bounds = content_bounds.MakeMenuBar();
459344d93782SGreg Clayton     Rect status_bounds = content_bounds.MakeStatusBar();
459444d93782SGreg Clayton     Rect source_bounds;
459544d93782SGreg Clayton     Rect variables_bounds;
459644d93782SGreg Clayton     Rect threads_bounds;
459744d93782SGreg Clayton     Rect source_variables_bounds;
4598b9c1b51eSKate Stone     content_bounds.VerticalSplitPercentage(0.80, source_variables_bounds,
4599b9c1b51eSKate Stone                                            threads_bounds);
4600b9c1b51eSKate Stone     source_variables_bounds.HorizontalSplitPercentage(0.70, source_bounds,
4601b9c1b51eSKate Stone                                                       variables_bounds);
460244d93782SGreg Clayton 
4603b9c1b51eSKate Stone     WindowSP menubar_window_sp =
4604b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Menubar", menubar_bounds, false);
460505097246SAdrian Prantl     // Let the menubar get keys if the active window doesn't handle the keys
460605097246SAdrian Prantl     // that are typed so it can respond to menubar key presses.
4607b9c1b51eSKate Stone     menubar_window_sp->SetCanBeActive(
4608b9c1b51eSKate Stone         false); // Don't let the menubar become the active window
460944d93782SGreg Clayton     menubar_window_sp->SetDelegate(menubar_sp);
461044d93782SGreg Clayton 
4611b9c1b51eSKate Stone     WindowSP source_window_sp(
4612b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Source", source_bounds, true));
4613b9c1b51eSKate Stone     WindowSP variables_window_sp(
4614b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Variables", variables_bounds, false));
4615b9c1b51eSKate Stone     WindowSP threads_window_sp(
4616b9c1b51eSKate Stone         main_window_sp->CreateSubWindow("Threads", threads_bounds, false));
4617b9c1b51eSKate Stone     WindowSP status_window_sp(
46186bb7e21fSPavel Labath         main_window_sp->CreateSubWindow("Status", status_bounds, false));
4619b9c1b51eSKate Stone     status_window_sp->SetCanBeActive(
4620b9c1b51eSKate Stone         false); // Don't let the status bar become the active window
4621b9c1b51eSKate Stone     main_window_sp->SetDelegate(
4622b9c1b51eSKate Stone         std::static_pointer_cast<WindowDelegate>(app_delegate_sp));
4623b9c1b51eSKate Stone     source_window_sp->SetDelegate(
4624b9c1b51eSKate Stone         WindowDelegateSP(new SourceFileWindowDelegate(m_debugger)));
4625b9c1b51eSKate Stone     variables_window_sp->SetDelegate(
4626b9c1b51eSKate Stone         WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
4627ec990867SGreg Clayton     TreeDelegateSP thread_delegate_sp(new ThreadsTreeDelegate(m_debugger));
4628b9c1b51eSKate Stone     threads_window_sp->SetDelegate(WindowDelegateSP(
4629b9c1b51eSKate Stone         new TreeWindowDelegate(m_debugger, thread_delegate_sp)));
4630b9c1b51eSKate Stone     status_window_sp->SetDelegate(
4631b9c1b51eSKate Stone         WindowDelegateSP(new StatusBarWindowDelegate(m_debugger)));
463244d93782SGreg Clayton 
46335fdb09bbSGreg Clayton     // Show the main help window once the first time the curses GUI is launched
46345fdb09bbSGreg Clayton     static bool g_showed_help = false;
4635b9c1b51eSKate Stone     if (!g_showed_help) {
46365fdb09bbSGreg Clayton       g_showed_help = true;
46375fdb09bbSGreg Clayton       main_window_sp->CreateHelpSubwindow();
46385fdb09bbSGreg Clayton     }
46395fdb09bbSGreg Clayton 
464044d93782SGreg Clayton     init_pair(1, COLOR_WHITE, COLOR_BLUE);
464144d93782SGreg Clayton     init_pair(2, COLOR_BLACK, COLOR_WHITE);
464244d93782SGreg Clayton     init_pair(3, COLOR_MAGENTA, COLOR_WHITE);
464344d93782SGreg Clayton     init_pair(4, COLOR_MAGENTA, COLOR_BLACK);
464444d93782SGreg Clayton     init_pair(5, COLOR_RED, COLOR_BLACK);
464544d93782SGreg Clayton   }
464644d93782SGreg Clayton }
464744d93782SGreg Clayton 
4648b9c1b51eSKate Stone void IOHandlerCursesGUI::Deactivate() { m_app_ap->Terminate(); }
464944d93782SGreg Clayton 
4650b9c1b51eSKate Stone void IOHandlerCursesGUI::Run() {
465144d93782SGreg Clayton   m_app_ap->Run(m_debugger);
465244d93782SGreg Clayton   SetIsDone(true);
465344d93782SGreg Clayton }
465444d93782SGreg Clayton 
4655315b6884SEugene Zelenko IOHandlerCursesGUI::~IOHandlerCursesGUI() = default;
465644d93782SGreg Clayton 
4657b9c1b51eSKate Stone void IOHandlerCursesGUI::Cancel() {}
465844d93782SGreg Clayton 
4659b9c1b51eSKate Stone bool IOHandlerCursesGUI::Interrupt() { return false; }
466044d93782SGreg Clayton 
4661b9c1b51eSKate Stone void IOHandlerCursesGUI::GotEOF() {}
466244d93782SGreg Clayton 
4663315b6884SEugene Zelenko #endif // LLDB_DISABLE_CURSES
4664