180814287SRaphael Isemann //===-- IOHandlerCursesGUI.cpp --------------------------------------------===//
27caa17caSRaphael Isemann //
37caa17caSRaphael Isemann // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
47caa17caSRaphael Isemann // See https://llvm.org/LICENSE.txt for license information.
57caa17caSRaphael Isemann // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
67caa17caSRaphael Isemann //
77caa17caSRaphael Isemann //===----------------------------------------------------------------------===//
87caa17caSRaphael Isemann 
97caa17caSRaphael Isemann #include "lldb/Core/IOHandlerCursesGUI.h"
1059998b7bSJonas Devlieghere #include "lldb/Host/Config.h"
117caa17caSRaphael Isemann 
12a4304f96SJonas Devlieghere #if LLDB_ENABLE_CURSES
13c952ec15SLuboš Luňák #if CURSES_HAVE_NCURSES_CURSES_H
14c952ec15SLuboš Luňák #include <ncurses/curses.h>
15c952ec15SLuboš Luňák #include <ncurses/panel.h>
16c952ec15SLuboš Luňák #else
177caa17caSRaphael Isemann #include <curses.h>
187caa17caSRaphael Isemann #include <panel.h>
197caa17caSRaphael Isemann #endif
20c952ec15SLuboš Luňák #endif
217caa17caSRaphael Isemann 
227caa17caSRaphael Isemann #if defined(__APPLE__)
237caa17caSRaphael Isemann #include <deque>
247caa17caSRaphael Isemann #endif
257caa17caSRaphael Isemann #include <string>
267caa17caSRaphael Isemann 
277caa17caSRaphael Isemann #include "lldb/Core/Debugger.h"
287caa17caSRaphael Isemann #include "lldb/Core/StreamFile.h"
294631afdeSRaphael Isemann #include "lldb/Core/ValueObjectUpdater.h"
307caa17caSRaphael Isemann #include "lldb/Host/File.h"
317caa17caSRaphael Isemann #include "lldb/Utility/Predicate.h"
327caa17caSRaphael Isemann #include "lldb/Utility/Status.h"
337caa17caSRaphael Isemann #include "lldb/Utility/StreamString.h"
347caa17caSRaphael Isemann #include "lldb/Utility/StringList.h"
357caa17caSRaphael Isemann #include "lldb/lldb-forward.h"
367caa17caSRaphael Isemann 
377caa17caSRaphael Isemann #include "lldb/Interpreter/CommandCompletions.h"
387caa17caSRaphael Isemann #include "lldb/Interpreter/CommandInterpreter.h"
397caa17caSRaphael Isemann 
40a4304f96SJonas Devlieghere #if LLDB_ENABLE_CURSES
417caa17caSRaphael Isemann #include "lldb/Breakpoint/BreakpointLocation.h"
427caa17caSRaphael Isemann #include "lldb/Core/Module.h"
43*0321dbc8SOmar Emara #include "lldb/Core/PluginManager.h"
447caa17caSRaphael Isemann #include "lldb/Core/ValueObject.h"
457caa17caSRaphael Isemann #include "lldb/Core/ValueObjectRegister.h"
467caa17caSRaphael Isemann #include "lldb/Symbol/Block.h"
477caa17caSRaphael Isemann #include "lldb/Symbol/Function.h"
487caa17caSRaphael Isemann #include "lldb/Symbol/Symbol.h"
497caa17caSRaphael Isemann #include "lldb/Symbol/VariableList.h"
507caa17caSRaphael Isemann #include "lldb/Target/Process.h"
517caa17caSRaphael Isemann #include "lldb/Target/RegisterContext.h"
527caa17caSRaphael Isemann #include "lldb/Target/StackFrame.h"
537caa17caSRaphael Isemann #include "lldb/Target/StopInfo.h"
547caa17caSRaphael Isemann #include "lldb/Target/Target.h"
557caa17caSRaphael Isemann #include "lldb/Target/Thread.h"
567caa17caSRaphael Isemann #include "lldb/Utility/State.h"
577caa17caSRaphael Isemann #endif
587caa17caSRaphael Isemann 
597caa17caSRaphael Isemann #include "llvm/ADT/StringRef.h"
607caa17caSRaphael Isemann 
617caa17caSRaphael Isemann #ifdef _WIN32
627caa17caSRaphael Isemann #include "lldb/Host/windows/windows.h"
637caa17caSRaphael Isemann #endif
647caa17caSRaphael Isemann 
657caa17caSRaphael Isemann #include <memory>
667caa17caSRaphael Isemann #include <mutex>
677caa17caSRaphael Isemann 
6876e47d48SRaphael Isemann #include <cassert>
6976e47d48SRaphael Isemann #include <cctype>
7076e47d48SRaphael Isemann #include <cerrno>
7176e47d48SRaphael Isemann #include <cstdint>
7276e47d48SRaphael Isemann #include <cstdio>
7376e47d48SRaphael Isemann #include <cstring>
7429cc50e1SOmar Emara #include <functional>
757caa17caSRaphael Isemann #include <type_traits>
767caa17caSRaphael Isemann 
777caa17caSRaphael Isemann using namespace lldb;
787caa17caSRaphael Isemann using namespace lldb_private;
797caa17caSRaphael Isemann using llvm::None;
807caa17caSRaphael Isemann using llvm::Optional;
817caa17caSRaphael Isemann using llvm::StringRef;
827caa17caSRaphael Isemann 
837caa17caSRaphael Isemann // we may want curses to be disabled for some builds for instance, windows
84a4304f96SJonas Devlieghere #if LLDB_ENABLE_CURSES
857caa17caSRaphael Isemann 
867caa17caSRaphael Isemann #define KEY_RETURN 10
877caa17caSRaphael Isemann #define KEY_ESCAPE 27
887caa17caSRaphael Isemann 
8929cc50e1SOmar Emara #define KEY_SHIFT_TAB (KEY_MAX + 1)
9029cc50e1SOmar Emara 
917caa17caSRaphael Isemann namespace curses {
927caa17caSRaphael Isemann class Menu;
937caa17caSRaphael Isemann class MenuDelegate;
947caa17caSRaphael Isemann class Window;
957caa17caSRaphael Isemann class WindowDelegate;
967caa17caSRaphael Isemann typedef std::shared_ptr<Menu> MenuSP;
977caa17caSRaphael Isemann typedef std::shared_ptr<MenuDelegate> MenuDelegateSP;
987caa17caSRaphael Isemann typedef std::shared_ptr<Window> WindowSP;
997caa17caSRaphael Isemann typedef std::shared_ptr<WindowDelegate> WindowDelegateSP;
1007caa17caSRaphael Isemann typedef std::vector<MenuSP> Menus;
1017caa17caSRaphael Isemann typedef std::vector<WindowSP> Windows;
1027caa17caSRaphael Isemann typedef std::vector<WindowDelegateSP> WindowDelegates;
1037caa17caSRaphael Isemann 
1047caa17caSRaphael Isemann #if 0
1057caa17caSRaphael Isemann type summary add -s "x=${var.x}, y=${var.y}" curses::Point
1067caa17caSRaphael Isemann type summary add -s "w=${var.width}, h=${var.height}" curses::Size
1077caa17caSRaphael Isemann type summary add -s "${var.origin%S} ${var.size%S}" curses::Rect
1087caa17caSRaphael Isemann #endif
1097caa17caSRaphael Isemann 
1107caa17caSRaphael Isemann struct Point {
1117caa17caSRaphael Isemann   int x;
1127caa17caSRaphael Isemann   int y;
1137caa17caSRaphael Isemann 
1147caa17caSRaphael Isemann   Point(int _x = 0, int _y = 0) : x(_x), y(_y) {}
1157caa17caSRaphael Isemann 
1167caa17caSRaphael Isemann   void Clear() {
1177caa17caSRaphael Isemann     x = 0;
1187caa17caSRaphael Isemann     y = 0;
1197caa17caSRaphael Isemann   }
1207caa17caSRaphael Isemann 
1217caa17caSRaphael Isemann   Point &operator+=(const Point &rhs) {
1227caa17caSRaphael Isemann     x += rhs.x;
1237caa17caSRaphael Isemann     y += rhs.y;
1247caa17caSRaphael Isemann     return *this;
1257caa17caSRaphael Isemann   }
1267caa17caSRaphael Isemann 
1277caa17caSRaphael Isemann   void Dump() { printf("(x=%i, y=%i)\n", x, y); }
1287caa17caSRaphael Isemann };
1297caa17caSRaphael Isemann 
1307caa17caSRaphael Isemann bool operator==(const Point &lhs, const Point &rhs) {
1317caa17caSRaphael Isemann   return lhs.x == rhs.x && lhs.y == rhs.y;
1327caa17caSRaphael Isemann }
1337caa17caSRaphael Isemann 
1347caa17caSRaphael Isemann bool operator!=(const Point &lhs, const Point &rhs) {
1357caa17caSRaphael Isemann   return lhs.x != rhs.x || lhs.y != rhs.y;
1367caa17caSRaphael Isemann }
1377caa17caSRaphael Isemann 
1387caa17caSRaphael Isemann struct Size {
1397caa17caSRaphael Isemann   int width;
1407caa17caSRaphael Isemann   int height;
1417caa17caSRaphael Isemann   Size(int w = 0, int h = 0) : width(w), height(h) {}
1427caa17caSRaphael Isemann 
1437caa17caSRaphael Isemann   void Clear() {
1447caa17caSRaphael Isemann     width = 0;
1457caa17caSRaphael Isemann     height = 0;
1467caa17caSRaphael Isemann   }
1477caa17caSRaphael Isemann 
1487caa17caSRaphael Isemann   void Dump() { printf("(w=%i, h=%i)\n", width, height); }
1497caa17caSRaphael Isemann };
1507caa17caSRaphael Isemann 
1517caa17caSRaphael Isemann bool operator==(const Size &lhs, const Size &rhs) {
1527caa17caSRaphael Isemann   return lhs.width == rhs.width && lhs.height == rhs.height;
1537caa17caSRaphael Isemann }
1547caa17caSRaphael Isemann 
1557caa17caSRaphael Isemann bool operator!=(const Size &lhs, const Size &rhs) {
1567caa17caSRaphael Isemann   return lhs.width != rhs.width || lhs.height != rhs.height;
1577caa17caSRaphael Isemann }
1587caa17caSRaphael Isemann 
1597caa17caSRaphael Isemann struct Rect {
1607caa17caSRaphael Isemann   Point origin;
1617caa17caSRaphael Isemann   Size size;
1627caa17caSRaphael Isemann 
1637caa17caSRaphael Isemann   Rect() : origin(), size() {}
1647caa17caSRaphael Isemann 
1657caa17caSRaphael Isemann   Rect(const Point &p, const Size &s) : origin(p), size(s) {}
1667caa17caSRaphael Isemann 
1677caa17caSRaphael Isemann   void Clear() {
1687caa17caSRaphael Isemann     origin.Clear();
1697caa17caSRaphael Isemann     size.Clear();
1707caa17caSRaphael Isemann   }
1717caa17caSRaphael Isemann 
1727caa17caSRaphael Isemann   void Dump() {
1737caa17caSRaphael Isemann     printf("(x=%i, y=%i), w=%i, h=%i)\n", origin.x, origin.y, size.width,
1747caa17caSRaphael Isemann            size.height);
1757caa17caSRaphael Isemann   }
1767caa17caSRaphael Isemann 
1777caa17caSRaphael Isemann   void Inset(int w, int h) {
1787caa17caSRaphael Isemann     if (size.width > w * 2)
1797caa17caSRaphael Isemann       size.width -= w * 2;
1807caa17caSRaphael Isemann     origin.x += w;
1817caa17caSRaphael Isemann 
1827caa17caSRaphael Isemann     if (size.height > h * 2)
1837caa17caSRaphael Isemann       size.height -= h * 2;
1847caa17caSRaphael Isemann     origin.y += h;
1857caa17caSRaphael Isemann   }
1867caa17caSRaphael Isemann 
1877caa17caSRaphael Isemann   // Return a status bar rectangle which is the last line of this rectangle.
1887caa17caSRaphael Isemann   // This rectangle will be modified to not include the status bar area.
1897caa17caSRaphael Isemann   Rect MakeStatusBar() {
1907caa17caSRaphael Isemann     Rect status_bar;
1917caa17caSRaphael Isemann     if (size.height > 1) {
1927caa17caSRaphael Isemann       status_bar.origin.x = origin.x;
1937caa17caSRaphael Isemann       status_bar.origin.y = size.height;
1947caa17caSRaphael Isemann       status_bar.size.width = size.width;
1957caa17caSRaphael Isemann       status_bar.size.height = 1;
1967caa17caSRaphael Isemann       --size.height;
1977caa17caSRaphael Isemann     }
1987caa17caSRaphael Isemann     return status_bar;
1997caa17caSRaphael Isemann   }
2007caa17caSRaphael Isemann 
2017caa17caSRaphael Isemann   // Return a menubar rectangle which is the first line of this rectangle. This
2027caa17caSRaphael Isemann   // rectangle will be modified to not include the menubar area.
2037caa17caSRaphael Isemann   Rect MakeMenuBar() {
2047caa17caSRaphael Isemann     Rect menubar;
2057caa17caSRaphael Isemann     if (size.height > 1) {
2067caa17caSRaphael Isemann       menubar.origin.x = origin.x;
2077caa17caSRaphael Isemann       menubar.origin.y = origin.y;
2087caa17caSRaphael Isemann       menubar.size.width = size.width;
2097caa17caSRaphael Isemann       menubar.size.height = 1;
2107caa17caSRaphael Isemann       ++origin.y;
2117caa17caSRaphael Isemann       --size.height;
2127caa17caSRaphael Isemann     }
2137caa17caSRaphael Isemann     return menubar;
2147caa17caSRaphael Isemann   }
2157caa17caSRaphael Isemann 
2167caa17caSRaphael Isemann   void HorizontalSplitPercentage(float top_percentage, Rect &top,
2177caa17caSRaphael Isemann                                  Rect &bottom) const {
2187caa17caSRaphael Isemann     float top_height = top_percentage * size.height;
2197caa17caSRaphael Isemann     HorizontalSplit(top_height, top, bottom);
2207caa17caSRaphael Isemann   }
2217caa17caSRaphael Isemann 
2227caa17caSRaphael Isemann   void HorizontalSplit(int top_height, Rect &top, Rect &bottom) const {
2237caa17caSRaphael Isemann     top = *this;
2247caa17caSRaphael Isemann     if (top_height < size.height) {
2257caa17caSRaphael Isemann       top.size.height = top_height;
2267caa17caSRaphael Isemann       bottom.origin.x = origin.x;
2277caa17caSRaphael Isemann       bottom.origin.y = origin.y + top.size.height;
2287caa17caSRaphael Isemann       bottom.size.width = size.width;
2297caa17caSRaphael Isemann       bottom.size.height = size.height - top.size.height;
2307caa17caSRaphael Isemann     } else {
2317caa17caSRaphael Isemann       bottom.Clear();
2327caa17caSRaphael Isemann     }
2337caa17caSRaphael Isemann   }
2347caa17caSRaphael Isemann 
2357caa17caSRaphael Isemann   void VerticalSplitPercentage(float left_percentage, Rect &left,
2367caa17caSRaphael Isemann                                Rect &right) const {
2377caa17caSRaphael Isemann     float left_width = left_percentage * size.width;
2387caa17caSRaphael Isemann     VerticalSplit(left_width, left, right);
2397caa17caSRaphael Isemann   }
2407caa17caSRaphael Isemann 
2417caa17caSRaphael Isemann   void VerticalSplit(int left_width, Rect &left, Rect &right) const {
2427caa17caSRaphael Isemann     left = *this;
2437caa17caSRaphael Isemann     if (left_width < size.width) {
2447caa17caSRaphael Isemann       left.size.width = left_width;
2457caa17caSRaphael Isemann       right.origin.x = origin.x + left.size.width;
2467caa17caSRaphael Isemann       right.origin.y = origin.y;
2477caa17caSRaphael Isemann       right.size.width = size.width - left.size.width;
2487caa17caSRaphael Isemann       right.size.height = size.height;
2497caa17caSRaphael Isemann     } else {
2507caa17caSRaphael Isemann       right.Clear();
2517caa17caSRaphael Isemann     }
2527caa17caSRaphael Isemann   }
2537caa17caSRaphael Isemann };
2547caa17caSRaphael Isemann 
2557caa17caSRaphael Isemann bool operator==(const Rect &lhs, const Rect &rhs) {
2567caa17caSRaphael Isemann   return lhs.origin == rhs.origin && lhs.size == rhs.size;
2577caa17caSRaphael Isemann }
2587caa17caSRaphael Isemann 
2597caa17caSRaphael Isemann bool operator!=(const Rect &lhs, const Rect &rhs) {
2607caa17caSRaphael Isemann   return lhs.origin != rhs.origin || lhs.size != rhs.size;
2617caa17caSRaphael Isemann }
2627caa17caSRaphael Isemann 
2637caa17caSRaphael Isemann enum HandleCharResult {
2647caa17caSRaphael Isemann   eKeyNotHandled = 0,
2657caa17caSRaphael Isemann   eKeyHandled = 1,
2667caa17caSRaphael Isemann   eQuitApplication = 2
2677caa17caSRaphael Isemann };
2687caa17caSRaphael Isemann 
2697caa17caSRaphael Isemann enum class MenuActionResult {
2707caa17caSRaphael Isemann   Handled,
2717caa17caSRaphael Isemann   NotHandled,
2727caa17caSRaphael Isemann   Quit // Exit all menus and quit
2737caa17caSRaphael Isemann };
2747caa17caSRaphael Isemann 
2757caa17caSRaphael Isemann struct KeyHelp {
2767caa17caSRaphael Isemann   int ch;
2777caa17caSRaphael Isemann   const char *description;
2787caa17caSRaphael Isemann };
2797caa17caSRaphael Isemann 
2804a8e4b5cSLuboš Luňák // COLOR_PAIR index names
2814a8e4b5cSLuboš Luňák enum {
2824a8e4b5cSLuboš Luňák   // First 16 colors are 8 black background and 8 blue background colors,
2834a8e4b5cSLuboš Luňák   // needed by OutputColoredStringTruncated().
2844a8e4b5cSLuboš Luňák   BlackOnBlack = 1,
2854a8e4b5cSLuboš Luňák   RedOnBlack,
2864a8e4b5cSLuboš Luňák   GreenOnBlack,
2874a8e4b5cSLuboš Luňák   YellowOnBlack,
2884a8e4b5cSLuboš Luňák   BlueOnBlack,
2894a8e4b5cSLuboš Luňák   MagentaOnBlack,
2904a8e4b5cSLuboš Luňák   CyanOnBlack,
2914a8e4b5cSLuboš Luňák   WhiteOnBlack,
2924a8e4b5cSLuboš Luňák   BlackOnBlue,
2934a8e4b5cSLuboš Luňák   RedOnBlue,
2944a8e4b5cSLuboš Luňák   GreenOnBlue,
2954a8e4b5cSLuboš Luňák   YellowOnBlue,
2964a8e4b5cSLuboš Luňák   BlueOnBlue,
2974a8e4b5cSLuboš Luňák   MagentaOnBlue,
2984a8e4b5cSLuboš Luňák   CyanOnBlue,
2994a8e4b5cSLuboš Luňák   WhiteOnBlue,
3004a8e4b5cSLuboš Luňák   // Other colors, as needed.
3014a8e4b5cSLuboš Luňák   BlackOnWhite,
3024a8e4b5cSLuboš Luňák   MagentaOnWhite,
3034a8e4b5cSLuboš Luňák   LastColorPairIndex = MagentaOnWhite
3044a8e4b5cSLuboš Luňák };
3054a8e4b5cSLuboš Luňák 
3067caa17caSRaphael Isemann class WindowDelegate {
3077caa17caSRaphael Isemann public:
3087caa17caSRaphael Isemann   virtual ~WindowDelegate() = default;
3097caa17caSRaphael Isemann 
3107caa17caSRaphael Isemann   virtual bool WindowDelegateDraw(Window &window, bool force) {
3117caa17caSRaphael Isemann     return false; // Drawing not handled
3127caa17caSRaphael Isemann   }
3137caa17caSRaphael Isemann 
3147caa17caSRaphael Isemann   virtual HandleCharResult WindowDelegateHandleChar(Window &window, int key) {
3157caa17caSRaphael Isemann     return eKeyNotHandled;
3167caa17caSRaphael Isemann   }
3177caa17caSRaphael Isemann 
3187caa17caSRaphael Isemann   virtual const char *WindowDelegateGetHelpText() { return nullptr; }
3197caa17caSRaphael Isemann 
3207caa17caSRaphael Isemann   virtual KeyHelp *WindowDelegateGetKeyHelp() { return nullptr; }
3217caa17caSRaphael Isemann };
3227caa17caSRaphael Isemann 
3237caa17caSRaphael Isemann class HelpDialogDelegate : public WindowDelegate {
3247caa17caSRaphael Isemann public:
3257caa17caSRaphael Isemann   HelpDialogDelegate(const char *text, KeyHelp *key_help_array);
3267caa17caSRaphael Isemann 
3277caa17caSRaphael Isemann   ~HelpDialogDelegate() override;
3287caa17caSRaphael Isemann 
3297caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override;
3307caa17caSRaphael Isemann 
3317caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
3327caa17caSRaphael Isemann 
3337caa17caSRaphael Isemann   size_t GetNumLines() const { return m_text.GetSize(); }
3347caa17caSRaphael Isemann 
3357caa17caSRaphael Isemann   size_t GetMaxLineLength() const { return m_text.GetMaxStringLength(); }
3367caa17caSRaphael Isemann 
3377caa17caSRaphael Isemann protected:
3387caa17caSRaphael Isemann   StringList m_text;
3397caa17caSRaphael Isemann   int m_first_visible_line;
3407caa17caSRaphael Isemann };
3417caa17caSRaphael Isemann 
34229cc50e1SOmar Emara // A surface is an abstraction for something than can be drawn on. The surface
34329cc50e1SOmar Emara // have a width, a height, a cursor position, and a multitude of drawing
34429cc50e1SOmar Emara // operations. This type should be sub-classed to get an actually useful ncurses
34529cc50e1SOmar Emara // object, such as a Window, SubWindow, Pad, or a SubPad.
34629cc50e1SOmar Emara class Surface {
3477caa17caSRaphael Isemann public:
34829cc50e1SOmar Emara   Surface() : m_window(nullptr) {}
3497caa17caSRaphael Isemann 
35029cc50e1SOmar Emara   WINDOW *get() { return m_window; }
35129cc50e1SOmar Emara 
35229cc50e1SOmar Emara   operator WINDOW *() { return m_window; }
35329cc50e1SOmar Emara 
35429cc50e1SOmar Emara   // Copy a region of the surface to another surface.
35529cc50e1SOmar Emara   void CopyToSurface(Surface &target, Point source_origin, Point target_origin,
35629cc50e1SOmar Emara                      Size size) {
35729cc50e1SOmar Emara     ::copywin(m_window, target.get(), source_origin.y, source_origin.x,
35829cc50e1SOmar Emara               target_origin.y, target_origin.x,
35929cc50e1SOmar Emara               target_origin.y + size.height - 1,
36029cc50e1SOmar Emara               target_origin.x + size.width - 1, false);
3617caa17caSRaphael Isemann   }
3627caa17caSRaphael Isemann 
36329cc50e1SOmar Emara   int GetCursorX() const { return getcurx(m_window); }
36429cc50e1SOmar Emara   int GetCursorY() const { return getcury(m_window); }
36529cc50e1SOmar Emara   void MoveCursor(int x, int y) { ::wmove(m_window, y, x); }
3667caa17caSRaphael Isemann 
3677caa17caSRaphael Isemann   void AttributeOn(attr_t attr) { ::wattron(m_window, attr); }
3687caa17caSRaphael Isemann   void AttributeOff(attr_t attr) { ::wattroff(m_window, attr); }
36929cc50e1SOmar Emara 
370fc0e8fb7SLuboš Luňák   int GetMaxX() const { return getmaxx(m_window); }
371fc0e8fb7SLuboš Luňák   int GetMaxY() const { return getmaxy(m_window); }
372fc0e8fb7SLuboš Luňák   int GetWidth() const { return GetMaxX(); }
373fc0e8fb7SLuboš Luňák   int GetHeight() const { return GetMaxY(); }
37429cc50e1SOmar Emara   Size GetSize() const { return Size(GetWidth(), GetHeight()); }
37529cc50e1SOmar Emara   // Get a zero origin rectangle width the surface size.
37629cc50e1SOmar Emara   Rect GetFrame() const { return Rect(Point(), GetSize()); }
37729cc50e1SOmar Emara 
37829cc50e1SOmar Emara   void Clear() { ::wclear(m_window); }
37929cc50e1SOmar Emara   void Erase() { ::werase(m_window); }
38029cc50e1SOmar Emara 
3817caa17caSRaphael Isemann   void SetBackground(int color_pair_idx) {
3827caa17caSRaphael Isemann     ::wbkgd(m_window, COLOR_PAIR(color_pair_idx));
3837caa17caSRaphael Isemann   }
3847caa17caSRaphael Isemann 
38529cc50e1SOmar Emara   void PutChar(int ch) { ::waddch(m_window, ch); }
38629cc50e1SOmar Emara   void PutCString(const char *s, int len = -1) { ::waddnstr(m_window, s, len); }
38729cc50e1SOmar Emara 
38814406ca0SLuboš Luňák   void PutCStringTruncated(int right_pad, const char *s, int len = -1) {
3897caa17caSRaphael Isemann     int bytes_left = GetWidth() - GetCursorX();
3907caa17caSRaphael Isemann     if (bytes_left > right_pad) {
3917caa17caSRaphael Isemann       bytes_left -= right_pad;
39214406ca0SLuboš Luňák       ::waddnstr(m_window, s, len < 0 ? bytes_left : std::min(bytes_left, len));
3937caa17caSRaphael Isemann     }
3947caa17caSRaphael Isemann   }
3957caa17caSRaphael Isemann 
3967caa17caSRaphael Isemann   void Printf(const char *format, ...) __attribute__((format(printf, 2, 3))) {
3977caa17caSRaphael Isemann     va_list args;
3987caa17caSRaphael Isemann     va_start(args, format);
3993b7795aeSRaphael Isemann     vw_printw(m_window, format, args);
4007caa17caSRaphael Isemann     va_end(args);
4017caa17caSRaphael Isemann   }
4027caa17caSRaphael Isemann 
403fc0e8fb7SLuboš Luňák   void PrintfTruncated(int right_pad, const char *format, ...)
404fc0e8fb7SLuboš Luňák       __attribute__((format(printf, 3, 4))) {
405fc0e8fb7SLuboš Luňák     va_list args;
406fc0e8fb7SLuboš Luňák     va_start(args, format);
407fc0e8fb7SLuboš Luňák     StreamString strm;
408fc0e8fb7SLuboš Luňák     strm.PrintfVarArg(format, args);
409fc0e8fb7SLuboš Luňák     va_end(args);
410fc0e8fb7SLuboš Luňák     PutCStringTruncated(right_pad, strm.GetData());
411fc0e8fb7SLuboš Luňák   }
412fc0e8fb7SLuboš Luňák 
41329cc50e1SOmar Emara   void VerticalLine(int n, chtype v_char = ACS_VLINE) {
41429cc50e1SOmar Emara     ::wvline(m_window, v_char, n);
41529cc50e1SOmar Emara   }
41629cc50e1SOmar Emara   void HorizontalLine(int n, chtype h_char = ACS_HLINE) {
41729cc50e1SOmar Emara     ::whline(m_window, h_char, n);
41829cc50e1SOmar Emara   }
41929cc50e1SOmar Emara   void Box(chtype v_char = ACS_VLINE, chtype h_char = ACS_HLINE) {
42029cc50e1SOmar Emara     ::box(m_window, v_char, h_char);
42129cc50e1SOmar Emara   }
42229cc50e1SOmar Emara 
42329cc50e1SOmar Emara   void TitledBox(const char *title, chtype v_char = ACS_VLINE,
42429cc50e1SOmar Emara                  chtype h_char = ACS_HLINE) {
42529cc50e1SOmar Emara     Box(v_char, h_char);
42629cc50e1SOmar Emara     int title_offset = 2;
42729cc50e1SOmar Emara     MoveCursor(title_offset, 0);
42829cc50e1SOmar Emara     PutChar('[');
42929cc50e1SOmar Emara     PutCString(title, GetWidth() - title_offset);
43029cc50e1SOmar Emara     PutChar(']');
43129cc50e1SOmar Emara   }
43229cc50e1SOmar Emara 
43329cc50e1SOmar Emara   void Box(const Rect &bounds, chtype v_char = ACS_VLINE,
43429cc50e1SOmar Emara            chtype h_char = ACS_HLINE) {
43529cc50e1SOmar Emara     MoveCursor(bounds.origin.x, bounds.origin.y);
43629cc50e1SOmar Emara     VerticalLine(bounds.size.height);
43729cc50e1SOmar Emara     HorizontalLine(bounds.size.width);
43829cc50e1SOmar Emara     PutChar(ACS_ULCORNER);
43929cc50e1SOmar Emara 
44029cc50e1SOmar Emara     MoveCursor(bounds.origin.x + bounds.size.width - 1, bounds.origin.y);
44129cc50e1SOmar Emara     VerticalLine(bounds.size.height);
44229cc50e1SOmar Emara     PutChar(ACS_URCORNER);
44329cc50e1SOmar Emara 
44429cc50e1SOmar Emara     MoveCursor(bounds.origin.x, bounds.origin.y + bounds.size.height - 1);
44529cc50e1SOmar Emara     HorizontalLine(bounds.size.width);
44629cc50e1SOmar Emara     PutChar(ACS_LLCORNER);
44729cc50e1SOmar Emara 
44829cc50e1SOmar Emara     MoveCursor(bounds.origin.x + bounds.size.width - 1,
44929cc50e1SOmar Emara                bounds.origin.y + bounds.size.height - 1);
45029cc50e1SOmar Emara     PutChar(ACS_LRCORNER);
45129cc50e1SOmar Emara   }
45229cc50e1SOmar Emara 
45329cc50e1SOmar Emara   void TitledBox(const Rect &bounds, const char *title,
45429cc50e1SOmar Emara                  chtype v_char = ACS_VLINE, chtype h_char = ACS_HLINE) {
45529cc50e1SOmar Emara     Box(bounds, v_char, h_char);
45629cc50e1SOmar Emara     int title_offset = 2;
45729cc50e1SOmar Emara     MoveCursor(bounds.origin.x + title_offset, bounds.origin.y);
45829cc50e1SOmar Emara     PutChar('[');
45929cc50e1SOmar Emara     PutCString(title, bounds.size.width - title_offset);
46029cc50e1SOmar Emara     PutChar(']');
461fc0e8fb7SLuboš Luňák   }
462fc0e8fb7SLuboš Luňák 
46314406ca0SLuboš Luňák   // Curses doesn't allow direct output of color escape sequences, but that's
46414406ca0SLuboš Luňák   // how we get source lines from the Highligher class. Read the line and
465dcd4589aSLuboš Luňák   // convert color escape sequences to curses color attributes. Use
466dcd4589aSLuboš Luňák   // first_skip_count to skip leading visible characters. Returns false if all
467dcd4589aSLuboš Luňák   // visible characters were skipped due to first_skip_count.
468dcd4589aSLuboš Luňák   bool OutputColoredStringTruncated(int right_pad, StringRef string,
469dcd4589aSLuboš Luňák                                     size_t skip_first_count,
47014406ca0SLuboš Luňák                                     bool use_blue_background) {
47114406ca0SLuboš Luňák     attr_t saved_attr;
47214406ca0SLuboš Luňák     short saved_pair;
473dcd4589aSLuboš Luňák     bool result = false;
4749dbdaea9SSterling Augustine     wattr_get(m_window, &saved_attr, &saved_pair, nullptr);
47514406ca0SLuboš Luňák     if (use_blue_background)
4764a8e4b5cSLuboš Luňák       ::wattron(m_window, COLOR_PAIR(WhiteOnBlue));
47714406ca0SLuboš Luňák     while (!string.empty()) {
47814406ca0SLuboš Luňák       size_t esc_pos = string.find('\x1b');
47914406ca0SLuboš Luňák       if (esc_pos == StringRef::npos) {
480dcd4589aSLuboš Luňák         string = string.substr(skip_first_count);
481dcd4589aSLuboš Luňák         if (!string.empty()) {
48214406ca0SLuboš Luňák           PutCStringTruncated(right_pad, string.data(), string.size());
483dcd4589aSLuboš Luňák           result = true;
484dcd4589aSLuboš Luňák         }
48514406ca0SLuboš Luňák         break;
48614406ca0SLuboš Luňák       }
48714406ca0SLuboš Luňák       if (esc_pos > 0) {
488dcd4589aSLuboš Luňák         if (skip_first_count > 0) {
489dcd4589aSLuboš Luňák           int skip = std::min(esc_pos, skip_first_count);
490dcd4589aSLuboš Luňák           string = string.substr(skip);
491dcd4589aSLuboš Luňák           skip_first_count -= skip;
492dcd4589aSLuboš Luňák           esc_pos -= skip;
493dcd4589aSLuboš Luňák         }
494dcd4589aSLuboš Luňák         if (esc_pos > 0) {
49514406ca0SLuboš Luňák           PutCStringTruncated(right_pad, string.data(), esc_pos);
496dcd4589aSLuboš Luňák           result = true;
49714406ca0SLuboš Luňák           string = string.drop_front(esc_pos);
49814406ca0SLuboš Luňák         }
499dcd4589aSLuboš Luňák       }
50014406ca0SLuboš Luňák       bool consumed = string.consume_front("\x1b");
50114406ca0SLuboš Luňák       assert(consumed);
50214406ca0SLuboš Luňák       UNUSED_IF_ASSERT_DISABLED(consumed);
50314406ca0SLuboš Luňák       // This is written to match our Highlighter classes, which seem to
50414406ca0SLuboš Luňák       // generate only foreground color escape sequences. If necessary, this
50514406ca0SLuboš Luňák       // will need to be extended.
50614406ca0SLuboš Luňák       if (!string.consume_front("[")) {
50714406ca0SLuboš Luňák         llvm::errs() << "Missing '[' in color escape sequence.\n";
50814406ca0SLuboš Luňák         continue;
50914406ca0SLuboš Luňák       }
51014406ca0SLuboš Luňák       // Only 8 basic foreground colors and reset, our Highlighter doesn't use
51114406ca0SLuboš Luňák       // anything else.
51214406ca0SLuboš Luňák       int value;
51314406ca0SLuboš Luňák       if (!!string.consumeInteger(10, value) || // Returns false on success.
51414406ca0SLuboš Luňák           !(value == 0 || (value >= 30 && value <= 37))) {
51514406ca0SLuboš Luňák         llvm::errs() << "No valid color code in color escape sequence.\n";
51614406ca0SLuboš Luňák         continue;
51714406ca0SLuboš Luňák       }
51814406ca0SLuboš Luňák       if (!string.consume_front("m")) {
51914406ca0SLuboš Luňák         llvm::errs() << "Missing 'm' in color escape sequence.\n";
52014406ca0SLuboš Luňák         continue;
52114406ca0SLuboš Luňák       }
52214406ca0SLuboš Luňák       if (value == 0) { // Reset.
5239dbdaea9SSterling Augustine         wattr_set(m_window, saved_attr, saved_pair, nullptr);
52414406ca0SLuboš Luňák         if (use_blue_background)
5254a8e4b5cSLuboš Luňák           ::wattron(m_window, COLOR_PAIR(WhiteOnBlue));
52614406ca0SLuboš Luňák       } else {
52714406ca0SLuboš Luňák         // Mapped directly to first 16 color pairs (black/blue background).
52814406ca0SLuboš Luňák         ::wattron(m_window,
52914406ca0SLuboš Luňák                   COLOR_PAIR(value - 30 + 1 + (use_blue_background ? 8 : 0)));
53014406ca0SLuboš Luňák       }
53114406ca0SLuboš Luňák     }
5329dbdaea9SSterling Augustine     wattr_set(m_window, saved_attr, saved_pair, nullptr);
533dcd4589aSLuboš Luňák     return result;
53414406ca0SLuboš Luňák   }
53514406ca0SLuboš Luňák 
53629cc50e1SOmar Emara protected:
53729cc50e1SOmar Emara   WINDOW *m_window;
53829cc50e1SOmar Emara };
53929cc50e1SOmar Emara 
54029cc50e1SOmar Emara class Pad : public Surface {
54129cc50e1SOmar Emara public:
54229cc50e1SOmar Emara   Pad(Size size) { m_window = ::newpad(size.height, size.width); }
54329cc50e1SOmar Emara 
54429cc50e1SOmar Emara   ~Pad() { ::delwin(m_window); }
54529cc50e1SOmar Emara };
54629cc50e1SOmar Emara 
54729cc50e1SOmar Emara class SubPad : public Surface {
54829cc50e1SOmar Emara public:
54929cc50e1SOmar Emara   SubPad(Pad &pad, Rect bounds) {
55029cc50e1SOmar Emara     m_window = ::subpad(pad.get(), bounds.size.height, bounds.size.width,
55129cc50e1SOmar Emara                         bounds.origin.y, bounds.origin.x);
55229cc50e1SOmar Emara   }
55329cc50e1SOmar Emara   SubPad(SubPad &subpad, Rect bounds) {
55429cc50e1SOmar Emara     m_window = ::subpad(subpad.get(), bounds.size.height, bounds.size.width,
55529cc50e1SOmar Emara                         bounds.origin.y, bounds.origin.x);
55629cc50e1SOmar Emara   }
55729cc50e1SOmar Emara 
55829cc50e1SOmar Emara   ~SubPad() { ::delwin(m_window); }
55929cc50e1SOmar Emara };
56029cc50e1SOmar Emara 
56129cc50e1SOmar Emara class Window : public Surface {
56229cc50e1SOmar Emara public:
56329cc50e1SOmar Emara   Window(const char *name)
56429cc50e1SOmar Emara       : m_name(name), m_panel(nullptr), m_parent(nullptr), m_subwindows(),
56529cc50e1SOmar Emara         m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
56629cc50e1SOmar Emara         m_prev_active_window_idx(UINT32_MAX), m_delete(false),
56729cc50e1SOmar Emara         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {}
56829cc50e1SOmar Emara 
56929cc50e1SOmar Emara   Window(const char *name, WINDOW *w, bool del = true)
57029cc50e1SOmar Emara       : m_name(name), m_panel(nullptr), m_parent(nullptr), m_subwindows(),
57129cc50e1SOmar Emara         m_delegate_sp(), m_curr_active_window_idx(UINT32_MAX),
57229cc50e1SOmar Emara         m_prev_active_window_idx(UINT32_MAX), m_delete(del),
57329cc50e1SOmar Emara         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
57429cc50e1SOmar Emara     if (w)
57529cc50e1SOmar Emara       Reset(w);
57629cc50e1SOmar Emara   }
57729cc50e1SOmar Emara 
57829cc50e1SOmar Emara   Window(const char *name, const Rect &bounds)
57929cc50e1SOmar Emara       : m_name(name), m_parent(nullptr), m_subwindows(), m_delegate_sp(),
58029cc50e1SOmar Emara         m_curr_active_window_idx(UINT32_MAX),
58129cc50e1SOmar Emara         m_prev_active_window_idx(UINT32_MAX), m_delete(true),
58229cc50e1SOmar Emara         m_needs_update(true), m_can_activate(true), m_is_subwin(false) {
58329cc50e1SOmar Emara     Reset(::newwin(bounds.size.height, bounds.size.width, bounds.origin.y,
58429cc50e1SOmar Emara                    bounds.origin.y));
58529cc50e1SOmar Emara   }
58629cc50e1SOmar Emara 
58729cc50e1SOmar Emara   virtual ~Window() {
58829cc50e1SOmar Emara     RemoveSubWindows();
58929cc50e1SOmar Emara     Reset();
59029cc50e1SOmar Emara   }
59129cc50e1SOmar Emara 
59229cc50e1SOmar Emara   void Reset(WINDOW *w = nullptr, bool del = true) {
59329cc50e1SOmar Emara     if (m_window == w)
59429cc50e1SOmar Emara       return;
59529cc50e1SOmar Emara 
59629cc50e1SOmar Emara     if (m_panel) {
59729cc50e1SOmar Emara       ::del_panel(m_panel);
59829cc50e1SOmar Emara       m_panel = nullptr;
59929cc50e1SOmar Emara     }
60029cc50e1SOmar Emara     if (m_window && m_delete) {
60129cc50e1SOmar Emara       ::delwin(m_window);
60229cc50e1SOmar Emara       m_window = nullptr;
60329cc50e1SOmar Emara       m_delete = false;
60429cc50e1SOmar Emara     }
60529cc50e1SOmar Emara     if (w) {
60629cc50e1SOmar Emara       m_window = w;
60729cc50e1SOmar Emara       m_panel = ::new_panel(m_window);
60829cc50e1SOmar Emara       m_delete = del;
60929cc50e1SOmar Emara     }
61029cc50e1SOmar Emara   }
611*0321dbc8SOmar Emara 
61229cc50e1SOmar Emara   // Get the rectangle in our parent window
61329cc50e1SOmar Emara   Rect GetBounds() const { return Rect(GetParentOrigin(), GetSize()); }
614*0321dbc8SOmar Emara 
615*0321dbc8SOmar Emara   Rect GetCenteredRect(int width, int height) {
616*0321dbc8SOmar Emara     Size size = GetSize();
617*0321dbc8SOmar Emara     width = std::min(size.width, width);
618*0321dbc8SOmar Emara     height = std::min(size.height, height);
619*0321dbc8SOmar Emara     int x = (size.width - width) / 2;
620*0321dbc8SOmar Emara     int y = (size.height - height) / 2;
621*0321dbc8SOmar Emara     return Rect(Point(x, y), Size(width, height));
622*0321dbc8SOmar Emara   }
623*0321dbc8SOmar Emara 
62429cc50e1SOmar Emara   int GetChar() { return ::wgetch(m_window); }
62529cc50e1SOmar Emara   Point GetParentOrigin() const { return Point(GetParentX(), GetParentY()); }
62629cc50e1SOmar Emara   int GetParentX() const { return getparx(m_window); }
62729cc50e1SOmar Emara   int GetParentY() const { return getpary(m_window); }
62829cc50e1SOmar Emara   void MoveWindow(int x, int y) { MoveWindow(Point(x, y)); }
62929cc50e1SOmar Emara   void Resize(int w, int h) { ::wresize(m_window, h, w); }
63029cc50e1SOmar Emara   void Resize(const Size &size) {
63129cc50e1SOmar Emara     ::wresize(m_window, size.height, size.width);
63229cc50e1SOmar Emara   }
63329cc50e1SOmar Emara   void MoveWindow(const Point &origin) {
63429cc50e1SOmar Emara     const bool moving_window = origin != GetParentOrigin();
63529cc50e1SOmar Emara     if (m_is_subwin && moving_window) {
63629cc50e1SOmar Emara       // Can't move subwindows, must delete and re-create
63729cc50e1SOmar Emara       Size size = GetSize();
63829cc50e1SOmar Emara       Reset(::subwin(m_parent->m_window, size.height, size.width, origin.y,
63929cc50e1SOmar Emara                      origin.x),
64029cc50e1SOmar Emara             true);
64129cc50e1SOmar Emara     } else {
64229cc50e1SOmar Emara       ::mvwin(m_window, origin.y, origin.x);
64329cc50e1SOmar Emara     }
64429cc50e1SOmar Emara   }
64529cc50e1SOmar Emara 
64629cc50e1SOmar Emara   void SetBounds(const Rect &bounds) {
64729cc50e1SOmar Emara     const bool moving_window = bounds.origin != GetParentOrigin();
64829cc50e1SOmar Emara     if (m_is_subwin && moving_window) {
64929cc50e1SOmar Emara       // Can't move subwindows, must delete and re-create
65029cc50e1SOmar Emara       Reset(::subwin(m_parent->m_window, bounds.size.height, bounds.size.width,
65129cc50e1SOmar Emara                      bounds.origin.y, bounds.origin.x),
65229cc50e1SOmar Emara             true);
65329cc50e1SOmar Emara     } else {
65429cc50e1SOmar Emara       if (moving_window)
65529cc50e1SOmar Emara         MoveWindow(bounds.origin);
65629cc50e1SOmar Emara       Resize(bounds.size);
65729cc50e1SOmar Emara     }
65829cc50e1SOmar Emara   }
65929cc50e1SOmar Emara 
6607caa17caSRaphael Isemann   void Touch() {
6617caa17caSRaphael Isemann     ::touchwin(m_window);
6627caa17caSRaphael Isemann     if (m_parent)
6637caa17caSRaphael Isemann       m_parent->Touch();
6647caa17caSRaphael Isemann   }
6657caa17caSRaphael Isemann 
6667caa17caSRaphael Isemann   WindowSP CreateSubWindow(const char *name, const Rect &bounds,
6677caa17caSRaphael Isemann                            bool make_active) {
6687caa17caSRaphael Isemann     auto get_window = [this, &bounds]() {
6697caa17caSRaphael Isemann       return m_window
6707caa17caSRaphael Isemann                  ? ::subwin(m_window, bounds.size.height, bounds.size.width,
6717caa17caSRaphael Isemann                             bounds.origin.y, bounds.origin.x)
6727caa17caSRaphael Isemann                  : ::newwin(bounds.size.height, bounds.size.width,
6737caa17caSRaphael Isemann                             bounds.origin.y, bounds.origin.x);
6747caa17caSRaphael Isemann     };
6757caa17caSRaphael Isemann     WindowSP subwindow_sp = std::make_shared<Window>(name, get_window(), true);
6767caa17caSRaphael Isemann     subwindow_sp->m_is_subwin = subwindow_sp.operator bool();
6777caa17caSRaphael Isemann     subwindow_sp->m_parent = this;
6787caa17caSRaphael Isemann     if (make_active) {
6797caa17caSRaphael Isemann       m_prev_active_window_idx = m_curr_active_window_idx;
6807caa17caSRaphael Isemann       m_curr_active_window_idx = m_subwindows.size();
6817caa17caSRaphael Isemann     }
6827caa17caSRaphael Isemann     m_subwindows.push_back(subwindow_sp);
6837caa17caSRaphael Isemann     ::top_panel(subwindow_sp->m_panel);
6847caa17caSRaphael Isemann     m_needs_update = true;
6857caa17caSRaphael Isemann     return subwindow_sp;
6867caa17caSRaphael Isemann   }
6877caa17caSRaphael Isemann 
6887caa17caSRaphael Isemann   bool RemoveSubWindow(Window *window) {
6897caa17caSRaphael Isemann     Windows::iterator pos, end = m_subwindows.end();
6907caa17caSRaphael Isemann     size_t i = 0;
6917caa17caSRaphael Isemann     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
6927caa17caSRaphael Isemann       if ((*pos).get() == window) {
6937caa17caSRaphael Isemann         if (m_prev_active_window_idx == i)
6947caa17caSRaphael Isemann           m_prev_active_window_idx = UINT32_MAX;
6957caa17caSRaphael Isemann         else if (m_prev_active_window_idx != UINT32_MAX &&
6967caa17caSRaphael Isemann                  m_prev_active_window_idx > i)
6977caa17caSRaphael Isemann           --m_prev_active_window_idx;
6987caa17caSRaphael Isemann 
6997caa17caSRaphael Isemann         if (m_curr_active_window_idx == i)
7007caa17caSRaphael Isemann           m_curr_active_window_idx = UINT32_MAX;
7017caa17caSRaphael Isemann         else if (m_curr_active_window_idx != UINT32_MAX &&
7027caa17caSRaphael Isemann                  m_curr_active_window_idx > i)
7037caa17caSRaphael Isemann           --m_curr_active_window_idx;
7047caa17caSRaphael Isemann         window->Erase();
7057caa17caSRaphael Isemann         m_subwindows.erase(pos);
7067caa17caSRaphael Isemann         m_needs_update = true;
7077caa17caSRaphael Isemann         if (m_parent)
7087caa17caSRaphael Isemann           m_parent->Touch();
7097caa17caSRaphael Isemann         else
7107caa17caSRaphael Isemann           ::touchwin(stdscr);
7117caa17caSRaphael Isemann         return true;
7127caa17caSRaphael Isemann       }
7137caa17caSRaphael Isemann     }
7147caa17caSRaphael Isemann     return false;
7157caa17caSRaphael Isemann   }
7167caa17caSRaphael Isemann 
7177caa17caSRaphael Isemann   WindowSP FindSubWindow(const char *name) {
7187caa17caSRaphael Isemann     Windows::iterator pos, end = m_subwindows.end();
7197caa17caSRaphael Isemann     size_t i = 0;
7207caa17caSRaphael Isemann     for (pos = m_subwindows.begin(); pos != end; ++pos, ++i) {
7217caa17caSRaphael Isemann       if ((*pos)->m_name == name)
7227caa17caSRaphael Isemann         return *pos;
7237caa17caSRaphael Isemann     }
7247caa17caSRaphael Isemann     return WindowSP();
7257caa17caSRaphael Isemann   }
7267caa17caSRaphael Isemann 
7277caa17caSRaphael Isemann   void RemoveSubWindows() {
7287caa17caSRaphael Isemann     m_curr_active_window_idx = UINT32_MAX;
7297caa17caSRaphael Isemann     m_prev_active_window_idx = UINT32_MAX;
7307caa17caSRaphael Isemann     for (Windows::iterator pos = m_subwindows.begin();
7317caa17caSRaphael Isemann          pos != m_subwindows.end(); pos = m_subwindows.erase(pos)) {
7327caa17caSRaphael Isemann       (*pos)->Erase();
7337caa17caSRaphael Isemann     }
7347caa17caSRaphael Isemann     if (m_parent)
7357caa17caSRaphael Isemann       m_parent->Touch();
7367caa17caSRaphael Isemann     else
7377caa17caSRaphael Isemann       ::touchwin(stdscr);
7387caa17caSRaphael Isemann   }
7397caa17caSRaphael Isemann 
7407caa17caSRaphael Isemann   // Window drawing utilities
7417caa17caSRaphael Isemann   void DrawTitleBox(const char *title, const char *bottom_message = nullptr) {
7427caa17caSRaphael Isemann     attr_t attr = 0;
7437caa17caSRaphael Isemann     if (IsActive())
7444a8e4b5cSLuboš Luňák       attr = A_BOLD | COLOR_PAIR(BlackOnWhite);
7457caa17caSRaphael Isemann     else
7467caa17caSRaphael Isemann       attr = 0;
7477caa17caSRaphael Isemann     if (attr)
7487caa17caSRaphael Isemann       AttributeOn(attr);
7497caa17caSRaphael Isemann 
7507caa17caSRaphael Isemann     Box();
7517caa17caSRaphael Isemann     MoveCursor(3, 0);
7527caa17caSRaphael Isemann 
7537caa17caSRaphael Isemann     if (title && title[0]) {
7547caa17caSRaphael Isemann       PutChar('<');
7557caa17caSRaphael Isemann       PutCString(title);
7567caa17caSRaphael Isemann       PutChar('>');
7577caa17caSRaphael Isemann     }
7587caa17caSRaphael Isemann 
7597caa17caSRaphael Isemann     if (bottom_message && bottom_message[0]) {
7607caa17caSRaphael Isemann       int bottom_message_length = strlen(bottom_message);
7617caa17caSRaphael Isemann       int x = GetWidth() - 3 - (bottom_message_length + 2);
7627caa17caSRaphael Isemann 
7637caa17caSRaphael Isemann       if (x > 0) {
7647caa17caSRaphael Isemann         MoveCursor(x, GetHeight() - 1);
7657caa17caSRaphael Isemann         PutChar('[');
7667caa17caSRaphael Isemann         PutCString(bottom_message);
7677caa17caSRaphael Isemann         PutChar(']');
7687caa17caSRaphael Isemann       } else {
7697caa17caSRaphael Isemann         MoveCursor(1, GetHeight() - 1);
7707caa17caSRaphael Isemann         PutChar('[');
771fc0e8fb7SLuboš Luňák         PutCStringTruncated(1, bottom_message);
7727caa17caSRaphael Isemann       }
7737caa17caSRaphael Isemann     }
7747caa17caSRaphael Isemann     if (attr)
7757caa17caSRaphael Isemann       AttributeOff(attr);
7767caa17caSRaphael Isemann   }
7777caa17caSRaphael Isemann 
7787caa17caSRaphael Isemann   virtual void Draw(bool force) {
7797caa17caSRaphael Isemann     if (m_delegate_sp && m_delegate_sp->WindowDelegateDraw(*this, force))
7807caa17caSRaphael Isemann       return;
7817caa17caSRaphael Isemann 
7827caa17caSRaphael Isemann     for (auto &subwindow_sp : m_subwindows)
7837caa17caSRaphael Isemann       subwindow_sp->Draw(force);
7847caa17caSRaphael Isemann   }
7857caa17caSRaphael Isemann 
7867caa17caSRaphael Isemann   bool CreateHelpSubwindow() {
7877caa17caSRaphael Isemann     if (m_delegate_sp) {
7887caa17caSRaphael Isemann       const char *text = m_delegate_sp->WindowDelegateGetHelpText();
7897caa17caSRaphael Isemann       KeyHelp *key_help = m_delegate_sp->WindowDelegateGetKeyHelp();
7907caa17caSRaphael Isemann       if ((text && text[0]) || key_help) {
7917caa17caSRaphael Isemann         std::unique_ptr<HelpDialogDelegate> help_delegate_up(
7927caa17caSRaphael Isemann             new HelpDialogDelegate(text, key_help));
7937caa17caSRaphael Isemann         const size_t num_lines = help_delegate_up->GetNumLines();
7947caa17caSRaphael Isemann         const size_t max_length = help_delegate_up->GetMaxLineLength();
7957caa17caSRaphael Isemann         Rect bounds = GetBounds();
7967caa17caSRaphael Isemann         bounds.Inset(1, 1);
7977caa17caSRaphael Isemann         if (max_length + 4 < static_cast<size_t>(bounds.size.width)) {
7987caa17caSRaphael Isemann           bounds.origin.x += (bounds.size.width - max_length + 4) / 2;
7997caa17caSRaphael Isemann           bounds.size.width = max_length + 4;
8007caa17caSRaphael Isemann         } else {
8017caa17caSRaphael Isemann           if (bounds.size.width > 100) {
8027caa17caSRaphael Isemann             const int inset_w = bounds.size.width / 4;
8037caa17caSRaphael Isemann             bounds.origin.x += inset_w;
8047caa17caSRaphael Isemann             bounds.size.width -= 2 * inset_w;
8057caa17caSRaphael Isemann           }
8067caa17caSRaphael Isemann         }
8077caa17caSRaphael Isemann 
8087caa17caSRaphael Isemann         if (num_lines + 2 < static_cast<size_t>(bounds.size.height)) {
8097caa17caSRaphael Isemann           bounds.origin.y += (bounds.size.height - num_lines + 2) / 2;
8107caa17caSRaphael Isemann           bounds.size.height = num_lines + 2;
8117caa17caSRaphael Isemann         } else {
8127caa17caSRaphael Isemann           if (bounds.size.height > 100) {
8137caa17caSRaphael Isemann             const int inset_h = bounds.size.height / 4;
8147caa17caSRaphael Isemann             bounds.origin.y += inset_h;
8157caa17caSRaphael Isemann             bounds.size.height -= 2 * inset_h;
8167caa17caSRaphael Isemann           }
8177caa17caSRaphael Isemann         }
8187caa17caSRaphael Isemann         WindowSP help_window_sp;
8197caa17caSRaphael Isemann         Window *parent_window = GetParent();
8207caa17caSRaphael Isemann         if (parent_window)
8217caa17caSRaphael Isemann           help_window_sp = parent_window->CreateSubWindow("Help", bounds, true);
8227caa17caSRaphael Isemann         else
8237caa17caSRaphael Isemann           help_window_sp = CreateSubWindow("Help", bounds, true);
8247caa17caSRaphael Isemann         help_window_sp->SetDelegate(
8257caa17caSRaphael Isemann             WindowDelegateSP(help_delegate_up.release()));
8267caa17caSRaphael Isemann         return true;
8277caa17caSRaphael Isemann       }
8287caa17caSRaphael Isemann     }
8297caa17caSRaphael Isemann     return false;
8307caa17caSRaphael Isemann   }
8317caa17caSRaphael Isemann 
8327caa17caSRaphael Isemann   virtual HandleCharResult HandleChar(int key) {
8337caa17caSRaphael Isemann     // Always check the active window first
8347caa17caSRaphael Isemann     HandleCharResult result = eKeyNotHandled;
8357caa17caSRaphael Isemann     WindowSP active_window_sp = GetActiveWindow();
8367caa17caSRaphael Isemann     if (active_window_sp) {
8377caa17caSRaphael Isemann       result = active_window_sp->HandleChar(key);
8387caa17caSRaphael Isemann       if (result != eKeyNotHandled)
8397caa17caSRaphael Isemann         return result;
8407caa17caSRaphael Isemann     }
8417caa17caSRaphael Isemann 
8427caa17caSRaphael Isemann     if (m_delegate_sp) {
8437caa17caSRaphael Isemann       result = m_delegate_sp->WindowDelegateHandleChar(*this, key);
8447caa17caSRaphael Isemann       if (result != eKeyNotHandled)
8457caa17caSRaphael Isemann         return result;
8467caa17caSRaphael Isemann     }
8477caa17caSRaphael Isemann 
8487caa17caSRaphael Isemann     // Then check for any windows that want any keys that weren't handled. This
8497caa17caSRaphael Isemann     // is typically only for a menubar. Make a copy of the subwindows in case
8507caa17caSRaphael Isemann     // any HandleChar() functions muck with the subwindows. If we don't do
8517caa17caSRaphael Isemann     // this, we can crash when iterating over the subwindows.
8527caa17caSRaphael Isemann     Windows subwindows(m_subwindows);
8537caa17caSRaphael Isemann     for (auto subwindow_sp : subwindows) {
8547caa17caSRaphael Isemann       if (!subwindow_sp->m_can_activate) {
8557caa17caSRaphael Isemann         HandleCharResult result = subwindow_sp->HandleChar(key);
8567caa17caSRaphael Isemann         if (result != eKeyNotHandled)
8577caa17caSRaphael Isemann           return result;
8587caa17caSRaphael Isemann       }
8597caa17caSRaphael Isemann     }
8607caa17caSRaphael Isemann 
8617caa17caSRaphael Isemann     return eKeyNotHandled;
8627caa17caSRaphael Isemann   }
8637caa17caSRaphael Isemann 
8647caa17caSRaphael Isemann   WindowSP GetActiveWindow() {
8657caa17caSRaphael Isemann     if (!m_subwindows.empty()) {
8667caa17caSRaphael Isemann       if (m_curr_active_window_idx >= m_subwindows.size()) {
8677caa17caSRaphael Isemann         if (m_prev_active_window_idx < m_subwindows.size()) {
8687caa17caSRaphael Isemann           m_curr_active_window_idx = m_prev_active_window_idx;
8697caa17caSRaphael Isemann           m_prev_active_window_idx = UINT32_MAX;
8707caa17caSRaphael Isemann         } else if (IsActive()) {
8717caa17caSRaphael Isemann           m_prev_active_window_idx = UINT32_MAX;
8727caa17caSRaphael Isemann           m_curr_active_window_idx = UINT32_MAX;
8737caa17caSRaphael Isemann 
8747caa17caSRaphael Isemann           // Find first window that wants to be active if this window is active
8757caa17caSRaphael Isemann           const size_t num_subwindows = m_subwindows.size();
8767caa17caSRaphael Isemann           for (size_t i = 0; i < num_subwindows; ++i) {
8777caa17caSRaphael Isemann             if (m_subwindows[i]->GetCanBeActive()) {
8787caa17caSRaphael Isemann               m_curr_active_window_idx = i;
8797caa17caSRaphael Isemann               break;
8807caa17caSRaphael Isemann             }
8817caa17caSRaphael Isemann           }
8827caa17caSRaphael Isemann         }
8837caa17caSRaphael Isemann       }
8847caa17caSRaphael Isemann 
8857caa17caSRaphael Isemann       if (m_curr_active_window_idx < m_subwindows.size())
8867caa17caSRaphael Isemann         return m_subwindows[m_curr_active_window_idx];
8877caa17caSRaphael Isemann     }
8887caa17caSRaphael Isemann     return WindowSP();
8897caa17caSRaphael Isemann   }
8907caa17caSRaphael Isemann 
8917caa17caSRaphael Isemann   bool GetCanBeActive() const { return m_can_activate; }
8927caa17caSRaphael Isemann 
8937caa17caSRaphael Isemann   void SetCanBeActive(bool b) { m_can_activate = b; }
8947caa17caSRaphael Isemann 
8957caa17caSRaphael Isemann   void SetDelegate(const WindowDelegateSP &delegate_sp) {
8967caa17caSRaphael Isemann     m_delegate_sp = delegate_sp;
8977caa17caSRaphael Isemann   }
8987caa17caSRaphael Isemann 
8997caa17caSRaphael Isemann   Window *GetParent() const { return m_parent; }
9007caa17caSRaphael Isemann 
9017caa17caSRaphael Isemann   bool IsActive() const {
9027caa17caSRaphael Isemann     if (m_parent)
9037caa17caSRaphael Isemann       return m_parent->GetActiveWindow().get() == this;
9047caa17caSRaphael Isemann     else
9057caa17caSRaphael Isemann       return true; // Top level window is always active
9067caa17caSRaphael Isemann   }
9077caa17caSRaphael Isemann 
9087caa17caSRaphael Isemann   void SelectNextWindowAsActive() {
9097caa17caSRaphael Isemann     // Move active focus to next window
9107a63dc53SLuboš Luňák     const int num_subwindows = m_subwindows.size();
9117a63dc53SLuboš Luňák     int start_idx = 0;
9127a63dc53SLuboš Luňák     if (m_curr_active_window_idx != UINT32_MAX) {
9137caa17caSRaphael Isemann       m_prev_active_window_idx = m_curr_active_window_idx;
9147a63dc53SLuboš Luňák       start_idx = m_curr_active_window_idx + 1;
9157a63dc53SLuboš Luňák     }
9167a63dc53SLuboš Luňák     for (int idx = start_idx; idx < num_subwindows; ++idx) {
9177caa17caSRaphael Isemann       if (m_subwindows[idx]->GetCanBeActive()) {
9187caa17caSRaphael Isemann         m_curr_active_window_idx = idx;
9197a63dc53SLuboš Luňák         return;
9207caa17caSRaphael Isemann       }
9217caa17caSRaphael Isemann     }
9227a63dc53SLuboš Luňák     for (int idx = 0; idx < start_idx; ++idx) {
9237caa17caSRaphael Isemann       if (m_subwindows[idx]->GetCanBeActive()) {
9247caa17caSRaphael Isemann         m_curr_active_window_idx = idx;
9257caa17caSRaphael Isemann         break;
9267caa17caSRaphael Isemann       }
9277caa17caSRaphael Isemann     }
9287caa17caSRaphael Isemann   }
9297a63dc53SLuboš Luňák 
9307a63dc53SLuboš Luňák   void SelectPreviousWindowAsActive() {
9317a63dc53SLuboš Luňák     // Move active focus to previous window
9327a63dc53SLuboš Luňák     const int num_subwindows = m_subwindows.size();
9337a63dc53SLuboš Luňák     int start_idx = num_subwindows - 1;
9347a63dc53SLuboš Luňák     if (m_curr_active_window_idx != UINT32_MAX) {
9357caa17caSRaphael Isemann       m_prev_active_window_idx = m_curr_active_window_idx;
9367a63dc53SLuboš Luňák       start_idx = m_curr_active_window_idx - 1;
9377a63dc53SLuboš Luňák     }
9387a63dc53SLuboš Luňák     for (int idx = start_idx; idx >= 0; --idx) {
9397a63dc53SLuboš Luňák       if (m_subwindows[idx]->GetCanBeActive()) {
9407a63dc53SLuboš Luňák         m_curr_active_window_idx = idx;
9417a63dc53SLuboš Luňák         return;
9427a63dc53SLuboš Luňák       }
9437a63dc53SLuboš Luňák     }
9447a63dc53SLuboš Luňák     for (int idx = num_subwindows - 1; idx > start_idx; --idx) {
9457caa17caSRaphael Isemann       if (m_subwindows[idx]->GetCanBeActive()) {
9467caa17caSRaphael Isemann         m_curr_active_window_idx = idx;
9477caa17caSRaphael Isemann         break;
9487caa17caSRaphael Isemann       }
9497caa17caSRaphael Isemann     }
9507caa17caSRaphael Isemann   }
9517caa17caSRaphael Isemann 
9527caa17caSRaphael Isemann   const char *GetName() const { return m_name.c_str(); }
9537caa17caSRaphael Isemann 
9547caa17caSRaphael Isemann protected:
9557caa17caSRaphael Isemann   std::string m_name;
9567caa17caSRaphael Isemann   PANEL *m_panel;
9577caa17caSRaphael Isemann   Window *m_parent;
9587caa17caSRaphael Isemann   Windows m_subwindows;
9597caa17caSRaphael Isemann   WindowDelegateSP m_delegate_sp;
9607caa17caSRaphael Isemann   uint32_t m_curr_active_window_idx;
9617caa17caSRaphael Isemann   uint32_t m_prev_active_window_idx;
9627caa17caSRaphael Isemann   bool m_delete;
9637caa17caSRaphael Isemann   bool m_needs_update;
9647caa17caSRaphael Isemann   bool m_can_activate;
9657caa17caSRaphael Isemann   bool m_is_subwin;
9667caa17caSRaphael Isemann 
9677caa17caSRaphael Isemann private:
968eaebcbc6SKonrad Kleine   Window(const Window &) = delete;
969eaebcbc6SKonrad Kleine   const Window &operator=(const Window &) = delete;
9707caa17caSRaphael Isemann };
9717caa17caSRaphael Isemann 
97229cc50e1SOmar Emara class DerivedWindow : public Surface {
97329cc50e1SOmar Emara public:
97429cc50e1SOmar Emara   DerivedWindow(Window &window, Rect bounds) {
97529cc50e1SOmar Emara     m_window = ::derwin(window.get(), bounds.size.height, bounds.size.width,
97629cc50e1SOmar Emara                         bounds.origin.y, bounds.origin.x);
97729cc50e1SOmar Emara   }
97829cc50e1SOmar Emara   DerivedWindow(DerivedWindow &derived_window, Rect bounds) {
97929cc50e1SOmar Emara     m_window = ::derwin(derived_window.get(), bounds.size.height,
98029cc50e1SOmar Emara                         bounds.size.width, bounds.origin.y, bounds.origin.x);
98129cc50e1SOmar Emara   }
98229cc50e1SOmar Emara 
98329cc50e1SOmar Emara   ~DerivedWindow() { ::delwin(m_window); }
98429cc50e1SOmar Emara };
98529cc50e1SOmar Emara 
98629cc50e1SOmar Emara /////////
98729cc50e1SOmar Emara // Forms
98829cc50e1SOmar Emara /////////
98929cc50e1SOmar Emara 
99029cc50e1SOmar Emara // A scroll context defines a vertical region that needs to be visible in a
99129cc50e1SOmar Emara // scrolling area. The region is defined by the index of the start and end lines
99229cc50e1SOmar Emara // of the region. The start and end lines may be equal, in which case, the
99329cc50e1SOmar Emara // region is a single line.
99429cc50e1SOmar Emara struct ScrollContext {
99529cc50e1SOmar Emara   int start;
99629cc50e1SOmar Emara   int end;
99729cc50e1SOmar Emara 
99829cc50e1SOmar Emara   ScrollContext(int line) : start(line), end(line) {}
99929cc50e1SOmar Emara   ScrollContext(int _start, int _end) : start(_start), end(_end) {}
100029cc50e1SOmar Emara 
100129cc50e1SOmar Emara   void Offset(int offset) {
100229cc50e1SOmar Emara     start += offset;
100329cc50e1SOmar Emara     end += offset;
100429cc50e1SOmar Emara   }
100529cc50e1SOmar Emara };
100629cc50e1SOmar Emara 
100729cc50e1SOmar Emara class FieldDelegate {
100829cc50e1SOmar Emara public:
100929cc50e1SOmar Emara   virtual ~FieldDelegate() = default;
101029cc50e1SOmar Emara 
101129cc50e1SOmar Emara   // Returns the number of lines needed to draw the field. The draw method will
101229cc50e1SOmar Emara   // be given a surface that have exactly this number of lines.
101329cc50e1SOmar Emara   virtual int FieldDelegateGetHeight() = 0;
101429cc50e1SOmar Emara 
101529cc50e1SOmar Emara   // Returns the scroll context in the local coordinates of the field. By
101629cc50e1SOmar Emara   // default, the scroll context spans the whole field. Bigger fields with
101729cc50e1SOmar Emara   // internal navigation should override this method to provide a finer context.
101829cc50e1SOmar Emara   // Typical override methods would first get the scroll context of the internal
101929cc50e1SOmar Emara   // element then add the offset of the element in the field.
102029cc50e1SOmar Emara   virtual ScrollContext FieldDelegateGetScrollContext() {
102129cc50e1SOmar Emara     return ScrollContext(0, FieldDelegateGetHeight() - 1);
102229cc50e1SOmar Emara   }
102329cc50e1SOmar Emara 
102429cc50e1SOmar Emara   // Draw the field in the given subpad surface. The surface have a height that
102529cc50e1SOmar Emara   // is equal to the height returned by FieldDelegateGetHeight(). If the field
102629cc50e1SOmar Emara   // is selected in the form window, then is_selected will be true.
102729cc50e1SOmar Emara   virtual void FieldDelegateDraw(SubPad &surface, bool is_selected) = 0;
102829cc50e1SOmar Emara 
102929cc50e1SOmar Emara   // Handle the key that wasn't handled by the form window or a container field.
103029cc50e1SOmar Emara   virtual HandleCharResult FieldDelegateHandleChar(int key) {
103129cc50e1SOmar Emara     return eKeyNotHandled;
103229cc50e1SOmar Emara   }
103329cc50e1SOmar Emara 
103429cc50e1SOmar Emara   // This is executed once the user exists the field, that is, once the user
103529cc50e1SOmar Emara   // navigates to the next or the previous field. This is particularly useful to
103629cc50e1SOmar Emara   // do in-field validation and error setting. Fields with internal navigation
103729cc50e1SOmar Emara   // should call this method on their fields.
103829cc50e1SOmar Emara   virtual void FieldDelegateExitCallback() { return; }
103929cc50e1SOmar Emara 
104029cc50e1SOmar Emara   // Fields may have internal navigation, for instance, a List Field have
104129cc50e1SOmar Emara   // multiple internal elements, which needs to be navigated. To allow for this
104229cc50e1SOmar Emara   // mechanism, the window shouldn't handle the navigation keys all the time,
104329cc50e1SOmar Emara   // and instead call the key handing method of the selected field. It should
104429cc50e1SOmar Emara   // only handle the navigation keys when the field contains a single element or
104529cc50e1SOmar Emara   // have the last or first element selected depending on if the user is
104629cc50e1SOmar Emara   // navigating forward or backward. Additionally, once a field is selected in
104729cc50e1SOmar Emara   // the forward or backward direction, its first or last internal element
104829cc50e1SOmar Emara   // should be selected. The following methods implements those mechanisms.
104929cc50e1SOmar Emara 
105029cc50e1SOmar Emara   // Returns true if the first element in the field is selected or if the field
105129cc50e1SOmar Emara   // contains a single element.
105229cc50e1SOmar Emara   virtual bool FieldDelegateOnFirstOrOnlyElement() { return true; }
105329cc50e1SOmar Emara 
105429cc50e1SOmar Emara   // Returns true if the last element in the field is selected or if the field
105529cc50e1SOmar Emara   // contains a single element.
105629cc50e1SOmar Emara   virtual bool FieldDelegateOnLastOrOnlyElement() { return true; }
105729cc50e1SOmar Emara 
105829cc50e1SOmar Emara   // Select the first element in the field if multiple elements exists.
105929cc50e1SOmar Emara   virtual void FieldDelegateSelectFirstElement() { return; }
106029cc50e1SOmar Emara 
106129cc50e1SOmar Emara   // Select the last element in the field if multiple elements exists.
106229cc50e1SOmar Emara   virtual void FieldDelegateSelectLastElement() { return; }
1063*0321dbc8SOmar Emara 
1064*0321dbc8SOmar Emara   bool FieldDelegateIsVisible() { return m_is_visible; }
1065*0321dbc8SOmar Emara 
1066*0321dbc8SOmar Emara   void FieldDelegateHide() { m_is_visible = false; }
1067*0321dbc8SOmar Emara 
1068*0321dbc8SOmar Emara   void FieldDelegateShow() { m_is_visible = true; }
1069*0321dbc8SOmar Emara 
1070*0321dbc8SOmar Emara protected:
1071*0321dbc8SOmar Emara   bool m_is_visible = true;
107229cc50e1SOmar Emara };
107329cc50e1SOmar Emara 
1074*0321dbc8SOmar Emara typedef std::unique_ptr<FieldDelegate> FieldDelegateUP;
107529cc50e1SOmar Emara 
107629cc50e1SOmar Emara class TextFieldDelegate : public FieldDelegate {
107729cc50e1SOmar Emara public:
107829cc50e1SOmar Emara   TextFieldDelegate(const char *label, const char *content)
107929cc50e1SOmar Emara       : m_label(label), m_cursor_position(0), m_first_visibile_char(0) {
108029cc50e1SOmar Emara     if (content)
108129cc50e1SOmar Emara       m_content = content;
108229cc50e1SOmar Emara   }
108329cc50e1SOmar Emara 
108429cc50e1SOmar Emara   // Text fields are drawn as titled boxes of a single line, with a possible
108529cc50e1SOmar Emara   // error messages at the end.
108629cc50e1SOmar Emara   //
108729cc50e1SOmar Emara   // __[Label]___________
108829cc50e1SOmar Emara   // |                  |
108929cc50e1SOmar Emara   // |__________________|
109029cc50e1SOmar Emara   // - Error message if it exists.
109129cc50e1SOmar Emara 
109229cc50e1SOmar Emara   // The text field has a height of 3 lines. 2 lines for borders and 1 line for
109329cc50e1SOmar Emara   // the content.
109429cc50e1SOmar Emara   int GetFieldHeight() { return 3; }
109529cc50e1SOmar Emara 
109629cc50e1SOmar Emara   // The text field has a full height of 3 or 4 lines. 3 lines for the actual
109729cc50e1SOmar Emara   // field and an optional line for an error if it exists.
109829cc50e1SOmar Emara   int FieldDelegateGetHeight() override {
109929cc50e1SOmar Emara     int height = GetFieldHeight();
110029cc50e1SOmar Emara     if (HasError())
110129cc50e1SOmar Emara       height++;
110229cc50e1SOmar Emara     return height;
110329cc50e1SOmar Emara   }
110429cc50e1SOmar Emara 
110529cc50e1SOmar Emara   // Get the cursor X position in the surface coordinate.
110629cc50e1SOmar Emara   int GetCursorXPosition() { return m_cursor_position - m_first_visibile_char; }
110729cc50e1SOmar Emara 
110829cc50e1SOmar Emara   int GetContentLength() { return m_content.length(); }
110929cc50e1SOmar Emara 
111029cc50e1SOmar Emara   void DrawContent(SubPad &surface, bool is_selected) {
111129cc50e1SOmar Emara     surface.MoveCursor(0, 0);
111229cc50e1SOmar Emara     const char *text = m_content.c_str() + m_first_visibile_char;
111329cc50e1SOmar Emara     surface.PutCString(text, surface.GetWidth());
111429cc50e1SOmar Emara     m_last_drawn_content_width = surface.GetWidth();
111529cc50e1SOmar Emara 
111629cc50e1SOmar Emara     // Highlight the cursor.
111729cc50e1SOmar Emara     surface.MoveCursor(GetCursorXPosition(), 0);
111829cc50e1SOmar Emara     if (is_selected)
111929cc50e1SOmar Emara       surface.AttributeOn(A_REVERSE);
112029cc50e1SOmar Emara     if (m_cursor_position == GetContentLength())
112129cc50e1SOmar Emara       // Cursor is past the last character. Highlight an empty space.
112229cc50e1SOmar Emara       surface.PutChar(' ');
112329cc50e1SOmar Emara     else
112429cc50e1SOmar Emara       surface.PutChar(m_content[m_cursor_position]);
112529cc50e1SOmar Emara     if (is_selected)
112629cc50e1SOmar Emara       surface.AttributeOff(A_REVERSE);
112729cc50e1SOmar Emara   }
112829cc50e1SOmar Emara 
112929cc50e1SOmar Emara   void DrawField(SubPad &surface, bool is_selected) {
113029cc50e1SOmar Emara     surface.TitledBox(m_label.c_str());
113129cc50e1SOmar Emara 
113229cc50e1SOmar Emara     Rect content_bounds = surface.GetFrame();
113329cc50e1SOmar Emara     content_bounds.Inset(1, 1);
113429cc50e1SOmar Emara     SubPad content_surface = SubPad(surface, content_bounds);
113529cc50e1SOmar Emara 
113629cc50e1SOmar Emara     DrawContent(content_surface, is_selected);
113729cc50e1SOmar Emara   }
113829cc50e1SOmar Emara 
113929cc50e1SOmar Emara   void DrawError(SubPad &surface) {
114029cc50e1SOmar Emara     if (!HasError())
114129cc50e1SOmar Emara       return;
114229cc50e1SOmar Emara     surface.MoveCursor(0, 0);
114329cc50e1SOmar Emara     surface.AttributeOn(COLOR_PAIR(RedOnBlack));
114429cc50e1SOmar Emara     surface.PutChar(ACS_DIAMOND);
114529cc50e1SOmar Emara     surface.PutChar(' ');
114629cc50e1SOmar Emara     surface.PutCStringTruncated(1, GetError().c_str());
114729cc50e1SOmar Emara     surface.AttributeOff(COLOR_PAIR(RedOnBlack));
114829cc50e1SOmar Emara   }
114929cc50e1SOmar Emara 
115029cc50e1SOmar Emara   void FieldDelegateDraw(SubPad &surface, bool is_selected) override {
115129cc50e1SOmar Emara     Rect frame = surface.GetFrame();
115229cc50e1SOmar Emara     Rect field_bounds, error_bounds;
115329cc50e1SOmar Emara     frame.HorizontalSplit(GetFieldHeight(), field_bounds, error_bounds);
115429cc50e1SOmar Emara     SubPad field_surface = SubPad(surface, field_bounds);
115529cc50e1SOmar Emara     SubPad error_surface = SubPad(surface, error_bounds);
115629cc50e1SOmar Emara 
115729cc50e1SOmar Emara     DrawField(field_surface, is_selected);
115829cc50e1SOmar Emara     DrawError(error_surface);
115929cc50e1SOmar Emara   }
116029cc50e1SOmar Emara 
116129cc50e1SOmar Emara   // The cursor is allowed to move one character past the string.
116229cc50e1SOmar Emara   // m_cursor_position is in range [0, GetContentLength()].
116329cc50e1SOmar Emara   void MoveCursorRight() {
116429cc50e1SOmar Emara     if (m_cursor_position < GetContentLength())
116529cc50e1SOmar Emara       m_cursor_position++;
116629cc50e1SOmar Emara   }
116729cc50e1SOmar Emara 
116829cc50e1SOmar Emara   void MoveCursorLeft() {
116929cc50e1SOmar Emara     if (m_cursor_position > 0)
117029cc50e1SOmar Emara       m_cursor_position--;
117129cc50e1SOmar Emara   }
117229cc50e1SOmar Emara 
117329cc50e1SOmar Emara   // If the cursor moved past the last visible character, scroll right by one
117429cc50e1SOmar Emara   // character.
117529cc50e1SOmar Emara   void ScrollRightIfNeeded() {
117629cc50e1SOmar Emara     if (m_cursor_position - m_first_visibile_char == m_last_drawn_content_width)
117729cc50e1SOmar Emara       m_first_visibile_char++;
117829cc50e1SOmar Emara   }
117929cc50e1SOmar Emara 
118029cc50e1SOmar Emara   void ScrollLeft() {
118129cc50e1SOmar Emara     if (m_first_visibile_char > 0)
118229cc50e1SOmar Emara       m_first_visibile_char--;
118329cc50e1SOmar Emara   }
118429cc50e1SOmar Emara 
118529cc50e1SOmar Emara   // If the cursor moved past the first visible character, scroll left by one
118629cc50e1SOmar Emara   // character.
118729cc50e1SOmar Emara   void ScrollLeftIfNeeded() {
118829cc50e1SOmar Emara     if (m_cursor_position < m_first_visibile_char)
118929cc50e1SOmar Emara       m_first_visibile_char--;
119029cc50e1SOmar Emara   }
119129cc50e1SOmar Emara 
119229cc50e1SOmar Emara   // Insert a character at the current cursor position, advance the cursor
119329cc50e1SOmar Emara   // position, and make sure to scroll right if needed.
119429cc50e1SOmar Emara   void InsertChar(char character) {
119529cc50e1SOmar Emara     m_content.insert(m_cursor_position, 1, character);
119629cc50e1SOmar Emara     m_cursor_position++;
119729cc50e1SOmar Emara     ScrollRightIfNeeded();
119829cc50e1SOmar Emara   }
119929cc50e1SOmar Emara 
120029cc50e1SOmar Emara   // Remove the character before the cursor position, retreat the cursor
120129cc50e1SOmar Emara   // position, and make sure to scroll left if needed.
120229cc50e1SOmar Emara   void RemoveChar() {
120329cc50e1SOmar Emara     if (m_cursor_position == 0)
120429cc50e1SOmar Emara       return;
120529cc50e1SOmar Emara 
120629cc50e1SOmar Emara     m_content.erase(m_cursor_position - 1, 1);
120729cc50e1SOmar Emara     m_cursor_position--;
120829cc50e1SOmar Emara     ScrollLeft();
120929cc50e1SOmar Emara   }
121029cc50e1SOmar Emara 
121129cc50e1SOmar Emara   // True if the key represents a char that can be inserted in the field
121229cc50e1SOmar Emara   // content, false otherwise.
121329cc50e1SOmar Emara   virtual bool IsAcceptableChar(int key) { return isprint(key); }
121429cc50e1SOmar Emara 
121529cc50e1SOmar Emara   HandleCharResult FieldDelegateHandleChar(int key) override {
121629cc50e1SOmar Emara     if (IsAcceptableChar(key)) {
121729cc50e1SOmar Emara       ClearError();
121829cc50e1SOmar Emara       InsertChar((char)key);
121929cc50e1SOmar Emara       return eKeyHandled;
122029cc50e1SOmar Emara     }
122129cc50e1SOmar Emara 
122229cc50e1SOmar Emara     switch (key) {
122329cc50e1SOmar Emara     case KEY_RIGHT:
122429cc50e1SOmar Emara       MoveCursorRight();
122529cc50e1SOmar Emara       ScrollRightIfNeeded();
122629cc50e1SOmar Emara       return eKeyHandled;
122729cc50e1SOmar Emara     case KEY_LEFT:
122829cc50e1SOmar Emara       MoveCursorLeft();
122929cc50e1SOmar Emara       ScrollLeftIfNeeded();
123029cc50e1SOmar Emara       return eKeyHandled;
123129cc50e1SOmar Emara     case KEY_BACKSPACE:
123229cc50e1SOmar Emara       ClearError();
123329cc50e1SOmar Emara       RemoveChar();
123429cc50e1SOmar Emara       return eKeyHandled;
123529cc50e1SOmar Emara     default:
123629cc50e1SOmar Emara       break;
123729cc50e1SOmar Emara     }
123829cc50e1SOmar Emara     return eKeyNotHandled;
123929cc50e1SOmar Emara   }
124029cc50e1SOmar Emara 
124129cc50e1SOmar Emara   bool HasError() { return !m_error.empty(); }
124229cc50e1SOmar Emara 
124329cc50e1SOmar Emara   void ClearError() { m_error.clear(); }
124429cc50e1SOmar Emara 
124529cc50e1SOmar Emara   const std::string &GetError() { return m_error; }
124629cc50e1SOmar Emara 
124729cc50e1SOmar Emara   void SetError(const char *error) { m_error = error; }
124829cc50e1SOmar Emara 
124929cc50e1SOmar Emara   const std::string &GetText() { return m_content; }
125029cc50e1SOmar Emara 
125129cc50e1SOmar Emara protected:
125229cc50e1SOmar Emara   std::string m_label;
125329cc50e1SOmar Emara   // The position of the top left corner character of the border.
125429cc50e1SOmar Emara   std::string m_content;
125529cc50e1SOmar Emara   // The cursor position in the content string itself. Can be in the range
125629cc50e1SOmar Emara   // [0, GetContentLength()].
125729cc50e1SOmar Emara   int m_cursor_position;
125829cc50e1SOmar Emara   // The index of the first visible character in the content.
125929cc50e1SOmar Emara   int m_first_visibile_char;
126029cc50e1SOmar Emara   // The width of the fields content that was last drawn. Width can change, so
126129cc50e1SOmar Emara   // this is used to determine if scrolling is needed dynamically.
126229cc50e1SOmar Emara   int m_last_drawn_content_width;
126329cc50e1SOmar Emara   // Optional error message. If empty, field is considered to have no error.
126429cc50e1SOmar Emara   std::string m_error;
126529cc50e1SOmar Emara };
126629cc50e1SOmar Emara 
126729cc50e1SOmar Emara class IntegerFieldDelegate : public TextFieldDelegate {
126829cc50e1SOmar Emara public:
126929cc50e1SOmar Emara   IntegerFieldDelegate(const char *label, int content)
127029cc50e1SOmar Emara       : TextFieldDelegate(label, std::to_string(content).c_str()) {}
127129cc50e1SOmar Emara 
127229cc50e1SOmar Emara   // Only accept digits.
127329cc50e1SOmar Emara   bool IsAcceptableChar(int key) override { return isdigit(key); }
127429cc50e1SOmar Emara 
127529cc50e1SOmar Emara   // Returns the integer content of the field.
127629cc50e1SOmar Emara   int GetInteger() { return std::stoi(m_content); }
127729cc50e1SOmar Emara };
127829cc50e1SOmar Emara 
127929cc50e1SOmar Emara class FileFieldDelegate : public TextFieldDelegate {
128029cc50e1SOmar Emara public:
128129cc50e1SOmar Emara   FileFieldDelegate(const char *label, const char *content,
128229cc50e1SOmar Emara                     bool need_to_exist = true)
128329cc50e1SOmar Emara       : TextFieldDelegate(label, content), m_need_to_exist(need_to_exist) {}
128429cc50e1SOmar Emara 
128529cc50e1SOmar Emara   // Set appropriate error messages if the file doesn't exists or is, in fact, a
128629cc50e1SOmar Emara   // directory.
128729cc50e1SOmar Emara   void FieldDelegateExitCallback() override {
128829cc50e1SOmar Emara     FileSpec file(GetPath());
128929cc50e1SOmar Emara     if (m_need_to_exist && !FileSystem::Instance().Exists(file)) {
129029cc50e1SOmar Emara       SetError("File doesn't exist!");
129129cc50e1SOmar Emara       return;
129229cc50e1SOmar Emara     }
129329cc50e1SOmar Emara     if (FileSystem::Instance().IsDirectory(file)) {
129429cc50e1SOmar Emara       SetError("Not a file!");
129529cc50e1SOmar Emara       return;
129629cc50e1SOmar Emara     }
129729cc50e1SOmar Emara   }
129829cc50e1SOmar Emara 
129929cc50e1SOmar Emara   // Returns the path of the file.
130029cc50e1SOmar Emara   const std::string &GetPath() { return m_content; }
130129cc50e1SOmar Emara 
130229cc50e1SOmar Emara protected:
130329cc50e1SOmar Emara   bool m_need_to_exist;
130429cc50e1SOmar Emara };
130529cc50e1SOmar Emara 
130629cc50e1SOmar Emara class DirectoryFieldDelegate : public TextFieldDelegate {
130729cc50e1SOmar Emara public:
130829cc50e1SOmar Emara   DirectoryFieldDelegate(const char *label, const char *content,
130929cc50e1SOmar Emara                          bool need_to_exist = true)
131029cc50e1SOmar Emara       : TextFieldDelegate(label, content), m_need_to_exist(need_to_exist) {}
131129cc50e1SOmar Emara 
131229cc50e1SOmar Emara   // Set appropriate error messages if the directory doesn't exists or is, in
131329cc50e1SOmar Emara   // fact, a file.
131429cc50e1SOmar Emara   void FieldDelegateExitCallback() override {
131529cc50e1SOmar Emara     FileSpec file(GetPath());
131629cc50e1SOmar Emara     if (m_need_to_exist && !FileSystem::Instance().Exists(file)) {
131729cc50e1SOmar Emara       SetError("Directory doesn't exist!");
131829cc50e1SOmar Emara       return;
131929cc50e1SOmar Emara     }
132029cc50e1SOmar Emara     if (!FileSystem::Instance().IsDirectory(file)) {
132129cc50e1SOmar Emara       SetError("Not a directory!");
132229cc50e1SOmar Emara       return;
132329cc50e1SOmar Emara     }
132429cc50e1SOmar Emara   }
132529cc50e1SOmar Emara 
132629cc50e1SOmar Emara   // Returns the path of the file.
132729cc50e1SOmar Emara   const std::string &GetPath() { return m_content; }
132829cc50e1SOmar Emara 
132929cc50e1SOmar Emara protected:
133029cc50e1SOmar Emara   bool m_need_to_exist;
133129cc50e1SOmar Emara };
133229cc50e1SOmar Emara 
133329cc50e1SOmar Emara class BooleanFieldDelegate : public FieldDelegate {
133429cc50e1SOmar Emara public:
133529cc50e1SOmar Emara   BooleanFieldDelegate(const char *label, bool content)
133629cc50e1SOmar Emara       : m_label(label), m_content(content) {}
133729cc50e1SOmar Emara 
133829cc50e1SOmar Emara   // Boolean fields are drawn as checkboxes.
133929cc50e1SOmar Emara   //
134029cc50e1SOmar Emara   // [X] Label  or [ ] Label
134129cc50e1SOmar Emara 
134229cc50e1SOmar Emara   // Boolean fields are have a single line.
134329cc50e1SOmar Emara   int FieldDelegateGetHeight() override { return 1; }
134429cc50e1SOmar Emara 
134529cc50e1SOmar Emara   void FieldDelegateDraw(SubPad &surface, bool is_selected) override {
134629cc50e1SOmar Emara     surface.MoveCursor(0, 0);
134729cc50e1SOmar Emara     surface.PutChar('[');
134829cc50e1SOmar Emara     if (is_selected)
134929cc50e1SOmar Emara       surface.AttributeOn(A_REVERSE);
135029cc50e1SOmar Emara     surface.PutChar(m_content ? ACS_DIAMOND : ' ');
135129cc50e1SOmar Emara     if (is_selected)
135229cc50e1SOmar Emara       surface.AttributeOff(A_REVERSE);
135329cc50e1SOmar Emara     surface.PutChar(']');
135429cc50e1SOmar Emara     surface.PutChar(' ');
135529cc50e1SOmar Emara     surface.PutCString(m_label.c_str());
135629cc50e1SOmar Emara   }
135729cc50e1SOmar Emara 
135829cc50e1SOmar Emara   void ToggleContent() { m_content = !m_content; }
135929cc50e1SOmar Emara 
136029cc50e1SOmar Emara   void SetContentToTrue() { m_content = true; }
136129cc50e1SOmar Emara 
136229cc50e1SOmar Emara   void SetContentToFalse() { m_content = false; }
136329cc50e1SOmar Emara 
136429cc50e1SOmar Emara   HandleCharResult FieldDelegateHandleChar(int key) override {
136529cc50e1SOmar Emara     switch (key) {
136629cc50e1SOmar Emara     case 't':
136729cc50e1SOmar Emara     case '1':
136829cc50e1SOmar Emara       SetContentToTrue();
136929cc50e1SOmar Emara       return eKeyHandled;
137029cc50e1SOmar Emara     case 'f':
137129cc50e1SOmar Emara     case '0':
137229cc50e1SOmar Emara       SetContentToFalse();
137329cc50e1SOmar Emara       return eKeyHandled;
137429cc50e1SOmar Emara     case ' ':
137529cc50e1SOmar Emara     case '\r':
137629cc50e1SOmar Emara     case '\n':
137729cc50e1SOmar Emara     case KEY_ENTER:
137829cc50e1SOmar Emara       ToggleContent();
137929cc50e1SOmar Emara       return eKeyHandled;
138029cc50e1SOmar Emara     default:
138129cc50e1SOmar Emara       break;
138229cc50e1SOmar Emara     }
138329cc50e1SOmar Emara     return eKeyNotHandled;
138429cc50e1SOmar Emara   }
138529cc50e1SOmar Emara 
138629cc50e1SOmar Emara   // Returns the boolean content of the field.
138729cc50e1SOmar Emara   bool GetBoolean() { return m_content; }
138829cc50e1SOmar Emara 
138929cc50e1SOmar Emara protected:
139029cc50e1SOmar Emara   std::string m_label;
139129cc50e1SOmar Emara   bool m_content;
139229cc50e1SOmar Emara };
139329cc50e1SOmar Emara 
139429cc50e1SOmar Emara class ChoicesFieldDelegate : public FieldDelegate {
139529cc50e1SOmar Emara public:
139629cc50e1SOmar Emara   ChoicesFieldDelegate(const char *label, int number_of_visible_choices,
139729cc50e1SOmar Emara                        std::vector<std::string> choices)
139829cc50e1SOmar Emara       : m_label(label), m_number_of_visible_choices(number_of_visible_choices),
139929cc50e1SOmar Emara         m_choices(choices), m_choice(0), m_first_visibile_choice(0) {}
140029cc50e1SOmar Emara 
140129cc50e1SOmar Emara   // Choices fields are drawn as titles boxses of a number of visible choices.
140229cc50e1SOmar Emara   // The rest of the choices become visible as the user scroll. The selected
140329cc50e1SOmar Emara   // choice is denoted by a diamond as the first character.
140429cc50e1SOmar Emara   //
140529cc50e1SOmar Emara   // __[Label]___________
140629cc50e1SOmar Emara   // |-Choice 1         |
140729cc50e1SOmar Emara   // | Choice 2         |
140829cc50e1SOmar Emara   // | Choice 3         |
140929cc50e1SOmar Emara   // |__________________|
141029cc50e1SOmar Emara 
141129cc50e1SOmar Emara   // Choices field have two border characters plus the number of visible
141229cc50e1SOmar Emara   // choices.
141329cc50e1SOmar Emara   int FieldDelegateGetHeight() override {
141429cc50e1SOmar Emara     return m_number_of_visible_choices + 2;
141529cc50e1SOmar Emara   }
141629cc50e1SOmar Emara 
141729cc50e1SOmar Emara   int GetNumberOfChoices() { return m_choices.size(); }
141829cc50e1SOmar Emara 
141929cc50e1SOmar Emara   // Get the index of the last visible choice.
142029cc50e1SOmar Emara   int GetLastVisibleChoice() {
142129cc50e1SOmar Emara     int index = m_first_visibile_choice + m_number_of_visible_choices;
142229cc50e1SOmar Emara     return std::min(index, GetNumberOfChoices()) - 1;
142329cc50e1SOmar Emara   }
142429cc50e1SOmar Emara 
142529cc50e1SOmar Emara   void DrawContent(SubPad &surface, bool is_selected) {
142629cc50e1SOmar Emara     int choices_to_draw = GetLastVisibleChoice() - m_first_visibile_choice + 1;
142729cc50e1SOmar Emara     for (int i = 0; i < choices_to_draw; i++) {
142829cc50e1SOmar Emara       surface.MoveCursor(0, i);
142929cc50e1SOmar Emara       int current_choice = m_first_visibile_choice + i;
143029cc50e1SOmar Emara       const char *text = m_choices[current_choice].c_str();
143129cc50e1SOmar Emara       bool highlight = is_selected && current_choice == m_choice;
143229cc50e1SOmar Emara       if (highlight)
143329cc50e1SOmar Emara         surface.AttributeOn(A_REVERSE);
143429cc50e1SOmar Emara       surface.PutChar(current_choice == m_choice ? ACS_DIAMOND : ' ');
143529cc50e1SOmar Emara       surface.PutCString(text);
143629cc50e1SOmar Emara       if (highlight)
143729cc50e1SOmar Emara         surface.AttributeOff(A_REVERSE);
143829cc50e1SOmar Emara     }
143929cc50e1SOmar Emara   }
144029cc50e1SOmar Emara 
144129cc50e1SOmar Emara   void FieldDelegateDraw(SubPad &surface, bool is_selected) override {
144229cc50e1SOmar Emara     surface.TitledBox(m_label.c_str());
144329cc50e1SOmar Emara 
144429cc50e1SOmar Emara     Rect content_bounds = surface.GetFrame();
144529cc50e1SOmar Emara     content_bounds.Inset(1, 1);
144629cc50e1SOmar Emara     SubPad content_surface = SubPad(surface, content_bounds);
144729cc50e1SOmar Emara 
144829cc50e1SOmar Emara     DrawContent(content_surface, is_selected);
144929cc50e1SOmar Emara   }
145029cc50e1SOmar Emara 
145129cc50e1SOmar Emara   void SelectPrevious() {
145229cc50e1SOmar Emara     if (m_choice > 0)
145329cc50e1SOmar Emara       m_choice--;
145429cc50e1SOmar Emara   }
145529cc50e1SOmar Emara 
145629cc50e1SOmar Emara   void SelectNext() {
145729cc50e1SOmar Emara     if (m_choice < GetNumberOfChoices() - 1)
145829cc50e1SOmar Emara       m_choice++;
145929cc50e1SOmar Emara   }
146029cc50e1SOmar Emara 
146129cc50e1SOmar Emara   // If the cursor moved past the first visible choice, scroll up by one
146229cc50e1SOmar Emara   // choice.
146329cc50e1SOmar Emara   void ScrollUpIfNeeded() {
146429cc50e1SOmar Emara     if (m_choice < m_first_visibile_choice)
146529cc50e1SOmar Emara       m_first_visibile_choice--;
146629cc50e1SOmar Emara   }
146729cc50e1SOmar Emara 
146829cc50e1SOmar Emara   // If the cursor moved past the last visible choice, scroll down by one
146929cc50e1SOmar Emara   // choice.
147029cc50e1SOmar Emara   void ScrollDownIfNeeded() {
147129cc50e1SOmar Emara     if (m_choice > GetLastVisibleChoice())
147229cc50e1SOmar Emara       m_first_visibile_choice++;
147329cc50e1SOmar Emara   }
147429cc50e1SOmar Emara 
147529cc50e1SOmar Emara   HandleCharResult FieldDelegateHandleChar(int key) override {
147629cc50e1SOmar Emara     switch (key) {
147729cc50e1SOmar Emara     case KEY_UP:
147829cc50e1SOmar Emara       SelectPrevious();
147929cc50e1SOmar Emara       ScrollUpIfNeeded();
148029cc50e1SOmar Emara       return eKeyHandled;
148129cc50e1SOmar Emara     case KEY_DOWN:
148229cc50e1SOmar Emara       SelectNext();
148329cc50e1SOmar Emara       ScrollDownIfNeeded();
148429cc50e1SOmar Emara       return eKeyHandled;
148529cc50e1SOmar Emara     default:
148629cc50e1SOmar Emara       break;
148729cc50e1SOmar Emara     }
148829cc50e1SOmar Emara     return eKeyNotHandled;
148929cc50e1SOmar Emara   }
149029cc50e1SOmar Emara 
149129cc50e1SOmar Emara   // Returns the content of the choice as a string.
149229cc50e1SOmar Emara   std::string GetChoiceContent() { return m_choices[m_choice]; }
149329cc50e1SOmar Emara 
149429cc50e1SOmar Emara   // Returns the index of the choice.
149529cc50e1SOmar Emara   int GetChoice() { return m_choice; }
149629cc50e1SOmar Emara 
149729cc50e1SOmar Emara protected:
149829cc50e1SOmar Emara   std::string m_label;
149929cc50e1SOmar Emara   int m_number_of_visible_choices;
150029cc50e1SOmar Emara   std::vector<std::string> m_choices;
150129cc50e1SOmar Emara   // The index of the selected choice.
150229cc50e1SOmar Emara   int m_choice;
150329cc50e1SOmar Emara   // The index of the first visible choice in the field.
150429cc50e1SOmar Emara   int m_first_visibile_choice;
150529cc50e1SOmar Emara };
150629cc50e1SOmar Emara 
150729cc50e1SOmar Emara template <class T> class ListFieldDelegate : public FieldDelegate {
150829cc50e1SOmar Emara public:
150929cc50e1SOmar Emara   ListFieldDelegate(const char *label, T default_field)
151029cc50e1SOmar Emara       : m_label(label), m_default_field(default_field), m_selection_index(0),
151129cc50e1SOmar Emara         m_selection_type(SelectionType::NewButton) {}
151229cc50e1SOmar Emara 
151329cc50e1SOmar Emara   // Signify which element is selected. If a field or a remove button is
151429cc50e1SOmar Emara   // selected, then m_selection_index signifies the particular field that
151529cc50e1SOmar Emara   // is selected or the field that the remove button belongs to.
151629cc50e1SOmar Emara   enum class SelectionType { Field, RemoveButton, NewButton };
151729cc50e1SOmar Emara 
151829cc50e1SOmar Emara   // A List field is drawn as a titled box of a number of other fields of the
151929cc50e1SOmar Emara   // same type. Each field has a Remove button next to it that removes the
152029cc50e1SOmar Emara   // corresponding field. Finally, the last line contains a New button to add a
152129cc50e1SOmar Emara   // new field.
152229cc50e1SOmar Emara   //
152329cc50e1SOmar Emara   // __[Label]___________
152429cc50e1SOmar Emara   // | Field 0 [Remove] |
152529cc50e1SOmar Emara   // | Field 1 [Remove] |
152629cc50e1SOmar Emara   // | Field 2 [Remove] |
152729cc50e1SOmar Emara   // |      [New]       |
152829cc50e1SOmar Emara   // |__________________|
152929cc50e1SOmar Emara 
153029cc50e1SOmar Emara   // List fields have two lines for border characters, 1 line for the New
153129cc50e1SOmar Emara   // button, and the total height of the available fields.
153229cc50e1SOmar Emara   int FieldDelegateGetHeight() override {
153329cc50e1SOmar Emara     // 2 border characters.
153429cc50e1SOmar Emara     int height = 2;
153529cc50e1SOmar Emara     // Total height of the fields.
153629cc50e1SOmar Emara     for (int i = 0; i < GetNumberOfFields(); i++) {
153729cc50e1SOmar Emara       height += m_fields[i].FieldDelegateGetHeight();
153829cc50e1SOmar Emara     }
153929cc50e1SOmar Emara     // A line for the New button.
154029cc50e1SOmar Emara     height++;
154129cc50e1SOmar Emara     return height;
154229cc50e1SOmar Emara   }
154329cc50e1SOmar Emara 
154429cc50e1SOmar Emara   ScrollContext FieldDelegateGetScrollContext() override {
154529cc50e1SOmar Emara     int height = FieldDelegateGetHeight();
154629cc50e1SOmar Emara     if (m_selection_type == SelectionType::NewButton)
154729cc50e1SOmar Emara       return ScrollContext(height - 2, height - 1);
154829cc50e1SOmar Emara 
154929cc50e1SOmar Emara     FieldDelegate &field = m_fields[m_selection_index];
155029cc50e1SOmar Emara     ScrollContext context = field.FieldDelegateGetScrollContext();
155129cc50e1SOmar Emara 
155229cc50e1SOmar Emara     // Start at 1 because of the top border.
155329cc50e1SOmar Emara     int offset = 1;
155429cc50e1SOmar Emara     for (int i = 0; i < m_selection_index; i++) {
155529cc50e1SOmar Emara       offset += m_fields[i].FieldDelegateGetHeight();
155629cc50e1SOmar Emara     }
155729cc50e1SOmar Emara     context.Offset(offset);
155829cc50e1SOmar Emara 
155929cc50e1SOmar Emara     // If the scroll context is touching the top border, include it in the
156029cc50e1SOmar Emara     // context to show the label.
156129cc50e1SOmar Emara     if (context.start == 1)
156229cc50e1SOmar Emara       context.start--;
156329cc50e1SOmar Emara 
156429cc50e1SOmar Emara     // If the scroll context is touching the new button, include it as well as
156529cc50e1SOmar Emara     // the bottom border in the context.
156629cc50e1SOmar Emara     if (context.end == height - 3)
156729cc50e1SOmar Emara       context.end += 2;
156829cc50e1SOmar Emara 
156929cc50e1SOmar Emara     return context;
157029cc50e1SOmar Emara   }
157129cc50e1SOmar Emara 
157229cc50e1SOmar Emara   void DrawRemoveButton(SubPad &surface, int highlight) {
157329cc50e1SOmar Emara     surface.MoveCursor(1, surface.GetHeight() / 2);
157429cc50e1SOmar Emara     if (highlight)
157529cc50e1SOmar Emara       surface.AttributeOn(A_REVERSE);
157629cc50e1SOmar Emara     surface.PutCString("[Remove]");
157729cc50e1SOmar Emara     if (highlight)
157829cc50e1SOmar Emara       surface.AttributeOff(A_REVERSE);
157929cc50e1SOmar Emara   }
158029cc50e1SOmar Emara 
158129cc50e1SOmar Emara   void DrawFields(SubPad &surface, bool is_selected) {
158229cc50e1SOmar Emara     int line = 0;
158329cc50e1SOmar Emara     int width = surface.GetWidth();
158429cc50e1SOmar Emara     for (int i = 0; i < GetNumberOfFields(); i++) {
158529cc50e1SOmar Emara       int height = m_fields[i].FieldDelegateGetHeight();
158629cc50e1SOmar Emara       Rect bounds = Rect(Point(0, line), Size(width, height));
158729cc50e1SOmar Emara       Rect field_bounds, remove_button_bounds;
158829cc50e1SOmar Emara       bounds.VerticalSplit(bounds.size.width - sizeof(" [Remove]"),
158929cc50e1SOmar Emara                            field_bounds, remove_button_bounds);
159029cc50e1SOmar Emara       SubPad field_surface = SubPad(surface, field_bounds);
159129cc50e1SOmar Emara       SubPad remove_button_surface = SubPad(surface, remove_button_bounds);
159229cc50e1SOmar Emara 
159329cc50e1SOmar Emara       bool is_element_selected = m_selection_index == i && is_selected;
159429cc50e1SOmar Emara       bool is_field_selected =
159529cc50e1SOmar Emara           is_element_selected && m_selection_type == SelectionType::Field;
159629cc50e1SOmar Emara       bool is_remove_button_selected =
159729cc50e1SOmar Emara           is_element_selected &&
159829cc50e1SOmar Emara           m_selection_type == SelectionType::RemoveButton;
159929cc50e1SOmar Emara       m_fields[i].FieldDelegateDraw(field_surface, is_field_selected);
160029cc50e1SOmar Emara       DrawRemoveButton(remove_button_surface, is_remove_button_selected);
160129cc50e1SOmar Emara 
160229cc50e1SOmar Emara       line += height;
160329cc50e1SOmar Emara     }
160429cc50e1SOmar Emara   }
160529cc50e1SOmar Emara 
160629cc50e1SOmar Emara   void DrawNewButton(SubPad &surface, bool is_selected) {
160729cc50e1SOmar Emara     const char *button_text = "[New]";
160829cc50e1SOmar Emara     int x = (surface.GetWidth() - sizeof(button_text) - 1) / 2;
160929cc50e1SOmar Emara     surface.MoveCursor(x, 0);
161029cc50e1SOmar Emara     bool highlight =
161129cc50e1SOmar Emara         is_selected && m_selection_type == SelectionType::NewButton;
161229cc50e1SOmar Emara     if (highlight)
161329cc50e1SOmar Emara       surface.AttributeOn(A_REVERSE);
161429cc50e1SOmar Emara     surface.PutCString(button_text);
161529cc50e1SOmar Emara     if (highlight)
161629cc50e1SOmar Emara       surface.AttributeOff(A_REVERSE);
161729cc50e1SOmar Emara   }
161829cc50e1SOmar Emara 
161929cc50e1SOmar Emara   void FieldDelegateDraw(SubPad &surface, bool is_selected) override {
162029cc50e1SOmar Emara     surface.TitledBox(m_label.c_str());
162129cc50e1SOmar Emara 
162229cc50e1SOmar Emara     Rect content_bounds = surface.GetFrame();
162329cc50e1SOmar Emara     content_bounds.Inset(1, 1);
162429cc50e1SOmar Emara     Rect fields_bounds, new_button_bounds;
162529cc50e1SOmar Emara     content_bounds.HorizontalSplit(content_bounds.size.height - 1,
162629cc50e1SOmar Emara                                    fields_bounds, new_button_bounds);
162729cc50e1SOmar Emara     SubPad fields_surface = SubPad(surface, fields_bounds);
162829cc50e1SOmar Emara     SubPad new_button_surface = SubPad(surface, new_button_bounds);
162929cc50e1SOmar Emara 
163029cc50e1SOmar Emara     DrawFields(fields_surface, is_selected);
163129cc50e1SOmar Emara     DrawNewButton(new_button_surface, is_selected);
163229cc50e1SOmar Emara   }
163329cc50e1SOmar Emara 
163429cc50e1SOmar Emara   void AddNewField() {
163529cc50e1SOmar Emara     m_fields.push_back(m_default_field);
163629cc50e1SOmar Emara     m_selection_index = GetNumberOfFields() - 1;
163729cc50e1SOmar Emara     m_selection_type = SelectionType::Field;
163829cc50e1SOmar Emara     FieldDelegate &field = m_fields[m_selection_index];
163929cc50e1SOmar Emara     field.FieldDelegateSelectFirstElement();
164029cc50e1SOmar Emara   }
164129cc50e1SOmar Emara 
164229cc50e1SOmar Emara   void RemoveField() {
164329cc50e1SOmar Emara     m_fields.erase(m_fields.begin() + m_selection_index);
164429cc50e1SOmar Emara     if (m_selection_index != 0)
164529cc50e1SOmar Emara       m_selection_index--;
164629cc50e1SOmar Emara 
164729cc50e1SOmar Emara     if (GetNumberOfFields() > 0) {
164829cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
164929cc50e1SOmar Emara       FieldDelegate &field = m_fields[m_selection_index];
165029cc50e1SOmar Emara       field.FieldDelegateSelectFirstElement();
165129cc50e1SOmar Emara     } else
165229cc50e1SOmar Emara       m_selection_type = SelectionType::NewButton;
165329cc50e1SOmar Emara   }
165429cc50e1SOmar Emara 
1655*0321dbc8SOmar Emara   HandleCharResult SelectNext(int key) {
165629cc50e1SOmar Emara     if (m_selection_type == SelectionType::NewButton)
165729cc50e1SOmar Emara       return eKeyNotHandled;
165829cc50e1SOmar Emara 
165929cc50e1SOmar Emara     if (m_selection_type == SelectionType::RemoveButton) {
166029cc50e1SOmar Emara       if (m_selection_index == GetNumberOfFields() - 1) {
166129cc50e1SOmar Emara         m_selection_type = SelectionType::NewButton;
166229cc50e1SOmar Emara         return eKeyHandled;
166329cc50e1SOmar Emara       }
166429cc50e1SOmar Emara       m_selection_index++;
166529cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
166629cc50e1SOmar Emara       FieldDelegate &next_field = m_fields[m_selection_index];
166729cc50e1SOmar Emara       next_field.FieldDelegateSelectFirstElement();
166829cc50e1SOmar Emara       return eKeyHandled;
166929cc50e1SOmar Emara     }
167029cc50e1SOmar Emara 
167129cc50e1SOmar Emara     FieldDelegate &field = m_fields[m_selection_index];
167229cc50e1SOmar Emara     if (!field.FieldDelegateOnLastOrOnlyElement()) {
167329cc50e1SOmar Emara       return field.FieldDelegateHandleChar(key);
167429cc50e1SOmar Emara     }
167529cc50e1SOmar Emara 
167629cc50e1SOmar Emara     field.FieldDelegateExitCallback();
167729cc50e1SOmar Emara 
167829cc50e1SOmar Emara     m_selection_type = SelectionType::RemoveButton;
167929cc50e1SOmar Emara     return eKeyHandled;
168029cc50e1SOmar Emara   }
168129cc50e1SOmar Emara 
168229cc50e1SOmar Emara   HandleCharResult SelectPrevious(int key) {
168329cc50e1SOmar Emara     if (FieldDelegateOnFirstOrOnlyElement())
168429cc50e1SOmar Emara       return eKeyNotHandled;
168529cc50e1SOmar Emara 
168629cc50e1SOmar Emara     if (m_selection_type == SelectionType::RemoveButton) {
168729cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
168829cc50e1SOmar Emara       FieldDelegate &field = m_fields[m_selection_index];
168929cc50e1SOmar Emara       field.FieldDelegateSelectLastElement();
169029cc50e1SOmar Emara       return eKeyHandled;
169129cc50e1SOmar Emara     }
169229cc50e1SOmar Emara 
169329cc50e1SOmar Emara     if (m_selection_type == SelectionType::NewButton) {
169429cc50e1SOmar Emara       m_selection_type = SelectionType::RemoveButton;
169529cc50e1SOmar Emara       m_selection_index = GetNumberOfFields() - 1;
169629cc50e1SOmar Emara       return eKeyHandled;
169729cc50e1SOmar Emara     }
169829cc50e1SOmar Emara 
169929cc50e1SOmar Emara     FieldDelegate &field = m_fields[m_selection_index];
170029cc50e1SOmar Emara     if (!field.FieldDelegateOnFirstOrOnlyElement()) {
170129cc50e1SOmar Emara       return field.FieldDelegateHandleChar(key);
170229cc50e1SOmar Emara     }
170329cc50e1SOmar Emara 
170429cc50e1SOmar Emara     field.FieldDelegateExitCallback();
170529cc50e1SOmar Emara 
170629cc50e1SOmar Emara     m_selection_type = SelectionType::RemoveButton;
170729cc50e1SOmar Emara     m_selection_index--;
170829cc50e1SOmar Emara     return eKeyHandled;
170929cc50e1SOmar Emara   }
171029cc50e1SOmar Emara 
171129cc50e1SOmar Emara   HandleCharResult FieldDelegateHandleChar(int key) override {
171229cc50e1SOmar Emara     switch (key) {
171329cc50e1SOmar Emara     case '\r':
171429cc50e1SOmar Emara     case '\n':
171529cc50e1SOmar Emara     case KEY_ENTER:
171629cc50e1SOmar Emara       switch (m_selection_type) {
171729cc50e1SOmar Emara       case SelectionType::NewButton:
171829cc50e1SOmar Emara         AddNewField();
171929cc50e1SOmar Emara         return eKeyHandled;
172029cc50e1SOmar Emara       case SelectionType::RemoveButton:
172129cc50e1SOmar Emara         RemoveField();
172229cc50e1SOmar Emara         return eKeyHandled;
172329cc50e1SOmar Emara       default:
172429cc50e1SOmar Emara         break;
172529cc50e1SOmar Emara       }
172629cc50e1SOmar Emara       break;
172729cc50e1SOmar Emara     case '\t':
1728*0321dbc8SOmar Emara       SelectNext(key);
172929cc50e1SOmar Emara       return eKeyHandled;
173029cc50e1SOmar Emara     case KEY_SHIFT_TAB:
173129cc50e1SOmar Emara       SelectPrevious(key);
173229cc50e1SOmar Emara       return eKeyHandled;
173329cc50e1SOmar Emara     default:
173429cc50e1SOmar Emara       break;
173529cc50e1SOmar Emara     }
173629cc50e1SOmar Emara 
173729cc50e1SOmar Emara     // If the key wasn't handled and one of the fields is selected, pass the key
173829cc50e1SOmar Emara     // to that field.
173929cc50e1SOmar Emara     if (m_selection_type == SelectionType::Field) {
174029cc50e1SOmar Emara       return m_fields[m_selection_index].FieldDelegateHandleChar(key);
174129cc50e1SOmar Emara     }
174229cc50e1SOmar Emara 
174329cc50e1SOmar Emara     return eKeyNotHandled;
174429cc50e1SOmar Emara   }
174529cc50e1SOmar Emara 
174629cc50e1SOmar Emara   bool FieldDelegateOnLastOrOnlyElement() override {
174729cc50e1SOmar Emara     if (m_selection_type == SelectionType::NewButton) {
174829cc50e1SOmar Emara       return true;
174929cc50e1SOmar Emara     }
175029cc50e1SOmar Emara     return false;
175129cc50e1SOmar Emara   }
175229cc50e1SOmar Emara 
175329cc50e1SOmar Emara   bool FieldDelegateOnFirstOrOnlyElement() override {
175429cc50e1SOmar Emara     if (m_selection_type == SelectionType::NewButton &&
175529cc50e1SOmar Emara         GetNumberOfFields() == 0)
175629cc50e1SOmar Emara       return true;
175729cc50e1SOmar Emara 
175829cc50e1SOmar Emara     if (m_selection_type == SelectionType::Field && m_selection_index == 0) {
175929cc50e1SOmar Emara       FieldDelegate &field = m_fields[m_selection_index];
176029cc50e1SOmar Emara       return field.FieldDelegateOnFirstOrOnlyElement();
176129cc50e1SOmar Emara     }
176229cc50e1SOmar Emara 
176329cc50e1SOmar Emara     return false;
176429cc50e1SOmar Emara   }
176529cc50e1SOmar Emara 
176629cc50e1SOmar Emara   void FieldDelegateSelectFirstElement() override {
176729cc50e1SOmar Emara     if (GetNumberOfFields() == 0) {
176829cc50e1SOmar Emara       m_selection_type = SelectionType::NewButton;
176929cc50e1SOmar Emara       return;
177029cc50e1SOmar Emara     }
177129cc50e1SOmar Emara 
177229cc50e1SOmar Emara     m_selection_type = SelectionType::Field;
177329cc50e1SOmar Emara     m_selection_index = 0;
177429cc50e1SOmar Emara   }
177529cc50e1SOmar Emara 
177629cc50e1SOmar Emara   void FieldDelegateSelectLastElement() override {
177729cc50e1SOmar Emara     m_selection_type = SelectionType::NewButton;
177829cc50e1SOmar Emara     return;
177929cc50e1SOmar Emara   }
178029cc50e1SOmar Emara 
178129cc50e1SOmar Emara   int GetNumberOfFields() { return m_fields.size(); }
178229cc50e1SOmar Emara 
178329cc50e1SOmar Emara   // Returns the form delegate at the current index.
178429cc50e1SOmar Emara   T &GetField(int index) { return m_fields[index]; }
178529cc50e1SOmar Emara 
178629cc50e1SOmar Emara protected:
178729cc50e1SOmar Emara   std::string m_label;
178829cc50e1SOmar Emara   // The default field delegate instance from which new field delegates will be
178929cc50e1SOmar Emara   // created though a copy.
179029cc50e1SOmar Emara   T m_default_field;
179129cc50e1SOmar Emara   std::vector<T> m_fields;
179229cc50e1SOmar Emara   int m_selection_index;
179329cc50e1SOmar Emara   // See SelectionType class enum.
179429cc50e1SOmar Emara   SelectionType m_selection_type;
179529cc50e1SOmar Emara };
179629cc50e1SOmar Emara 
179729cc50e1SOmar Emara class FormAction {
179829cc50e1SOmar Emara public:
179929cc50e1SOmar Emara   FormAction(const char *label, std::function<void(Window &)> action)
180029cc50e1SOmar Emara       : m_action(action) {
180129cc50e1SOmar Emara     if (label)
180229cc50e1SOmar Emara       m_label = label;
180329cc50e1SOmar Emara   }
180429cc50e1SOmar Emara 
180529cc50e1SOmar Emara   // Draw a centered [Label].
180629cc50e1SOmar Emara   void Draw(SubPad &surface, bool is_selected) {
180729cc50e1SOmar Emara     int x = (surface.GetWidth() - m_label.length()) / 2;
180829cc50e1SOmar Emara     surface.MoveCursor(x, 0);
180929cc50e1SOmar Emara     if (is_selected)
181029cc50e1SOmar Emara       surface.AttributeOn(A_REVERSE);
181129cc50e1SOmar Emara     surface.PutChar('[');
181229cc50e1SOmar Emara     surface.PutCString(m_label.c_str());
181329cc50e1SOmar Emara     surface.PutChar(']');
181429cc50e1SOmar Emara     if (is_selected)
181529cc50e1SOmar Emara       surface.AttributeOff(A_REVERSE);
181629cc50e1SOmar Emara   }
181729cc50e1SOmar Emara 
181829cc50e1SOmar Emara   void Execute(Window &window) { m_action(window); }
181929cc50e1SOmar Emara 
182029cc50e1SOmar Emara   const std::string &GetLabel() { return m_label; }
182129cc50e1SOmar Emara 
182229cc50e1SOmar Emara protected:
182329cc50e1SOmar Emara   std::string m_label;
182429cc50e1SOmar Emara   std::function<void(Window &)> m_action;
182529cc50e1SOmar Emara };
182629cc50e1SOmar Emara 
182729cc50e1SOmar Emara class FormDelegate {
182829cc50e1SOmar Emara public:
182929cc50e1SOmar Emara   FormDelegate() {}
183029cc50e1SOmar Emara 
183129cc50e1SOmar Emara   virtual ~FormDelegate() = default;
183229cc50e1SOmar Emara 
1833*0321dbc8SOmar Emara   virtual std::string GetName() = 0;
1834*0321dbc8SOmar Emara 
1835*0321dbc8SOmar Emara   virtual void UpdateFieldsVisibility() { return; }
1836*0321dbc8SOmar Emara 
1837*0321dbc8SOmar Emara   FieldDelegate *GetField(uint32_t field_index) {
1838*0321dbc8SOmar Emara     if (field_index < m_fields.size())
1839*0321dbc8SOmar Emara       return m_fields[field_index].get();
1840*0321dbc8SOmar Emara     return nullptr;
1841*0321dbc8SOmar Emara   }
184229cc50e1SOmar Emara 
184329cc50e1SOmar Emara   FormAction &GetAction(int action_index) { return m_actions[action_index]; }
184429cc50e1SOmar Emara 
184529cc50e1SOmar Emara   int GetNumberOfFields() { return m_fields.size(); }
184629cc50e1SOmar Emara 
184729cc50e1SOmar Emara   int GetNumberOfActions() { return m_actions.size(); }
184829cc50e1SOmar Emara 
184929cc50e1SOmar Emara   bool HasError() { return !m_error.empty(); }
185029cc50e1SOmar Emara 
185129cc50e1SOmar Emara   void ClearError() { m_error.clear(); }
185229cc50e1SOmar Emara 
185329cc50e1SOmar Emara   const std::string &GetError() { return m_error; }
185429cc50e1SOmar Emara 
185529cc50e1SOmar Emara   void SetError(const char *error) { m_error = error; }
185629cc50e1SOmar Emara 
185729cc50e1SOmar Emara   // Factory methods to create and add fields of specific types.
185829cc50e1SOmar Emara 
185929cc50e1SOmar Emara   TextFieldDelegate *AddTextField(const char *label, const char *content) {
186029cc50e1SOmar Emara     TextFieldDelegate *delegate = new TextFieldDelegate(label, content);
1861*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
186229cc50e1SOmar Emara     return delegate;
186329cc50e1SOmar Emara   }
186429cc50e1SOmar Emara 
186529cc50e1SOmar Emara   FileFieldDelegate *AddFileField(const char *label, const char *content,
186629cc50e1SOmar Emara                                   bool need_to_exist = true) {
186729cc50e1SOmar Emara     FileFieldDelegate *delegate =
186829cc50e1SOmar Emara         new FileFieldDelegate(label, content, need_to_exist);
1869*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
187029cc50e1SOmar Emara     return delegate;
187129cc50e1SOmar Emara   }
187229cc50e1SOmar Emara 
187329cc50e1SOmar Emara   DirectoryFieldDelegate *AddDirectoryField(const char *label,
187429cc50e1SOmar Emara                                             const char *content,
187529cc50e1SOmar Emara                                             bool need_to_exist = true) {
187629cc50e1SOmar Emara     DirectoryFieldDelegate *delegate =
187729cc50e1SOmar Emara         new DirectoryFieldDelegate(label, content, need_to_exist);
1878*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
187929cc50e1SOmar Emara     return delegate;
188029cc50e1SOmar Emara   }
188129cc50e1SOmar Emara 
188229cc50e1SOmar Emara   IntegerFieldDelegate *AddIntegerField(const char *label, int content) {
188329cc50e1SOmar Emara     IntegerFieldDelegate *delegate = new IntegerFieldDelegate(label, content);
1884*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
188529cc50e1SOmar Emara     return delegate;
188629cc50e1SOmar Emara   }
188729cc50e1SOmar Emara 
188829cc50e1SOmar Emara   BooleanFieldDelegate *AddBooleanField(const char *label, bool content) {
188929cc50e1SOmar Emara     BooleanFieldDelegate *delegate = new BooleanFieldDelegate(label, content);
1890*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
189129cc50e1SOmar Emara     return delegate;
189229cc50e1SOmar Emara   }
189329cc50e1SOmar Emara 
189429cc50e1SOmar Emara   ChoicesFieldDelegate *AddChoicesField(const char *label, int height,
189529cc50e1SOmar Emara                                         std::vector<std::string> choices) {
189629cc50e1SOmar Emara     ChoicesFieldDelegate *delegate =
189729cc50e1SOmar Emara         new ChoicesFieldDelegate(label, height, choices);
1898*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
189929cc50e1SOmar Emara     return delegate;
190029cc50e1SOmar Emara   }
190129cc50e1SOmar Emara 
190229cc50e1SOmar Emara   template <class T>
190329cc50e1SOmar Emara   ListFieldDelegate<T> *AddListField(const char *label, T default_field) {
190429cc50e1SOmar Emara     ListFieldDelegate<T> *delegate =
190529cc50e1SOmar Emara         new ListFieldDelegate<T>(label, default_field);
1906*0321dbc8SOmar Emara     m_fields.push_back(FieldDelegateUP(delegate));
190729cc50e1SOmar Emara     return delegate;
190829cc50e1SOmar Emara   }
190929cc50e1SOmar Emara 
191029cc50e1SOmar Emara   // Factory methods for adding actions.
191129cc50e1SOmar Emara 
191229cc50e1SOmar Emara   void AddAction(const char *label, std::function<void(Window &)> action) {
191329cc50e1SOmar Emara     m_actions.push_back(FormAction(label, action));
191429cc50e1SOmar Emara   }
191529cc50e1SOmar Emara 
191629cc50e1SOmar Emara protected:
1917*0321dbc8SOmar Emara   std::vector<FieldDelegateUP> m_fields;
191829cc50e1SOmar Emara   std::vector<FormAction> m_actions;
191929cc50e1SOmar Emara   // Optional error message. If empty, form is considered to have no error.
192029cc50e1SOmar Emara   std::string m_error;
192129cc50e1SOmar Emara };
192229cc50e1SOmar Emara 
192329cc50e1SOmar Emara typedef std::shared_ptr<FormDelegate> FormDelegateSP;
192429cc50e1SOmar Emara 
192529cc50e1SOmar Emara class FormWindowDelegate : public WindowDelegate {
192629cc50e1SOmar Emara public:
192729cc50e1SOmar Emara   FormWindowDelegate(FormDelegateSP &delegate_sp)
192829cc50e1SOmar Emara       : m_delegate_sp(delegate_sp), m_selection_index(0),
1929*0321dbc8SOmar Emara         m_first_visible_line(0) {
1930*0321dbc8SOmar Emara     assert(m_delegate_sp->GetNumberOfActions() > 0);
1931*0321dbc8SOmar Emara     if (m_delegate_sp->GetNumberOfFields() > 0)
1932*0321dbc8SOmar Emara       m_selection_type = SelectionType::Field;
1933*0321dbc8SOmar Emara     else
1934*0321dbc8SOmar Emara       m_selection_type = SelectionType::Action;
1935*0321dbc8SOmar Emara   }
193629cc50e1SOmar Emara 
193729cc50e1SOmar Emara   // Signify which element is selected. If a field or an action is selected,
193829cc50e1SOmar Emara   // then m_selection_index signifies the particular field or action that is
193929cc50e1SOmar Emara   // selected.
194029cc50e1SOmar Emara   enum class SelectionType { Field, Action };
194129cc50e1SOmar Emara 
194229cc50e1SOmar Emara   // A form window is padded by one character from all sides. First, if an error
194329cc50e1SOmar Emara   // message exists, it is drawn followed by a separator. Then one or more
194429cc50e1SOmar Emara   // fields are drawn. Finally, all available actions are drawn on a single
194529cc50e1SOmar Emara   // line.
194629cc50e1SOmar Emara   //
194729cc50e1SOmar Emara   // ___<Form Name>_________________________________________________
194829cc50e1SOmar Emara   // |                                                             |
194929cc50e1SOmar Emara   // | - Error message if it exists.                               |
195029cc50e1SOmar Emara   // |-------------------------------------------------------------|
195129cc50e1SOmar Emara   // | Form elements here.                                         |
195229cc50e1SOmar Emara   // |                       Form actions here.                    |
195329cc50e1SOmar Emara   // |                                                             |
195429cc50e1SOmar Emara   // |______________________________________[Press Esc to cancel]__|
195529cc50e1SOmar Emara   //
195629cc50e1SOmar Emara 
195729cc50e1SOmar Emara   // One line for the error and another for the horizontal line.
195829cc50e1SOmar Emara   int GetErrorHeight() {
195929cc50e1SOmar Emara     if (m_delegate_sp->HasError())
196029cc50e1SOmar Emara       return 2;
196129cc50e1SOmar Emara     return 0;
196229cc50e1SOmar Emara   }
196329cc50e1SOmar Emara 
196429cc50e1SOmar Emara   // Actions span a single line.
196529cc50e1SOmar Emara   int GetActionsHeight() {
196629cc50e1SOmar Emara     if (m_delegate_sp->GetNumberOfActions() > 0)
196729cc50e1SOmar Emara       return 1;
196829cc50e1SOmar Emara     return 0;
196929cc50e1SOmar Emara   }
197029cc50e1SOmar Emara 
197129cc50e1SOmar Emara   // Get the total number of needed lines to draw the contents.
197229cc50e1SOmar Emara   int GetContentHeight() {
197329cc50e1SOmar Emara     int height = 0;
197429cc50e1SOmar Emara     height += GetErrorHeight();
197529cc50e1SOmar Emara     for (int i = 0; i < m_delegate_sp->GetNumberOfFields(); i++) {
1976*0321dbc8SOmar Emara       if (!m_delegate_sp->GetField(i)->FieldDelegateIsVisible())
1977*0321dbc8SOmar Emara         continue;
197829cc50e1SOmar Emara       height += m_delegate_sp->GetField(i)->FieldDelegateGetHeight();
197929cc50e1SOmar Emara     }
198029cc50e1SOmar Emara     height += GetActionsHeight();
198129cc50e1SOmar Emara     return height;
198229cc50e1SOmar Emara   }
198329cc50e1SOmar Emara 
198429cc50e1SOmar Emara   ScrollContext GetScrollContext() {
198529cc50e1SOmar Emara     if (m_selection_type == SelectionType::Action)
198629cc50e1SOmar Emara       return ScrollContext(GetContentHeight() - 1);
198729cc50e1SOmar Emara 
1988*0321dbc8SOmar Emara     FieldDelegate *field = m_delegate_sp->GetField(m_selection_index);
198929cc50e1SOmar Emara     ScrollContext context = field->FieldDelegateGetScrollContext();
199029cc50e1SOmar Emara 
199129cc50e1SOmar Emara     int offset = GetErrorHeight();
199229cc50e1SOmar Emara     for (int i = 0; i < m_selection_index; i++) {
1993*0321dbc8SOmar Emara       if (!m_delegate_sp->GetField(i)->FieldDelegateIsVisible())
1994*0321dbc8SOmar Emara         continue;
199529cc50e1SOmar Emara       offset += m_delegate_sp->GetField(i)->FieldDelegateGetHeight();
199629cc50e1SOmar Emara     }
199729cc50e1SOmar Emara     context.Offset(offset);
199829cc50e1SOmar Emara 
199929cc50e1SOmar Emara     // If the context is touching the error, include the error in the context as
200029cc50e1SOmar Emara     // well.
200129cc50e1SOmar Emara     if (context.start == GetErrorHeight())
200229cc50e1SOmar Emara       context.start = 0;
200329cc50e1SOmar Emara 
200429cc50e1SOmar Emara     return context;
200529cc50e1SOmar Emara   }
200629cc50e1SOmar Emara 
200729cc50e1SOmar Emara   void UpdateScrolling(DerivedWindow &surface) {
200829cc50e1SOmar Emara     ScrollContext context = GetScrollContext();
200929cc50e1SOmar Emara     int content_height = GetContentHeight();
201029cc50e1SOmar Emara     int surface_height = surface.GetHeight();
201129cc50e1SOmar Emara     int visible_height = std::min(content_height, surface_height);
201229cc50e1SOmar Emara     int last_visible_line = m_first_visible_line + visible_height - 1;
201329cc50e1SOmar Emara 
201429cc50e1SOmar Emara     // If the last visible line is bigger than the content, then it is invalid
201529cc50e1SOmar Emara     // and needs to be set to the last line in the content. This can happen when
201629cc50e1SOmar Emara     // a field has shrunk in height.
201729cc50e1SOmar Emara     if (last_visible_line > content_height - 1) {
201829cc50e1SOmar Emara       m_first_visible_line = content_height - visible_height;
201929cc50e1SOmar Emara     }
202029cc50e1SOmar Emara 
202129cc50e1SOmar Emara     if (context.start < m_first_visible_line) {
202229cc50e1SOmar Emara       m_first_visible_line = context.start;
202329cc50e1SOmar Emara       return;
202429cc50e1SOmar Emara     }
202529cc50e1SOmar Emara 
202629cc50e1SOmar Emara     if (context.end > last_visible_line) {
202729cc50e1SOmar Emara       m_first_visible_line = context.end - visible_height + 1;
202829cc50e1SOmar Emara     }
202929cc50e1SOmar Emara   }
203029cc50e1SOmar Emara 
203129cc50e1SOmar Emara   void DrawError(SubPad &surface) {
203229cc50e1SOmar Emara     if (!m_delegate_sp->HasError())
203329cc50e1SOmar Emara       return;
203429cc50e1SOmar Emara     surface.MoveCursor(0, 0);
203529cc50e1SOmar Emara     surface.AttributeOn(COLOR_PAIR(RedOnBlack));
203629cc50e1SOmar Emara     surface.PutChar(ACS_DIAMOND);
203729cc50e1SOmar Emara     surface.PutChar(' ');
203829cc50e1SOmar Emara     surface.PutCStringTruncated(1, m_delegate_sp->GetError().c_str());
203929cc50e1SOmar Emara     surface.AttributeOff(COLOR_PAIR(RedOnBlack));
204029cc50e1SOmar Emara 
204129cc50e1SOmar Emara     surface.MoveCursor(0, 1);
204229cc50e1SOmar Emara     surface.HorizontalLine(surface.GetWidth());
204329cc50e1SOmar Emara   }
204429cc50e1SOmar Emara 
204529cc50e1SOmar Emara   void DrawFields(SubPad &surface) {
204629cc50e1SOmar Emara     int line = 0;
204729cc50e1SOmar Emara     int width = surface.GetWidth();
204829cc50e1SOmar Emara     bool a_field_is_selected = m_selection_type == SelectionType::Field;
204929cc50e1SOmar Emara     for (int i = 0; i < m_delegate_sp->GetNumberOfFields(); i++) {
2050*0321dbc8SOmar Emara       FieldDelegate *field = m_delegate_sp->GetField(i);
2051*0321dbc8SOmar Emara       if (!field->FieldDelegateIsVisible())
2052*0321dbc8SOmar Emara         continue;
205329cc50e1SOmar Emara       bool is_field_selected = a_field_is_selected && m_selection_index == i;
205429cc50e1SOmar Emara       int height = field->FieldDelegateGetHeight();
205529cc50e1SOmar Emara       Rect bounds = Rect(Point(0, line), Size(width, height));
205629cc50e1SOmar Emara       SubPad field_surface = SubPad(surface, bounds);
205729cc50e1SOmar Emara       field->FieldDelegateDraw(field_surface, is_field_selected);
205829cc50e1SOmar Emara       line += height;
205929cc50e1SOmar Emara     }
206029cc50e1SOmar Emara   }
206129cc50e1SOmar Emara 
206229cc50e1SOmar Emara   void DrawActions(SubPad &surface) {
206329cc50e1SOmar Emara     int number_of_actions = m_delegate_sp->GetNumberOfActions();
206429cc50e1SOmar Emara     int width = surface.GetWidth() / number_of_actions;
206529cc50e1SOmar Emara     bool an_action_is_selected = m_selection_type == SelectionType::Action;
206629cc50e1SOmar Emara     int x = 0;
206729cc50e1SOmar Emara     for (int i = 0; i < number_of_actions; i++) {
206829cc50e1SOmar Emara       bool is_action_selected = an_action_is_selected && m_selection_index == i;
206929cc50e1SOmar Emara       FormAction &action = m_delegate_sp->GetAction(i);
207029cc50e1SOmar Emara       Rect bounds = Rect(Point(x, 0), Size(width, 1));
207129cc50e1SOmar Emara       SubPad action_surface = SubPad(surface, bounds);
207229cc50e1SOmar Emara       action.Draw(action_surface, is_action_selected);
207329cc50e1SOmar Emara       x += width;
207429cc50e1SOmar Emara     }
207529cc50e1SOmar Emara   }
207629cc50e1SOmar Emara 
207729cc50e1SOmar Emara   void DrawElements(SubPad &surface) {
207829cc50e1SOmar Emara     Rect frame = surface.GetFrame();
207929cc50e1SOmar Emara     Rect fields_bounds, actions_bounds;
208029cc50e1SOmar Emara     frame.HorizontalSplit(surface.GetHeight() - GetActionsHeight(),
208129cc50e1SOmar Emara                           fields_bounds, actions_bounds);
208229cc50e1SOmar Emara     SubPad fields_surface = SubPad(surface, fields_bounds);
208329cc50e1SOmar Emara     SubPad actions_surface = SubPad(surface, actions_bounds);
208429cc50e1SOmar Emara 
208529cc50e1SOmar Emara     DrawFields(fields_surface);
208629cc50e1SOmar Emara     DrawActions(actions_surface);
208729cc50e1SOmar Emara   }
208829cc50e1SOmar Emara 
208929cc50e1SOmar Emara   // Contents are first drawn on a pad. Then a subset of that pad is copied to
209029cc50e1SOmar Emara   // the derived window starting at the first visible line. This essentially
209129cc50e1SOmar Emara   // provides scrolling functionality.
209229cc50e1SOmar Emara   void DrawContent(DerivedWindow &surface) {
209329cc50e1SOmar Emara     UpdateScrolling(surface);
209429cc50e1SOmar Emara 
209529cc50e1SOmar Emara     int width = surface.GetWidth();
209629cc50e1SOmar Emara     int height = GetContentHeight();
209729cc50e1SOmar Emara     Pad pad = Pad(Size(width, height));
209829cc50e1SOmar Emara 
209929cc50e1SOmar Emara     Rect frame = pad.GetFrame();
210029cc50e1SOmar Emara     Rect error_bounds, elements_bounds;
210129cc50e1SOmar Emara     frame.HorizontalSplit(GetErrorHeight(), error_bounds, elements_bounds);
210229cc50e1SOmar Emara     SubPad error_surface = SubPad(pad, error_bounds);
210329cc50e1SOmar Emara     SubPad elements_surface = SubPad(pad, elements_bounds);
210429cc50e1SOmar Emara 
210529cc50e1SOmar Emara     DrawError(error_surface);
210629cc50e1SOmar Emara     DrawElements(elements_surface);
210729cc50e1SOmar Emara 
210829cc50e1SOmar Emara     int copy_height = std::min(surface.GetHeight(), pad.GetHeight());
210929cc50e1SOmar Emara     pad.CopyToSurface(surface, Point(0, m_first_visible_line), Point(),
211029cc50e1SOmar Emara                       Size(width, copy_height));
211129cc50e1SOmar Emara   }
211229cc50e1SOmar Emara 
211329cc50e1SOmar Emara   bool WindowDelegateDraw(Window &window, bool force) override {
2114*0321dbc8SOmar Emara     m_delegate_sp->UpdateFieldsVisibility();
211529cc50e1SOmar Emara 
211629cc50e1SOmar Emara     window.Erase();
211729cc50e1SOmar Emara 
2118*0321dbc8SOmar Emara     window.DrawTitleBox(m_delegate_sp->GetName().c_str(),
2119*0321dbc8SOmar Emara                         "Press Esc to cancel");
212029cc50e1SOmar Emara 
212129cc50e1SOmar Emara     Rect content_bounds = window.GetFrame();
212229cc50e1SOmar Emara     content_bounds.Inset(2, 2);
212329cc50e1SOmar Emara     DerivedWindow content_surface = DerivedWindow(window, content_bounds);
212429cc50e1SOmar Emara 
212529cc50e1SOmar Emara     DrawContent(content_surface);
212629cc50e1SOmar Emara     return true;
212729cc50e1SOmar Emara   }
212829cc50e1SOmar Emara 
2129*0321dbc8SOmar Emara   void SkipNextHiddenFields() {
2130*0321dbc8SOmar Emara     while (true) {
2131*0321dbc8SOmar Emara       if (m_delegate_sp->GetField(m_selection_index)->FieldDelegateIsVisible())
2132*0321dbc8SOmar Emara         return;
2133*0321dbc8SOmar Emara 
2134*0321dbc8SOmar Emara       if (m_selection_index == m_delegate_sp->GetNumberOfFields() - 1) {
2135*0321dbc8SOmar Emara         m_selection_type = SelectionType::Action;
2136*0321dbc8SOmar Emara         m_selection_index = 0;
2137*0321dbc8SOmar Emara         return;
2138*0321dbc8SOmar Emara       }
2139*0321dbc8SOmar Emara 
2140*0321dbc8SOmar Emara       m_selection_index++;
2141*0321dbc8SOmar Emara     }
2142*0321dbc8SOmar Emara   }
2143*0321dbc8SOmar Emara 
2144*0321dbc8SOmar Emara   HandleCharResult SelectNext(int key) {
214529cc50e1SOmar Emara     if (m_selection_type == SelectionType::Action) {
214629cc50e1SOmar Emara       if (m_selection_index < m_delegate_sp->GetNumberOfActions() - 1) {
214729cc50e1SOmar Emara         m_selection_index++;
214829cc50e1SOmar Emara         return eKeyHandled;
214929cc50e1SOmar Emara       }
215029cc50e1SOmar Emara 
215129cc50e1SOmar Emara       m_selection_index = 0;
215229cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
2153*0321dbc8SOmar Emara       SkipNextHiddenFields();
2154*0321dbc8SOmar Emara       if (m_selection_type == SelectionType::Field) {
2155*0321dbc8SOmar Emara         FieldDelegate *next_field = m_delegate_sp->GetField(m_selection_index);
215629cc50e1SOmar Emara         next_field->FieldDelegateSelectFirstElement();
2157*0321dbc8SOmar Emara       }
215829cc50e1SOmar Emara       return eKeyHandled;
215929cc50e1SOmar Emara     }
216029cc50e1SOmar Emara 
2161*0321dbc8SOmar Emara     FieldDelegate *field = m_delegate_sp->GetField(m_selection_index);
216229cc50e1SOmar Emara     if (!field->FieldDelegateOnLastOrOnlyElement()) {
216329cc50e1SOmar Emara       return field->FieldDelegateHandleChar(key);
216429cc50e1SOmar Emara     }
216529cc50e1SOmar Emara 
216629cc50e1SOmar Emara     field->FieldDelegateExitCallback();
216729cc50e1SOmar Emara 
216829cc50e1SOmar Emara     if (m_selection_index == m_delegate_sp->GetNumberOfFields() - 1) {
216929cc50e1SOmar Emara       m_selection_type = SelectionType::Action;
217029cc50e1SOmar Emara       m_selection_index = 0;
217129cc50e1SOmar Emara       return eKeyHandled;
217229cc50e1SOmar Emara     }
217329cc50e1SOmar Emara 
217429cc50e1SOmar Emara     m_selection_index++;
2175*0321dbc8SOmar Emara     SkipNextHiddenFields();
217629cc50e1SOmar Emara 
2177*0321dbc8SOmar Emara     if (m_selection_type == SelectionType::Field) {
2178*0321dbc8SOmar Emara       FieldDelegate *next_field = m_delegate_sp->GetField(m_selection_index);
217929cc50e1SOmar Emara       next_field->FieldDelegateSelectFirstElement();
2180*0321dbc8SOmar Emara     }
218129cc50e1SOmar Emara 
218229cc50e1SOmar Emara     return eKeyHandled;
218329cc50e1SOmar Emara   }
218429cc50e1SOmar Emara 
2185*0321dbc8SOmar Emara   void SkipPreviousHiddenFields() {
2186*0321dbc8SOmar Emara     while (true) {
2187*0321dbc8SOmar Emara       if (m_delegate_sp->GetField(m_selection_index)->FieldDelegateIsVisible())
2188*0321dbc8SOmar Emara         return;
2189*0321dbc8SOmar Emara 
2190*0321dbc8SOmar Emara       if (m_selection_index == 0) {
2191*0321dbc8SOmar Emara         m_selection_type = SelectionType::Action;
2192*0321dbc8SOmar Emara         m_selection_index = 0;
2193*0321dbc8SOmar Emara         return;
2194*0321dbc8SOmar Emara       }
2195*0321dbc8SOmar Emara 
2196*0321dbc8SOmar Emara       m_selection_index--;
2197*0321dbc8SOmar Emara     }
2198*0321dbc8SOmar Emara   }
2199*0321dbc8SOmar Emara 
2200*0321dbc8SOmar Emara   HandleCharResult SelectPrevious(int key) {
220129cc50e1SOmar Emara     if (m_selection_type == SelectionType::Action) {
220229cc50e1SOmar Emara       if (m_selection_index > 0) {
220329cc50e1SOmar Emara         m_selection_index--;
220429cc50e1SOmar Emara         return eKeyHandled;
220529cc50e1SOmar Emara       }
220629cc50e1SOmar Emara       m_selection_index = m_delegate_sp->GetNumberOfFields() - 1;
220729cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
2208*0321dbc8SOmar Emara       SkipPreviousHiddenFields();
2209*0321dbc8SOmar Emara       if (m_selection_type == SelectionType::Field) {
2210*0321dbc8SOmar Emara         FieldDelegate *previous_field =
221129cc50e1SOmar Emara             m_delegate_sp->GetField(m_selection_index);
221229cc50e1SOmar Emara         previous_field->FieldDelegateSelectLastElement();
2213*0321dbc8SOmar Emara       }
221429cc50e1SOmar Emara       return eKeyHandled;
221529cc50e1SOmar Emara     }
221629cc50e1SOmar Emara 
2217*0321dbc8SOmar Emara     FieldDelegate *field = m_delegate_sp->GetField(m_selection_index);
221829cc50e1SOmar Emara     if (!field->FieldDelegateOnFirstOrOnlyElement()) {
221929cc50e1SOmar Emara       return field->FieldDelegateHandleChar(key);
222029cc50e1SOmar Emara     }
222129cc50e1SOmar Emara 
222229cc50e1SOmar Emara     field->FieldDelegateExitCallback();
222329cc50e1SOmar Emara 
222429cc50e1SOmar Emara     if (m_selection_index == 0) {
222529cc50e1SOmar Emara       m_selection_type = SelectionType::Action;
222629cc50e1SOmar Emara       m_selection_index = m_delegate_sp->GetNumberOfActions() - 1;
222729cc50e1SOmar Emara       return eKeyHandled;
222829cc50e1SOmar Emara     }
222929cc50e1SOmar Emara 
223029cc50e1SOmar Emara     m_selection_index--;
2231*0321dbc8SOmar Emara     SkipPreviousHiddenFields();
223229cc50e1SOmar Emara 
2233*0321dbc8SOmar Emara     if (m_selection_type == SelectionType::Field) {
2234*0321dbc8SOmar Emara       FieldDelegate *previous_field =
223529cc50e1SOmar Emara           m_delegate_sp->GetField(m_selection_index);
223629cc50e1SOmar Emara       previous_field->FieldDelegateSelectLastElement();
2237*0321dbc8SOmar Emara     }
223829cc50e1SOmar Emara 
223929cc50e1SOmar Emara     return eKeyHandled;
224029cc50e1SOmar Emara   }
224129cc50e1SOmar Emara 
224229cc50e1SOmar Emara   void ExecuteAction(Window &window) {
224329cc50e1SOmar Emara     FormAction &action = m_delegate_sp->GetAction(m_selection_index);
224429cc50e1SOmar Emara     action.Execute(window);
2245*0321dbc8SOmar Emara     if (m_delegate_sp->HasError()) {
224629cc50e1SOmar Emara       m_first_visible_line = 0;
224729cc50e1SOmar Emara       m_selection_index = 0;
224829cc50e1SOmar Emara       m_selection_type = SelectionType::Field;
224929cc50e1SOmar Emara     }
2250*0321dbc8SOmar Emara   }
225129cc50e1SOmar Emara 
225229cc50e1SOmar Emara   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override {
225329cc50e1SOmar Emara     switch (key) {
225429cc50e1SOmar Emara     case '\r':
225529cc50e1SOmar Emara     case '\n':
225629cc50e1SOmar Emara     case KEY_ENTER:
225729cc50e1SOmar Emara       if (m_selection_type == SelectionType::Action) {
225829cc50e1SOmar Emara         ExecuteAction(window);
225929cc50e1SOmar Emara         return eKeyHandled;
226029cc50e1SOmar Emara       }
226129cc50e1SOmar Emara       break;
226229cc50e1SOmar Emara     case '\t':
2263*0321dbc8SOmar Emara       return SelectNext(key);
226429cc50e1SOmar Emara     case KEY_SHIFT_TAB:
2265*0321dbc8SOmar Emara       return SelectPrevious(key);
226629cc50e1SOmar Emara     case KEY_ESCAPE:
226729cc50e1SOmar Emara       window.GetParent()->RemoveSubWindow(&window);
226829cc50e1SOmar Emara       return eKeyHandled;
226929cc50e1SOmar Emara     default:
227029cc50e1SOmar Emara       break;
227129cc50e1SOmar Emara     }
227229cc50e1SOmar Emara 
227329cc50e1SOmar Emara     // If the key wasn't handled and one of the fields is selected, pass the key
227429cc50e1SOmar Emara     // to that field.
227529cc50e1SOmar Emara     if (m_selection_type == SelectionType::Field) {
2276*0321dbc8SOmar Emara       FieldDelegate *field = m_delegate_sp->GetField(m_selection_index);
227729cc50e1SOmar Emara       return field->FieldDelegateHandleChar(key);
227829cc50e1SOmar Emara     }
227929cc50e1SOmar Emara 
228029cc50e1SOmar Emara     return eKeyNotHandled;
228129cc50e1SOmar Emara   }
228229cc50e1SOmar Emara 
228329cc50e1SOmar Emara protected:
228429cc50e1SOmar Emara   FormDelegateSP m_delegate_sp;
228529cc50e1SOmar Emara   // The index of the currently selected SelectionType.
228629cc50e1SOmar Emara   int m_selection_index;
228729cc50e1SOmar Emara   // See SelectionType class enum.
228829cc50e1SOmar Emara   SelectionType m_selection_type;
228929cc50e1SOmar Emara   // The first visible line from the pad.
229029cc50e1SOmar Emara   int m_first_visible_line;
229129cc50e1SOmar Emara };
229229cc50e1SOmar Emara 
2293*0321dbc8SOmar Emara ///////////////////////////
2294*0321dbc8SOmar Emara // Form Delegate Instances
2295*0321dbc8SOmar Emara ///////////////////////////
2296*0321dbc8SOmar Emara 
2297*0321dbc8SOmar Emara class DetachOrKillProcessFormDelegate : public FormDelegate {
2298*0321dbc8SOmar Emara public:
2299*0321dbc8SOmar Emara   DetachOrKillProcessFormDelegate(Process *process) : m_process(process) {
2300*0321dbc8SOmar Emara     SetError("There is a running process, either detach or kill it.");
2301*0321dbc8SOmar Emara 
2302*0321dbc8SOmar Emara     m_keep_stopped_field =
2303*0321dbc8SOmar Emara         AddBooleanField("Keep process stopped when detaching.", false);
2304*0321dbc8SOmar Emara 
2305*0321dbc8SOmar Emara     AddAction("Detach", [this](Window &window) { Detach(window); });
2306*0321dbc8SOmar Emara     AddAction("Kill", [this](Window &window) { Kill(window); });
2307*0321dbc8SOmar Emara   }
2308*0321dbc8SOmar Emara 
2309*0321dbc8SOmar Emara   std::string GetName() override { return "Detach/Kill Process"; }
2310*0321dbc8SOmar Emara 
2311*0321dbc8SOmar Emara   void Kill(Window &window) {
2312*0321dbc8SOmar Emara     Status destroy_status(m_process->Destroy(false));
2313*0321dbc8SOmar Emara     if (destroy_status.Fail()) {
2314*0321dbc8SOmar Emara       SetError("Failed to kill process.");
2315*0321dbc8SOmar Emara       return;
2316*0321dbc8SOmar Emara     }
2317*0321dbc8SOmar Emara     window.GetParent()->RemoveSubWindow(&window);
2318*0321dbc8SOmar Emara   }
2319*0321dbc8SOmar Emara 
2320*0321dbc8SOmar Emara   void Detach(Window &window) {
2321*0321dbc8SOmar Emara     Status detach_status(m_process->Detach(m_keep_stopped_field->GetBoolean()));
2322*0321dbc8SOmar Emara     if (detach_status.Fail()) {
2323*0321dbc8SOmar Emara       SetError("Failed to detach from process.");
2324*0321dbc8SOmar Emara       return;
2325*0321dbc8SOmar Emara     }
2326*0321dbc8SOmar Emara     window.GetParent()->RemoveSubWindow(&window);
2327*0321dbc8SOmar Emara   }
2328*0321dbc8SOmar Emara 
2329*0321dbc8SOmar Emara protected:
2330*0321dbc8SOmar Emara   Process *m_process;
2331*0321dbc8SOmar Emara   BooleanFieldDelegate *m_keep_stopped_field;
2332*0321dbc8SOmar Emara };
2333*0321dbc8SOmar Emara 
2334*0321dbc8SOmar Emara class ProcessAttachFormDelegate : public FormDelegate {
2335*0321dbc8SOmar Emara public:
2336*0321dbc8SOmar Emara   ProcessAttachFormDelegate(Debugger &debugger, WindowSP main_window_sp)
2337*0321dbc8SOmar Emara       : m_debugger(debugger), m_main_window_sp(main_window_sp) {
2338*0321dbc8SOmar Emara     std::vector<std::string> types;
2339*0321dbc8SOmar Emara     types.push_back(std::string("Name"));
2340*0321dbc8SOmar Emara     types.push_back(std::string("PID"));
2341*0321dbc8SOmar Emara     m_type_field = AddChoicesField("Attach By", 2, types);
2342*0321dbc8SOmar Emara     m_pid_field = AddIntegerField("PID", 0);
2343*0321dbc8SOmar Emara     m_name_field =
2344*0321dbc8SOmar Emara         AddTextField("Process Name", GetDefaultProcessName().c_str());
2345*0321dbc8SOmar Emara     m_continue_field = AddBooleanField("Continue once attached.", false);
2346*0321dbc8SOmar Emara     m_wait_for_field = AddBooleanField("Wait for process to launch.", false);
2347*0321dbc8SOmar Emara     m_include_existing_field =
2348*0321dbc8SOmar Emara         AddBooleanField("Include existing processes.", false);
2349*0321dbc8SOmar Emara     m_show_advanced_field = AddBooleanField("Show advanced settings.", false);
2350*0321dbc8SOmar Emara     m_plugin_field =
2351*0321dbc8SOmar Emara         AddChoicesField("Plugin Name", 3, GetPossiblePluginNames());
2352*0321dbc8SOmar Emara 
2353*0321dbc8SOmar Emara     AddAction("Attach", [this](Window &window) { Attach(window); });
2354*0321dbc8SOmar Emara   }
2355*0321dbc8SOmar Emara 
2356*0321dbc8SOmar Emara   std::string GetName() override { return "Attach Process"; }
2357*0321dbc8SOmar Emara 
2358*0321dbc8SOmar Emara   void UpdateFieldsVisibility() override {
2359*0321dbc8SOmar Emara     if (m_type_field->GetChoiceContent() == "Name") {
2360*0321dbc8SOmar Emara       m_pid_field->FieldDelegateHide();
2361*0321dbc8SOmar Emara       m_name_field->FieldDelegateShow();
2362*0321dbc8SOmar Emara       m_wait_for_field->FieldDelegateShow();
2363*0321dbc8SOmar Emara       if (m_wait_for_field->GetBoolean())
2364*0321dbc8SOmar Emara         m_include_existing_field->FieldDelegateShow();
2365*0321dbc8SOmar Emara       else
2366*0321dbc8SOmar Emara         m_include_existing_field->FieldDelegateHide();
2367*0321dbc8SOmar Emara     } else {
2368*0321dbc8SOmar Emara       m_pid_field->FieldDelegateShow();
2369*0321dbc8SOmar Emara       m_name_field->FieldDelegateHide();
2370*0321dbc8SOmar Emara       m_wait_for_field->FieldDelegateHide();
2371*0321dbc8SOmar Emara       m_include_existing_field->FieldDelegateHide();
2372*0321dbc8SOmar Emara     }
2373*0321dbc8SOmar Emara     if (m_show_advanced_field->GetBoolean())
2374*0321dbc8SOmar Emara       m_plugin_field->FieldDelegateShow();
2375*0321dbc8SOmar Emara     else
2376*0321dbc8SOmar Emara       m_plugin_field->FieldDelegateHide();
2377*0321dbc8SOmar Emara   }
2378*0321dbc8SOmar Emara 
2379*0321dbc8SOmar Emara   // Get the basename of the target's main executable if available, empty string
2380*0321dbc8SOmar Emara   // otherwise.
2381*0321dbc8SOmar Emara   std::string GetDefaultProcessName() {
2382*0321dbc8SOmar Emara     Target *target = m_debugger.GetSelectedTarget().get();
2383*0321dbc8SOmar Emara     if (target == nullptr)
2384*0321dbc8SOmar Emara       return "";
2385*0321dbc8SOmar Emara 
2386*0321dbc8SOmar Emara     ModuleSP module_sp = target->GetExecutableModule();
2387*0321dbc8SOmar Emara     if (!module_sp->IsExecutable())
2388*0321dbc8SOmar Emara       return "";
2389*0321dbc8SOmar Emara 
2390*0321dbc8SOmar Emara     return module_sp->GetFileSpec().GetFilename().AsCString();
2391*0321dbc8SOmar Emara   }
2392*0321dbc8SOmar Emara 
2393*0321dbc8SOmar Emara   std::vector<std::string> GetPossiblePluginNames() {
2394*0321dbc8SOmar Emara     std::vector<std::string> names;
2395*0321dbc8SOmar Emara     names.push_back("<default>");
2396*0321dbc8SOmar Emara 
2397*0321dbc8SOmar Emara     size_t i = 0;
2398*0321dbc8SOmar Emara     while (auto name = PluginManager::GetProcessPluginNameAtIndex(i++))
2399*0321dbc8SOmar Emara       names.push_back(name);
2400*0321dbc8SOmar Emara     return names;
2401*0321dbc8SOmar Emara   }
2402*0321dbc8SOmar Emara 
2403*0321dbc8SOmar Emara   bool StopRunningProcess() {
2404*0321dbc8SOmar Emara     ExecutionContext exe_ctx =
2405*0321dbc8SOmar Emara         m_debugger.GetCommandInterpreter().GetExecutionContext();
2406*0321dbc8SOmar Emara 
2407*0321dbc8SOmar Emara     if (!exe_ctx.HasProcessScope())
2408*0321dbc8SOmar Emara       return false;
2409*0321dbc8SOmar Emara 
2410*0321dbc8SOmar Emara     Process *process = exe_ctx.GetProcessPtr();
2411*0321dbc8SOmar Emara     if (!(process && process->IsAlive()))
2412*0321dbc8SOmar Emara       return false;
2413*0321dbc8SOmar Emara 
2414*0321dbc8SOmar Emara     FormDelegateSP form_delegate_sp =
2415*0321dbc8SOmar Emara         FormDelegateSP(new DetachOrKillProcessFormDelegate(process));
2416*0321dbc8SOmar Emara     Rect bounds = m_main_window_sp->GetCenteredRect(85, 8);
2417*0321dbc8SOmar Emara     WindowSP form_window_sp = m_main_window_sp->CreateSubWindow(
2418*0321dbc8SOmar Emara         form_delegate_sp->GetName().c_str(), bounds, true);
2419*0321dbc8SOmar Emara     WindowDelegateSP window_delegate_sp =
2420*0321dbc8SOmar Emara         WindowDelegateSP(new FormWindowDelegate(form_delegate_sp));
2421*0321dbc8SOmar Emara     form_window_sp->SetDelegate(window_delegate_sp);
2422*0321dbc8SOmar Emara 
2423*0321dbc8SOmar Emara     return true;
2424*0321dbc8SOmar Emara   }
2425*0321dbc8SOmar Emara 
2426*0321dbc8SOmar Emara   Target *GetTarget() {
2427*0321dbc8SOmar Emara     Target *target = m_debugger.GetSelectedTarget().get();
2428*0321dbc8SOmar Emara 
2429*0321dbc8SOmar Emara     if (target != nullptr)
2430*0321dbc8SOmar Emara       return target;
2431*0321dbc8SOmar Emara 
2432*0321dbc8SOmar Emara     TargetSP new_target_sp;
2433*0321dbc8SOmar Emara     m_debugger.GetTargetList().CreateTarget(
2434*0321dbc8SOmar Emara         m_debugger, "", "", eLoadDependentsNo, nullptr, new_target_sp);
2435*0321dbc8SOmar Emara 
2436*0321dbc8SOmar Emara     target = new_target_sp.get();
2437*0321dbc8SOmar Emara 
2438*0321dbc8SOmar Emara     if (target == nullptr)
2439*0321dbc8SOmar Emara       SetError("Failed to create target.");
2440*0321dbc8SOmar Emara 
2441*0321dbc8SOmar Emara     m_debugger.GetTargetList().SetSelectedTarget(new_target_sp);
2442*0321dbc8SOmar Emara 
2443*0321dbc8SOmar Emara     return target;
2444*0321dbc8SOmar Emara   }
2445*0321dbc8SOmar Emara 
2446*0321dbc8SOmar Emara   ProcessAttachInfo GetAttachInfo() {
2447*0321dbc8SOmar Emara     ProcessAttachInfo attach_info;
2448*0321dbc8SOmar Emara     attach_info.SetContinueOnceAttached(m_continue_field->GetBoolean());
2449*0321dbc8SOmar Emara     if (m_type_field->GetChoiceContent() == "Name") {
2450*0321dbc8SOmar Emara       attach_info.GetExecutableFile().SetFile(m_name_field->GetText(),
2451*0321dbc8SOmar Emara                                               FileSpec::Style::native);
2452*0321dbc8SOmar Emara       attach_info.SetWaitForLaunch(m_wait_for_field->GetBoolean());
2453*0321dbc8SOmar Emara       if (m_wait_for_field->GetBoolean())
2454*0321dbc8SOmar Emara         attach_info.SetIgnoreExisting(!m_include_existing_field->GetBoolean());
2455*0321dbc8SOmar Emara     } else {
2456*0321dbc8SOmar Emara       attach_info.SetProcessID(m_pid_field->GetInteger());
2457*0321dbc8SOmar Emara     }
2458*0321dbc8SOmar Emara     if (m_plugin_field->GetChoiceContent() != "<default>")
2459*0321dbc8SOmar Emara       attach_info.SetProcessPluginName(m_plugin_field->GetChoiceContent());
2460*0321dbc8SOmar Emara 
2461*0321dbc8SOmar Emara     return attach_info;
2462*0321dbc8SOmar Emara   }
2463*0321dbc8SOmar Emara 
2464*0321dbc8SOmar Emara   void Attach(Window &window) {
2465*0321dbc8SOmar Emara     ClearError();
2466*0321dbc8SOmar Emara 
2467*0321dbc8SOmar Emara     bool process_is_running = StopRunningProcess();
2468*0321dbc8SOmar Emara     if (process_is_running)
2469*0321dbc8SOmar Emara       return;
2470*0321dbc8SOmar Emara 
2471*0321dbc8SOmar Emara     Target *target = GetTarget();
2472*0321dbc8SOmar Emara     if (HasError())
2473*0321dbc8SOmar Emara       return;
2474*0321dbc8SOmar Emara 
2475*0321dbc8SOmar Emara     StreamString stream;
2476*0321dbc8SOmar Emara     ProcessAttachInfo attach_info = GetAttachInfo();
2477*0321dbc8SOmar Emara     Status status = target->Attach(attach_info, &stream);
2478*0321dbc8SOmar Emara 
2479*0321dbc8SOmar Emara     if (status.Fail()) {
2480*0321dbc8SOmar Emara       SetError(status.AsCString());
2481*0321dbc8SOmar Emara       return;
2482*0321dbc8SOmar Emara     }
2483*0321dbc8SOmar Emara 
2484*0321dbc8SOmar Emara     ProcessSP process_sp(target->GetProcessSP());
2485*0321dbc8SOmar Emara     if (!process_sp) {
2486*0321dbc8SOmar Emara       SetError("Attached sucessfully but target has no process.");
2487*0321dbc8SOmar Emara       return;
2488*0321dbc8SOmar Emara     }
2489*0321dbc8SOmar Emara 
2490*0321dbc8SOmar Emara     if (attach_info.GetContinueOnceAttached())
2491*0321dbc8SOmar Emara       process_sp->Resume();
2492*0321dbc8SOmar Emara 
2493*0321dbc8SOmar Emara     window.GetParent()->RemoveSubWindow(&window);
2494*0321dbc8SOmar Emara   }
2495*0321dbc8SOmar Emara 
2496*0321dbc8SOmar Emara protected:
2497*0321dbc8SOmar Emara   Debugger &m_debugger;
2498*0321dbc8SOmar Emara   WindowSP m_main_window_sp;
2499*0321dbc8SOmar Emara 
2500*0321dbc8SOmar Emara   ChoicesFieldDelegate *m_type_field;
2501*0321dbc8SOmar Emara   IntegerFieldDelegate *m_pid_field;
2502*0321dbc8SOmar Emara   TextFieldDelegate *m_name_field;
2503*0321dbc8SOmar Emara   BooleanFieldDelegate *m_continue_field;
2504*0321dbc8SOmar Emara   BooleanFieldDelegate *m_wait_for_field;
2505*0321dbc8SOmar Emara   BooleanFieldDelegate *m_include_existing_field;
2506*0321dbc8SOmar Emara   BooleanFieldDelegate *m_show_advanced_field;
2507*0321dbc8SOmar Emara   ChoicesFieldDelegate *m_plugin_field;
2508*0321dbc8SOmar Emara };
2509*0321dbc8SOmar Emara 
25107caa17caSRaphael Isemann class MenuDelegate {
25117caa17caSRaphael Isemann public:
25127caa17caSRaphael Isemann   virtual ~MenuDelegate() = default;
25137caa17caSRaphael Isemann 
25147caa17caSRaphael Isemann   virtual MenuActionResult MenuDelegateAction(Menu &menu) = 0;
25157caa17caSRaphael Isemann };
25167caa17caSRaphael Isemann 
25177caa17caSRaphael Isemann class Menu : public WindowDelegate {
25187caa17caSRaphael Isemann public:
25197caa17caSRaphael Isemann   enum class Type { Invalid, Bar, Item, Separator };
25207caa17caSRaphael Isemann 
25217caa17caSRaphael Isemann   // Menubar or separator constructor
25227caa17caSRaphael Isemann   Menu(Type type);
25237caa17caSRaphael Isemann 
25247caa17caSRaphael Isemann   // Menuitem constructor
25257caa17caSRaphael Isemann   Menu(const char *name, const char *key_name, int key_value,
25267caa17caSRaphael Isemann        uint64_t identifier);
25277caa17caSRaphael Isemann 
25287caa17caSRaphael Isemann   ~Menu() override = default;
25297caa17caSRaphael Isemann 
25307caa17caSRaphael Isemann   const MenuDelegateSP &GetDelegate() const { return m_delegate_sp; }
25317caa17caSRaphael Isemann 
25327caa17caSRaphael Isemann   void SetDelegate(const MenuDelegateSP &delegate_sp) {
25337caa17caSRaphael Isemann     m_delegate_sp = delegate_sp;
25347caa17caSRaphael Isemann   }
25357caa17caSRaphael Isemann 
25367caa17caSRaphael Isemann   void RecalculateNameLengths();
25377caa17caSRaphael Isemann 
25387caa17caSRaphael Isemann   void AddSubmenu(const MenuSP &menu_sp);
25397caa17caSRaphael Isemann 
25407caa17caSRaphael Isemann   int DrawAndRunMenu(Window &window);
25417caa17caSRaphael Isemann 
25427caa17caSRaphael Isemann   void DrawMenuTitle(Window &window, bool highlight);
25437caa17caSRaphael Isemann 
25447caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override;
25457caa17caSRaphael Isemann 
25467caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override;
25477caa17caSRaphael Isemann 
25487caa17caSRaphael Isemann   MenuActionResult ActionPrivate(Menu &menu) {
25497caa17caSRaphael Isemann     MenuActionResult result = MenuActionResult::NotHandled;
25507caa17caSRaphael Isemann     if (m_delegate_sp) {
25517caa17caSRaphael Isemann       result = m_delegate_sp->MenuDelegateAction(menu);
25527caa17caSRaphael Isemann       if (result != MenuActionResult::NotHandled)
25537caa17caSRaphael Isemann         return result;
25547caa17caSRaphael Isemann     } else if (m_parent) {
25557caa17caSRaphael Isemann       result = m_parent->ActionPrivate(menu);
25567caa17caSRaphael Isemann       if (result != MenuActionResult::NotHandled)
25577caa17caSRaphael Isemann         return result;
25587caa17caSRaphael Isemann     }
25597caa17caSRaphael Isemann     return m_canned_result;
25607caa17caSRaphael Isemann   }
25617caa17caSRaphael Isemann 
25627caa17caSRaphael Isemann   MenuActionResult Action() {
25637caa17caSRaphael Isemann     // Call the recursive action so it can try to handle it with the menu
25647caa17caSRaphael Isemann     // delegate, and if not, try our parent menu
25657caa17caSRaphael Isemann     return ActionPrivate(*this);
25667caa17caSRaphael Isemann   }
25677caa17caSRaphael Isemann 
25687caa17caSRaphael Isemann   void SetCannedResult(MenuActionResult result) { m_canned_result = result; }
25697caa17caSRaphael Isemann 
25707caa17caSRaphael Isemann   Menus &GetSubmenus() { return m_submenus; }
25717caa17caSRaphael Isemann 
25727caa17caSRaphael Isemann   const Menus &GetSubmenus() const { return m_submenus; }
25737caa17caSRaphael Isemann 
25747caa17caSRaphael Isemann   int GetSelectedSubmenuIndex() const { return m_selected; }
25757caa17caSRaphael Isemann 
25767caa17caSRaphael Isemann   void SetSelectedSubmenuIndex(int idx) { m_selected = idx; }
25777caa17caSRaphael Isemann 
25787caa17caSRaphael Isemann   Type GetType() const { return m_type; }
25797caa17caSRaphael Isemann 
25807caa17caSRaphael Isemann   int GetStartingColumn() const { return m_start_col; }
25817caa17caSRaphael Isemann 
25827caa17caSRaphael Isemann   void SetStartingColumn(int col) { m_start_col = col; }
25837caa17caSRaphael Isemann 
25847caa17caSRaphael Isemann   int GetKeyValue() const { return m_key_value; }
25857caa17caSRaphael Isemann 
25867caa17caSRaphael Isemann   std::string &GetName() { return m_name; }
25877caa17caSRaphael Isemann 
25887caa17caSRaphael Isemann   int GetDrawWidth() const {
25897caa17caSRaphael Isemann     return m_max_submenu_name_length + m_max_submenu_key_name_length + 8;
25907caa17caSRaphael Isemann   }
25917caa17caSRaphael Isemann 
25927caa17caSRaphael Isemann   uint64_t GetIdentifier() const { return m_identifier; }
25937caa17caSRaphael Isemann 
25947caa17caSRaphael Isemann   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
25957caa17caSRaphael Isemann 
25967caa17caSRaphael Isemann protected:
25977caa17caSRaphael Isemann   std::string m_name;
25987caa17caSRaphael Isemann   std::string m_key_name;
25997caa17caSRaphael Isemann   uint64_t m_identifier;
26007caa17caSRaphael Isemann   Type m_type;
26017caa17caSRaphael Isemann   int m_key_value;
26027caa17caSRaphael Isemann   int m_start_col;
26037caa17caSRaphael Isemann   int m_max_submenu_name_length;
26047caa17caSRaphael Isemann   int m_max_submenu_key_name_length;
26057caa17caSRaphael Isemann   int m_selected;
26067caa17caSRaphael Isemann   Menu *m_parent;
26077caa17caSRaphael Isemann   Menus m_submenus;
26087caa17caSRaphael Isemann   WindowSP m_menu_window_sp;
26097caa17caSRaphael Isemann   MenuActionResult m_canned_result;
26107caa17caSRaphael Isemann   MenuDelegateSP m_delegate_sp;
26117caa17caSRaphael Isemann };
26127caa17caSRaphael Isemann 
26137caa17caSRaphael Isemann // Menubar or separator constructor
26147caa17caSRaphael Isemann Menu::Menu(Type type)
26157caa17caSRaphael Isemann     : m_name(), m_key_name(), m_identifier(0), m_type(type), m_key_value(0),
26167caa17caSRaphael Isemann       m_start_col(0), m_max_submenu_name_length(0),
26177caa17caSRaphael Isemann       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
26187caa17caSRaphael Isemann       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
26197caa17caSRaphael Isemann       m_delegate_sp() {}
26207caa17caSRaphael Isemann 
26217caa17caSRaphael Isemann // Menuitem constructor
26227caa17caSRaphael Isemann Menu::Menu(const char *name, const char *key_name, int key_value,
26237caa17caSRaphael Isemann            uint64_t identifier)
26247caa17caSRaphael Isemann     : m_name(), m_key_name(), m_identifier(identifier), m_type(Type::Invalid),
26257caa17caSRaphael Isemann       m_key_value(key_value), m_start_col(0), m_max_submenu_name_length(0),
26267caa17caSRaphael Isemann       m_max_submenu_key_name_length(0), m_selected(0), m_parent(nullptr),
26277caa17caSRaphael Isemann       m_submenus(), m_canned_result(MenuActionResult::NotHandled),
26287caa17caSRaphael Isemann       m_delegate_sp() {
26297caa17caSRaphael Isemann   if (name && name[0]) {
26307caa17caSRaphael Isemann     m_name = name;
26317caa17caSRaphael Isemann     m_type = Type::Item;
26327caa17caSRaphael Isemann     if (key_name && key_name[0])
26337caa17caSRaphael Isemann       m_key_name = key_name;
26347caa17caSRaphael Isemann   } else {
26357caa17caSRaphael Isemann     m_type = Type::Separator;
26367caa17caSRaphael Isemann   }
26377caa17caSRaphael Isemann }
26387caa17caSRaphael Isemann 
26397caa17caSRaphael Isemann void Menu::RecalculateNameLengths() {
26407caa17caSRaphael Isemann   m_max_submenu_name_length = 0;
26417caa17caSRaphael Isemann   m_max_submenu_key_name_length = 0;
26427caa17caSRaphael Isemann   Menus &submenus = GetSubmenus();
26437caa17caSRaphael Isemann   const size_t num_submenus = submenus.size();
26447caa17caSRaphael Isemann   for (size_t i = 0; i < num_submenus; ++i) {
26457caa17caSRaphael Isemann     Menu *submenu = submenus[i].get();
26467caa17caSRaphael Isemann     if (static_cast<size_t>(m_max_submenu_name_length) < submenu->m_name.size())
26477caa17caSRaphael Isemann       m_max_submenu_name_length = submenu->m_name.size();
26487caa17caSRaphael Isemann     if (static_cast<size_t>(m_max_submenu_key_name_length) <
26497caa17caSRaphael Isemann         submenu->m_key_name.size())
26507caa17caSRaphael Isemann       m_max_submenu_key_name_length = submenu->m_key_name.size();
26517caa17caSRaphael Isemann   }
26527caa17caSRaphael Isemann }
26537caa17caSRaphael Isemann 
26547caa17caSRaphael Isemann void Menu::AddSubmenu(const MenuSP &menu_sp) {
26557caa17caSRaphael Isemann   menu_sp->m_parent = this;
26567caa17caSRaphael Isemann   if (static_cast<size_t>(m_max_submenu_name_length) < menu_sp->m_name.size())
26577caa17caSRaphael Isemann     m_max_submenu_name_length = menu_sp->m_name.size();
26587caa17caSRaphael Isemann   if (static_cast<size_t>(m_max_submenu_key_name_length) <
26597caa17caSRaphael Isemann       menu_sp->m_key_name.size())
26607caa17caSRaphael Isemann     m_max_submenu_key_name_length = menu_sp->m_key_name.size();
26617caa17caSRaphael Isemann   m_submenus.push_back(menu_sp);
26627caa17caSRaphael Isemann }
26637caa17caSRaphael Isemann 
26647caa17caSRaphael Isemann void Menu::DrawMenuTitle(Window &window, bool highlight) {
26657caa17caSRaphael Isemann   if (m_type == Type::Separator) {
26667caa17caSRaphael Isemann     window.MoveCursor(0, window.GetCursorY());
26677caa17caSRaphael Isemann     window.PutChar(ACS_LTEE);
26687caa17caSRaphael Isemann     int width = window.GetWidth();
26697caa17caSRaphael Isemann     if (width > 2) {
26707caa17caSRaphael Isemann       width -= 2;
26717caa17caSRaphael Isemann       for (int i = 0; i < width; ++i)
26727caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
26737caa17caSRaphael Isemann     }
26747caa17caSRaphael Isemann     window.PutChar(ACS_RTEE);
26757caa17caSRaphael Isemann   } else {
26767caa17caSRaphael Isemann     const int shortcut_key = m_key_value;
26777caa17caSRaphael Isemann     bool underlined_shortcut = false;
2678daa1c6d9SLuboš Luňák     const attr_t highlight_attr = A_REVERSE;
26797caa17caSRaphael Isemann     if (highlight)
2680daa1c6d9SLuboš Luňák       window.AttributeOn(highlight_attr);
2681f5eaa2afSRaphael Isemann     if (llvm::isPrint(shortcut_key)) {
26827caa17caSRaphael Isemann       size_t lower_pos = m_name.find(tolower(shortcut_key));
26837caa17caSRaphael Isemann       size_t upper_pos = m_name.find(toupper(shortcut_key));
26847caa17caSRaphael Isemann       const char *name = m_name.c_str();
26857caa17caSRaphael Isemann       size_t pos = std::min<size_t>(lower_pos, upper_pos);
26867caa17caSRaphael Isemann       if (pos != std::string::npos) {
26877caa17caSRaphael Isemann         underlined_shortcut = true;
26887caa17caSRaphael Isemann         if (pos > 0) {
26897caa17caSRaphael Isemann           window.PutCString(name, pos);
26907caa17caSRaphael Isemann           name += pos;
26917caa17caSRaphael Isemann         }
26927caa17caSRaphael Isemann         const attr_t shortcut_attr = A_UNDERLINE | A_BOLD;
26937caa17caSRaphael Isemann         window.AttributeOn(shortcut_attr);
26947caa17caSRaphael Isemann         window.PutChar(name[0]);
26957caa17caSRaphael Isemann         window.AttributeOff(shortcut_attr);
26967caa17caSRaphael Isemann         name++;
26977caa17caSRaphael Isemann         if (name[0])
26987caa17caSRaphael Isemann           window.PutCString(name);
26997caa17caSRaphael Isemann       }
27007caa17caSRaphael Isemann     }
27017caa17caSRaphael Isemann 
27027caa17caSRaphael Isemann     if (!underlined_shortcut) {
27037caa17caSRaphael Isemann       window.PutCString(m_name.c_str());
27047caa17caSRaphael Isemann     }
27057caa17caSRaphael Isemann 
27067caa17caSRaphael Isemann     if (highlight)
2707daa1c6d9SLuboš Luňák       window.AttributeOff(highlight_attr);
27087caa17caSRaphael Isemann 
27097caa17caSRaphael Isemann     if (m_key_name.empty()) {
2710f5eaa2afSRaphael Isemann       if (!underlined_shortcut && llvm::isPrint(m_key_value)) {
27114a8e4b5cSLuboš Luňák         window.AttributeOn(COLOR_PAIR(MagentaOnWhite));
27127caa17caSRaphael Isemann         window.Printf(" (%c)", m_key_value);
27134a8e4b5cSLuboš Luňák         window.AttributeOff(COLOR_PAIR(MagentaOnWhite));
27147caa17caSRaphael Isemann       }
27157caa17caSRaphael Isemann     } else {
27164a8e4b5cSLuboš Luňák       window.AttributeOn(COLOR_PAIR(MagentaOnWhite));
27177caa17caSRaphael Isemann       window.Printf(" (%s)", m_key_name.c_str());
27184a8e4b5cSLuboš Luňák       window.AttributeOff(COLOR_PAIR(MagentaOnWhite));
27197caa17caSRaphael Isemann     }
27207caa17caSRaphael Isemann   }
27217caa17caSRaphael Isemann }
27227caa17caSRaphael Isemann 
27237caa17caSRaphael Isemann bool Menu::WindowDelegateDraw(Window &window, bool force) {
27247caa17caSRaphael Isemann   Menus &submenus = GetSubmenus();
27257caa17caSRaphael Isemann   const size_t num_submenus = submenus.size();
27267caa17caSRaphael Isemann   const int selected_idx = GetSelectedSubmenuIndex();
27277caa17caSRaphael Isemann   Menu::Type menu_type = GetType();
27287caa17caSRaphael Isemann   switch (menu_type) {
27297caa17caSRaphael Isemann   case Menu::Type::Bar: {
27304a8e4b5cSLuboš Luňák     window.SetBackground(BlackOnWhite);
27317caa17caSRaphael Isemann     window.MoveCursor(0, 0);
27327caa17caSRaphael Isemann     for (size_t i = 0; i < num_submenus; ++i) {
27337caa17caSRaphael Isemann       Menu *menu = submenus[i].get();
27347caa17caSRaphael Isemann       if (i > 0)
27357caa17caSRaphael Isemann         window.PutChar(' ');
27367caa17caSRaphael Isemann       menu->SetStartingColumn(window.GetCursorX());
27377caa17caSRaphael Isemann       window.PutCString("| ");
27387caa17caSRaphael Isemann       menu->DrawMenuTitle(window, false);
27397caa17caSRaphael Isemann     }
27407caa17caSRaphael Isemann     window.PutCString(" |");
27417caa17caSRaphael Isemann   } break;
27427caa17caSRaphael Isemann 
27437caa17caSRaphael Isemann   case Menu::Type::Item: {
27447caa17caSRaphael Isemann     int y = 1;
27457caa17caSRaphael Isemann     int x = 3;
27467caa17caSRaphael Isemann     // Draw the menu
27477caa17caSRaphael Isemann     int cursor_x = 0;
27487caa17caSRaphael Isemann     int cursor_y = 0;
27497caa17caSRaphael Isemann     window.Erase();
27504a8e4b5cSLuboš Luňák     window.SetBackground(BlackOnWhite);
27517caa17caSRaphael Isemann     window.Box();
27527caa17caSRaphael Isemann     for (size_t i = 0; i < num_submenus; ++i) {
27537caa17caSRaphael Isemann       const bool is_selected = (i == static_cast<size_t>(selected_idx));
27547caa17caSRaphael Isemann       window.MoveCursor(x, y + i);
27557caa17caSRaphael Isemann       if (is_selected) {
27567caa17caSRaphael Isemann         // Remember where we want the cursor to be
27577caa17caSRaphael Isemann         cursor_x = x - 1;
27587caa17caSRaphael Isemann         cursor_y = y + i;
27597caa17caSRaphael Isemann       }
27607caa17caSRaphael Isemann       submenus[i]->DrawMenuTitle(window, is_selected);
27617caa17caSRaphael Isemann     }
27627caa17caSRaphael Isemann     window.MoveCursor(cursor_x, cursor_y);
27637caa17caSRaphael Isemann   } break;
27647caa17caSRaphael Isemann 
27657caa17caSRaphael Isemann   default:
27667caa17caSRaphael Isemann   case Menu::Type::Separator:
27677caa17caSRaphael Isemann     break;
27687caa17caSRaphael Isemann   }
27697caa17caSRaphael Isemann   return true; // Drawing handled...
27707caa17caSRaphael Isemann }
27717caa17caSRaphael Isemann 
27727caa17caSRaphael Isemann HandleCharResult Menu::WindowDelegateHandleChar(Window &window, int key) {
27737caa17caSRaphael Isemann   HandleCharResult result = eKeyNotHandled;
27747caa17caSRaphael Isemann 
27757caa17caSRaphael Isemann   Menus &submenus = GetSubmenus();
27767caa17caSRaphael Isemann   const size_t num_submenus = submenus.size();
27777caa17caSRaphael Isemann   const int selected_idx = GetSelectedSubmenuIndex();
27787caa17caSRaphael Isemann   Menu::Type menu_type = GetType();
27797caa17caSRaphael Isemann   if (menu_type == Menu::Type::Bar) {
27807caa17caSRaphael Isemann     MenuSP run_menu_sp;
27817caa17caSRaphael Isemann     switch (key) {
27827caa17caSRaphael Isemann     case KEY_DOWN:
27837caa17caSRaphael Isemann     case KEY_UP:
27847caa17caSRaphael Isemann       // Show last menu or first menu
27857caa17caSRaphael Isemann       if (selected_idx < static_cast<int>(num_submenus))
27867caa17caSRaphael Isemann         run_menu_sp = submenus[selected_idx];
27877caa17caSRaphael Isemann       else if (!submenus.empty())
27887caa17caSRaphael Isemann         run_menu_sp = submenus.front();
27897caa17caSRaphael Isemann       result = eKeyHandled;
27907caa17caSRaphael Isemann       break;
27917caa17caSRaphael Isemann 
27927caa17caSRaphael Isemann     case KEY_RIGHT:
27937caa17caSRaphael Isemann       ++m_selected;
27947caa17caSRaphael Isemann       if (m_selected >= static_cast<int>(num_submenus))
27957caa17caSRaphael Isemann         m_selected = 0;
27967caa17caSRaphael Isemann       if (m_selected < static_cast<int>(num_submenus))
27977caa17caSRaphael Isemann         run_menu_sp = submenus[m_selected];
27987caa17caSRaphael Isemann       else if (!submenus.empty())
27997caa17caSRaphael Isemann         run_menu_sp = submenus.front();
28007caa17caSRaphael Isemann       result = eKeyHandled;
28017caa17caSRaphael Isemann       break;
28027caa17caSRaphael Isemann 
28037caa17caSRaphael Isemann     case KEY_LEFT:
28047caa17caSRaphael Isemann       --m_selected;
28057caa17caSRaphael Isemann       if (m_selected < 0)
28067caa17caSRaphael Isemann         m_selected = num_submenus - 1;
28077caa17caSRaphael Isemann       if (m_selected < static_cast<int>(num_submenus))
28087caa17caSRaphael Isemann         run_menu_sp = submenus[m_selected];
28097caa17caSRaphael Isemann       else if (!submenus.empty())
28107caa17caSRaphael Isemann         run_menu_sp = submenus.front();
28117caa17caSRaphael Isemann       result = eKeyHandled;
28127caa17caSRaphael Isemann       break;
28137caa17caSRaphael Isemann 
28147caa17caSRaphael Isemann     default:
28157caa17caSRaphael Isemann       for (size_t i = 0; i < num_submenus; ++i) {
28167caa17caSRaphael Isemann         if (submenus[i]->GetKeyValue() == key) {
28177caa17caSRaphael Isemann           SetSelectedSubmenuIndex(i);
28187caa17caSRaphael Isemann           run_menu_sp = submenus[i];
28197caa17caSRaphael Isemann           result = eKeyHandled;
28207caa17caSRaphael Isemann           break;
28217caa17caSRaphael Isemann         }
28227caa17caSRaphael Isemann       }
28237caa17caSRaphael Isemann       break;
28247caa17caSRaphael Isemann     }
28257caa17caSRaphael Isemann 
28267caa17caSRaphael Isemann     if (run_menu_sp) {
28277caa17caSRaphael Isemann       // Run the action on this menu in case we need to populate the menu with
28287caa17caSRaphael Isemann       // dynamic content and also in case check marks, and any other menu
28297caa17caSRaphael Isemann       // decorations need to be calculated
28307caa17caSRaphael Isemann       if (run_menu_sp->Action() == MenuActionResult::Quit)
28317caa17caSRaphael Isemann         return eQuitApplication;
28327caa17caSRaphael Isemann 
28337caa17caSRaphael Isemann       Rect menu_bounds;
28347caa17caSRaphael Isemann       menu_bounds.origin.x = run_menu_sp->GetStartingColumn();
28357caa17caSRaphael Isemann       menu_bounds.origin.y = 1;
28367caa17caSRaphael Isemann       menu_bounds.size.width = run_menu_sp->GetDrawWidth();
28377caa17caSRaphael Isemann       menu_bounds.size.height = run_menu_sp->GetSubmenus().size() + 2;
28387caa17caSRaphael Isemann       if (m_menu_window_sp)
28397caa17caSRaphael Isemann         window.GetParent()->RemoveSubWindow(m_menu_window_sp.get());
28407caa17caSRaphael Isemann 
28417caa17caSRaphael Isemann       m_menu_window_sp = window.GetParent()->CreateSubWindow(
28427caa17caSRaphael Isemann           run_menu_sp->GetName().c_str(), menu_bounds, true);
28437caa17caSRaphael Isemann       m_menu_window_sp->SetDelegate(run_menu_sp);
28447caa17caSRaphael Isemann     }
28457caa17caSRaphael Isemann   } else if (menu_type == Menu::Type::Item) {
28467caa17caSRaphael Isemann     switch (key) {
28477caa17caSRaphael Isemann     case KEY_DOWN:
28487caa17caSRaphael Isemann       if (m_submenus.size() > 1) {
28497caa17caSRaphael Isemann         const int start_select = m_selected;
28507caa17caSRaphael Isemann         while (++m_selected != start_select) {
28517caa17caSRaphael Isemann           if (static_cast<size_t>(m_selected) >= num_submenus)
28527caa17caSRaphael Isemann             m_selected = 0;
28537caa17caSRaphael Isemann           if (m_submenus[m_selected]->GetType() == Type::Separator)
28547caa17caSRaphael Isemann             continue;
28557caa17caSRaphael Isemann           else
28567caa17caSRaphael Isemann             break;
28577caa17caSRaphael Isemann         }
28587caa17caSRaphael Isemann         return eKeyHandled;
28597caa17caSRaphael Isemann       }
28607caa17caSRaphael Isemann       break;
28617caa17caSRaphael Isemann 
28627caa17caSRaphael Isemann     case KEY_UP:
28637caa17caSRaphael Isemann       if (m_submenus.size() > 1) {
28647caa17caSRaphael Isemann         const int start_select = m_selected;
28657caa17caSRaphael Isemann         while (--m_selected != start_select) {
28667caa17caSRaphael Isemann           if (m_selected < static_cast<int>(0))
28677caa17caSRaphael Isemann             m_selected = num_submenus - 1;
28687caa17caSRaphael Isemann           if (m_submenus[m_selected]->GetType() == Type::Separator)
28697caa17caSRaphael Isemann             continue;
28707caa17caSRaphael Isemann           else
28717caa17caSRaphael Isemann             break;
28727caa17caSRaphael Isemann         }
28737caa17caSRaphael Isemann         return eKeyHandled;
28747caa17caSRaphael Isemann       }
28757caa17caSRaphael Isemann       break;
28767caa17caSRaphael Isemann 
28777caa17caSRaphael Isemann     case KEY_RETURN:
28787caa17caSRaphael Isemann       if (static_cast<size_t>(selected_idx) < num_submenus) {
28797caa17caSRaphael Isemann         if (submenus[selected_idx]->Action() == MenuActionResult::Quit)
28807caa17caSRaphael Isemann           return eQuitApplication;
28817caa17caSRaphael Isemann         window.GetParent()->RemoveSubWindow(&window);
28827caa17caSRaphael Isemann         return eKeyHandled;
28837caa17caSRaphael Isemann       }
28847caa17caSRaphael Isemann       break;
28857caa17caSRaphael Isemann 
28867caa17caSRaphael Isemann     case KEY_ESCAPE: // Beware: pressing escape key has 1 to 2 second delay in
28877caa17caSRaphael Isemann                      // case other chars are entered for escaped sequences
28887caa17caSRaphael Isemann       window.GetParent()->RemoveSubWindow(&window);
28897caa17caSRaphael Isemann       return eKeyHandled;
28907caa17caSRaphael Isemann 
28917caa17caSRaphael Isemann     default:
28927caa17caSRaphael Isemann       for (size_t i = 0; i < num_submenus; ++i) {
28937caa17caSRaphael Isemann         Menu *menu = submenus[i].get();
28947caa17caSRaphael Isemann         if (menu->GetKeyValue() == key) {
28957caa17caSRaphael Isemann           SetSelectedSubmenuIndex(i);
28967caa17caSRaphael Isemann           window.GetParent()->RemoveSubWindow(&window);
28977caa17caSRaphael Isemann           if (menu->Action() == MenuActionResult::Quit)
28987caa17caSRaphael Isemann             return eQuitApplication;
28997caa17caSRaphael Isemann           return eKeyHandled;
29007caa17caSRaphael Isemann         }
29017caa17caSRaphael Isemann       }
29027caa17caSRaphael Isemann       break;
29037caa17caSRaphael Isemann     }
29047caa17caSRaphael Isemann   } else if (menu_type == Menu::Type::Separator) {
29057caa17caSRaphael Isemann   }
29067caa17caSRaphael Isemann   return result;
29077caa17caSRaphael Isemann }
29087caa17caSRaphael Isemann 
29097caa17caSRaphael Isemann class Application {
29107caa17caSRaphael Isemann public:
29117caa17caSRaphael Isemann   Application(FILE *in, FILE *out)
29127caa17caSRaphael Isemann       : m_window_sp(), m_screen(nullptr), m_in(in), m_out(out) {}
29137caa17caSRaphael Isemann 
29147caa17caSRaphael Isemann   ~Application() {
29157caa17caSRaphael Isemann     m_window_delegates.clear();
29167caa17caSRaphael Isemann     m_window_sp.reset();
29177caa17caSRaphael Isemann     if (m_screen) {
29187caa17caSRaphael Isemann       ::delscreen(m_screen);
29197caa17caSRaphael Isemann       m_screen = nullptr;
29207caa17caSRaphael Isemann     }
29217caa17caSRaphael Isemann   }
29227caa17caSRaphael Isemann 
29237caa17caSRaphael Isemann   void Initialize() {
29247caa17caSRaphael Isemann     m_screen = ::newterm(nullptr, m_out, m_in);
29257caa17caSRaphael Isemann     ::start_color();
29267caa17caSRaphael Isemann     ::curs_set(0);
29277caa17caSRaphael Isemann     ::noecho();
29287caa17caSRaphael Isemann     ::keypad(stdscr, TRUE);
29297caa17caSRaphael Isemann   }
29307caa17caSRaphael Isemann 
29317caa17caSRaphael Isemann   void Terminate() { ::endwin(); }
29327caa17caSRaphael Isemann 
29337caa17caSRaphael Isemann   void Run(Debugger &debugger) {
29347caa17caSRaphael Isemann     bool done = false;
29357caa17caSRaphael Isemann     int delay_in_tenths_of_a_second = 1;
29367caa17caSRaphael Isemann 
29377caa17caSRaphael Isemann     // Alas the threading model in curses is a bit lame so we need to resort to
29387caa17caSRaphael Isemann     // polling every 0.5 seconds. We could poll for stdin ourselves and then
29397caa17caSRaphael Isemann     // pass the keys down but then we need to translate all of the escape
29407caa17caSRaphael Isemann     // sequences ourselves. So we resort to polling for input because we need
29417caa17caSRaphael Isemann     // to receive async process events while in this loop.
29427caa17caSRaphael Isemann 
29437caa17caSRaphael Isemann     halfdelay(delay_in_tenths_of_a_second); // Poll using some number of tenths
29447caa17caSRaphael Isemann                                             // of seconds seconds when calling
29457caa17caSRaphael Isemann                                             // Window::GetChar()
29467caa17caSRaphael Isemann 
29477caa17caSRaphael Isemann     ListenerSP listener_sp(
29487caa17caSRaphael Isemann         Listener::MakeListener("lldb.IOHandler.curses.Application"));
29497caa17caSRaphael Isemann     ConstString broadcaster_class_process(Process::GetStaticBroadcasterClass());
29507caa17caSRaphael Isemann     debugger.EnableForwardEvents(listener_sp);
29517caa17caSRaphael Isemann 
29522f1b24b7SLuboš Luňák     m_update_screen = true;
29537caa17caSRaphael Isemann #if defined(__APPLE__)
29547caa17caSRaphael Isemann     std::deque<int> escape_chars;
29557caa17caSRaphael Isemann #endif
29567caa17caSRaphael Isemann 
29577caa17caSRaphael Isemann     while (!done) {
29582f1b24b7SLuboš Luňák       if (m_update_screen) {
29597caa17caSRaphael Isemann         m_window_sp->Draw(false);
29607caa17caSRaphael Isemann         // All windows should be calling Window::DeferredRefresh() instead of
29617caa17caSRaphael Isemann         // Window::Refresh() so we can do a single update and avoid any screen
29627caa17caSRaphael Isemann         // blinking
29637caa17caSRaphael Isemann         update_panels();
29647caa17caSRaphael Isemann 
29657caa17caSRaphael Isemann         // Cursor hiding isn't working on MacOSX, so hide it in the top left
29667caa17caSRaphael Isemann         // corner
29677caa17caSRaphael Isemann         m_window_sp->MoveCursor(0, 0);
29687caa17caSRaphael Isemann 
29697caa17caSRaphael Isemann         doupdate();
29702f1b24b7SLuboš Luňák         m_update_screen = false;
29717caa17caSRaphael Isemann       }
29727caa17caSRaphael Isemann 
29737caa17caSRaphael Isemann #if defined(__APPLE__)
29747caa17caSRaphael Isemann       // Terminal.app doesn't map its function keys correctly, F1-F4 default
29757caa17caSRaphael Isemann       // to: \033OP, \033OQ, \033OR, \033OS, so lets take care of this here if
29767caa17caSRaphael Isemann       // possible
29777caa17caSRaphael Isemann       int ch;
29787caa17caSRaphael Isemann       if (escape_chars.empty())
29797caa17caSRaphael Isemann         ch = m_window_sp->GetChar();
29807caa17caSRaphael Isemann       else {
29817caa17caSRaphael Isemann         ch = escape_chars.front();
29827caa17caSRaphael Isemann         escape_chars.pop_front();
29837caa17caSRaphael Isemann       }
29847caa17caSRaphael Isemann       if (ch == KEY_ESCAPE) {
29857caa17caSRaphael Isemann         int ch2 = m_window_sp->GetChar();
29867caa17caSRaphael Isemann         if (ch2 == 'O') {
29877caa17caSRaphael Isemann           int ch3 = m_window_sp->GetChar();
29887caa17caSRaphael Isemann           switch (ch3) {
29897caa17caSRaphael Isemann           case 'P':
29907caa17caSRaphael Isemann             ch = KEY_F(1);
29917caa17caSRaphael Isemann             break;
29927caa17caSRaphael Isemann           case 'Q':
29937caa17caSRaphael Isemann             ch = KEY_F(2);
29947caa17caSRaphael Isemann             break;
29957caa17caSRaphael Isemann           case 'R':
29967caa17caSRaphael Isemann             ch = KEY_F(3);
29977caa17caSRaphael Isemann             break;
29987caa17caSRaphael Isemann           case 'S':
29997caa17caSRaphael Isemann             ch = KEY_F(4);
30007caa17caSRaphael Isemann             break;
30017caa17caSRaphael Isemann           default:
30027caa17caSRaphael Isemann             escape_chars.push_back(ch2);
30037caa17caSRaphael Isemann             if (ch3 != -1)
30047caa17caSRaphael Isemann               escape_chars.push_back(ch3);
30057caa17caSRaphael Isemann             break;
30067caa17caSRaphael Isemann           }
30077caa17caSRaphael Isemann         } else if (ch2 != -1)
30087caa17caSRaphael Isemann           escape_chars.push_back(ch2);
30097caa17caSRaphael Isemann       }
30107caa17caSRaphael Isemann #else
30117caa17caSRaphael Isemann       int ch = m_window_sp->GetChar();
30127caa17caSRaphael Isemann 
30137caa17caSRaphael Isemann #endif
30147caa17caSRaphael Isemann       if (ch == -1) {
30157caa17caSRaphael Isemann         if (feof(m_in) || ferror(m_in)) {
30167caa17caSRaphael Isemann           done = true;
30177caa17caSRaphael Isemann         } else {
30187caa17caSRaphael Isemann           // Just a timeout from using halfdelay(), check for events
30197caa17caSRaphael Isemann           EventSP event_sp;
30207caa17caSRaphael Isemann           while (listener_sp->PeekAtNextEvent()) {
30217caa17caSRaphael Isemann             listener_sp->GetEvent(event_sp, std::chrono::seconds(0));
30227caa17caSRaphael Isemann 
30237caa17caSRaphael Isemann             if (event_sp) {
30247caa17caSRaphael Isemann               Broadcaster *broadcaster = event_sp->GetBroadcaster();
30257caa17caSRaphael Isemann               if (broadcaster) {
30267caa17caSRaphael Isemann                 // uint32_t event_type = event_sp->GetType();
30277caa17caSRaphael Isemann                 ConstString broadcaster_class(
30287caa17caSRaphael Isemann                     broadcaster->GetBroadcasterClass());
30297caa17caSRaphael Isemann                 if (broadcaster_class == broadcaster_class_process) {
30302f1b24b7SLuboš Luňák                   m_update_screen = true;
30317caa17caSRaphael Isemann                   continue; // Don't get any key, just update our view
30327caa17caSRaphael Isemann                 }
30337caa17caSRaphael Isemann               }
30347caa17caSRaphael Isemann             }
30357caa17caSRaphael Isemann           }
30367caa17caSRaphael Isemann         }
30377caa17caSRaphael Isemann       } else {
30387caa17caSRaphael Isemann         HandleCharResult key_result = m_window_sp->HandleChar(ch);
30397caa17caSRaphael Isemann         switch (key_result) {
30407caa17caSRaphael Isemann         case eKeyHandled:
30412f1b24b7SLuboš Luňák           m_update_screen = true;
30427caa17caSRaphael Isemann           break;
30437caa17caSRaphael Isemann         case eKeyNotHandled:
30448739445eSLuboš Luňák           if (ch == 12) { // Ctrl+L, force full redraw
30458739445eSLuboš Luňák             redrawwin(m_window_sp->get());
30462f1b24b7SLuboš Luňák             m_update_screen = true;
30478739445eSLuboš Luňák           }
30487caa17caSRaphael Isemann           break;
30497caa17caSRaphael Isemann         case eQuitApplication:
30507caa17caSRaphael Isemann           done = true;
30517caa17caSRaphael Isemann           break;
30527caa17caSRaphael Isemann         }
30537caa17caSRaphael Isemann       }
30547caa17caSRaphael Isemann     }
30557caa17caSRaphael Isemann 
30567caa17caSRaphael Isemann     debugger.CancelForwardEvents(listener_sp);
30577caa17caSRaphael Isemann   }
30587caa17caSRaphael Isemann 
30597caa17caSRaphael Isemann   WindowSP &GetMainWindow() {
30607caa17caSRaphael Isemann     if (!m_window_sp)
30617caa17caSRaphael Isemann       m_window_sp = std::make_shared<Window>("main", stdscr, false);
30627caa17caSRaphael Isemann     return m_window_sp;
30637caa17caSRaphael Isemann   }
30647caa17caSRaphael Isemann 
30652f1b24b7SLuboš Luňák   void TerminalSizeChanged() {
30662f1b24b7SLuboš Luňák     ::endwin();
30672f1b24b7SLuboš Luňák     ::refresh();
30682f1b24b7SLuboš Luňák     Rect content_bounds = m_window_sp->GetFrame();
30692f1b24b7SLuboš Luňák     m_window_sp->SetBounds(content_bounds);
30702f1b24b7SLuboš Luňák     if (WindowSP menubar_window_sp = m_window_sp->FindSubWindow("Menubar"))
30712f1b24b7SLuboš Luňák       menubar_window_sp->SetBounds(content_bounds.MakeMenuBar());
30722f1b24b7SLuboš Luňák     if (WindowSP status_window_sp = m_window_sp->FindSubWindow("Status"))
30732f1b24b7SLuboš Luňák       status_window_sp->SetBounds(content_bounds.MakeStatusBar());
30742f1b24b7SLuboš Luňák 
30752f1b24b7SLuboš Luňák     WindowSP source_window_sp = m_window_sp->FindSubWindow("Source");
30762f1b24b7SLuboš Luňák     WindowSP variables_window_sp = m_window_sp->FindSubWindow("Variables");
30772f1b24b7SLuboš Luňák     WindowSP registers_window_sp = m_window_sp->FindSubWindow("Registers");
30782f1b24b7SLuboš Luňák     WindowSP threads_window_sp = m_window_sp->FindSubWindow("Threads");
30792f1b24b7SLuboš Luňák 
30802f1b24b7SLuboš Luňák     Rect threads_bounds;
30812f1b24b7SLuboš Luňák     Rect source_variables_bounds;
30822f1b24b7SLuboš Luňák     content_bounds.VerticalSplitPercentage(0.80, source_variables_bounds,
30832f1b24b7SLuboš Luňák                                            threads_bounds);
30842f1b24b7SLuboš Luňák     if (threads_window_sp)
30852f1b24b7SLuboš Luňák       threads_window_sp->SetBounds(threads_bounds);
30862f1b24b7SLuboš Luňák     else
30872f1b24b7SLuboš Luňák       source_variables_bounds = content_bounds;
30882f1b24b7SLuboš Luňák 
30892f1b24b7SLuboš Luňák     Rect source_bounds;
30902f1b24b7SLuboš Luňák     Rect variables_registers_bounds;
30912f1b24b7SLuboš Luňák     source_variables_bounds.HorizontalSplitPercentage(
30922f1b24b7SLuboš Luňák         0.70, source_bounds, variables_registers_bounds);
30932f1b24b7SLuboš Luňák     if (variables_window_sp || registers_window_sp) {
30942f1b24b7SLuboš Luňák       if (variables_window_sp && registers_window_sp) {
30952f1b24b7SLuboš Luňák         Rect variables_bounds;
30962f1b24b7SLuboš Luňák         Rect registers_bounds;
30972f1b24b7SLuboš Luňák         variables_registers_bounds.VerticalSplitPercentage(
30982f1b24b7SLuboš Luňák             0.50, variables_bounds, registers_bounds);
30992f1b24b7SLuboš Luňák         variables_window_sp->SetBounds(variables_bounds);
31002f1b24b7SLuboš Luňák         registers_window_sp->SetBounds(registers_bounds);
31012f1b24b7SLuboš Luňák       } else if (variables_window_sp) {
31022f1b24b7SLuboš Luňák         variables_window_sp->SetBounds(variables_registers_bounds);
31032f1b24b7SLuboš Luňák       } else {
31042f1b24b7SLuboš Luňák         registers_window_sp->SetBounds(variables_registers_bounds);
31052f1b24b7SLuboš Luňák       }
31062f1b24b7SLuboš Luňák     } else {
31072f1b24b7SLuboš Luňák       source_bounds = source_variables_bounds;
31082f1b24b7SLuboš Luňák     }
31092f1b24b7SLuboš Luňák 
31102f1b24b7SLuboš Luňák     source_window_sp->SetBounds(source_bounds);
31112f1b24b7SLuboš Luňák 
31122f1b24b7SLuboš Luňák     touchwin(stdscr);
31132f1b24b7SLuboš Luňák     redrawwin(m_window_sp->get());
31142f1b24b7SLuboš Luňák     m_update_screen = true;
31152f1b24b7SLuboš Luňák   }
31162f1b24b7SLuboš Luňák 
31177caa17caSRaphael Isemann protected:
31187caa17caSRaphael Isemann   WindowSP m_window_sp;
31197caa17caSRaphael Isemann   WindowDelegates m_window_delegates;
31207caa17caSRaphael Isemann   SCREEN *m_screen;
31217caa17caSRaphael Isemann   FILE *m_in;
31227caa17caSRaphael Isemann   FILE *m_out;
31232f1b24b7SLuboš Luňák   bool m_update_screen = false;
31247caa17caSRaphael Isemann };
31257caa17caSRaphael Isemann 
31267caa17caSRaphael Isemann } // namespace curses
31277caa17caSRaphael Isemann 
31287caa17caSRaphael Isemann using namespace curses;
31297caa17caSRaphael Isemann 
31307caa17caSRaphael Isemann struct Row {
31314631afdeSRaphael Isemann   ValueObjectUpdater value;
31327caa17caSRaphael Isemann   Row *parent;
31337caa17caSRaphael Isemann   // The process stop ID when the children were calculated.
3134f7e04b71SJordan Rupprecht   uint32_t children_stop_id = 0;
3135f7e04b71SJordan Rupprecht   int row_idx = 0;
3136f7e04b71SJordan Rupprecht   int x = 1;
3137f7e04b71SJordan Rupprecht   int y = 1;
31387caa17caSRaphael Isemann   bool might_have_children;
3139f7e04b71SJordan Rupprecht   bool expanded = false;
3140f7e04b71SJordan Rupprecht   bool calculated_children = false;
31417caa17caSRaphael Isemann   std::vector<Row> children;
31427caa17caSRaphael Isemann 
31437caa17caSRaphael Isemann   Row(const ValueObjectSP &v, Row *p)
31444631afdeSRaphael Isemann       : value(v), parent(p),
3145f7e04b71SJordan Rupprecht         might_have_children(v ? v->MightHaveChildren() : false) {}
31467caa17caSRaphael Isemann 
31477caa17caSRaphael Isemann   size_t GetDepth() const {
31487caa17caSRaphael Isemann     if (parent)
31497caa17caSRaphael Isemann       return 1 + parent->GetDepth();
31507caa17caSRaphael Isemann     return 0;
31517caa17caSRaphael Isemann   }
31527caa17caSRaphael Isemann 
31537caa17caSRaphael Isemann   void Expand() { expanded = true; }
31547caa17caSRaphael Isemann 
31557caa17caSRaphael Isemann   std::vector<Row> &GetChildren() {
31567caa17caSRaphael Isemann     ProcessSP process_sp = value.GetProcessSP();
31577caa17caSRaphael Isemann     auto stop_id = process_sp->GetStopID();
31587caa17caSRaphael Isemann     if (process_sp && stop_id != children_stop_id) {
31597caa17caSRaphael Isemann       children_stop_id = stop_id;
31607caa17caSRaphael Isemann       calculated_children = false;
31617caa17caSRaphael Isemann     }
31627caa17caSRaphael Isemann     if (!calculated_children) {
31637caa17caSRaphael Isemann       children.clear();
31647caa17caSRaphael Isemann       calculated_children = true;
31657caa17caSRaphael Isemann       ValueObjectSP valobj = value.GetSP();
31667caa17caSRaphael Isemann       if (valobj) {
31677caa17caSRaphael Isemann         const size_t num_children = valobj->GetNumChildren();
31687caa17caSRaphael Isemann         for (size_t i = 0; i < num_children; ++i) {
31697caa17caSRaphael Isemann           children.push_back(Row(valobj->GetChildAtIndex(i, true), this));
31707caa17caSRaphael Isemann         }
31717caa17caSRaphael Isemann       }
31727caa17caSRaphael Isemann     }
31737caa17caSRaphael Isemann     return children;
31747caa17caSRaphael Isemann   }
31757caa17caSRaphael Isemann 
31767caa17caSRaphael Isemann   void Unexpand() {
31777caa17caSRaphael Isemann     expanded = false;
31787caa17caSRaphael Isemann     calculated_children = false;
31797caa17caSRaphael Isemann     children.clear();
31807caa17caSRaphael Isemann   }
31817caa17caSRaphael Isemann 
31827caa17caSRaphael Isemann   void DrawTree(Window &window) {
31837caa17caSRaphael Isemann     if (parent)
31847caa17caSRaphael Isemann       parent->DrawTreeForChild(window, this, 0);
31857caa17caSRaphael Isemann 
31867caa17caSRaphael Isemann     if (might_have_children) {
31877caa17caSRaphael Isemann       // It we can get UTF8 characters to work we should try to use the
31887caa17caSRaphael Isemann       // "symbol" UTF8 string below
31897caa17caSRaphael Isemann       //            const char *symbol = "";
31907caa17caSRaphael Isemann       //            if (row.expanded)
31917caa17caSRaphael Isemann       //                symbol = "\xe2\x96\xbd ";
31927caa17caSRaphael Isemann       //            else
31937caa17caSRaphael Isemann       //                symbol = "\xe2\x96\xb7 ";
31947caa17caSRaphael Isemann       //            window.PutCString (symbol);
31957caa17caSRaphael Isemann 
31967caa17caSRaphael Isemann       // The ACS_DARROW and ACS_RARROW don't look very nice they are just a 'v'
31977caa17caSRaphael Isemann       // or '>' character...
31987caa17caSRaphael Isemann       //            if (expanded)
31997caa17caSRaphael Isemann       //                window.PutChar (ACS_DARROW);
32007caa17caSRaphael Isemann       //            else
32017caa17caSRaphael Isemann       //                window.PutChar (ACS_RARROW);
32027caa17caSRaphael Isemann       // Since we can't find any good looking right arrow/down arrow symbols,
32037caa17caSRaphael Isemann       // just use a diamond...
32047caa17caSRaphael Isemann       window.PutChar(ACS_DIAMOND);
32057caa17caSRaphael Isemann       window.PutChar(ACS_HLINE);
32067caa17caSRaphael Isemann     }
32077caa17caSRaphael Isemann   }
32087caa17caSRaphael Isemann 
32097caa17caSRaphael Isemann   void DrawTreeForChild(Window &window, Row *child, uint32_t reverse_depth) {
32107caa17caSRaphael Isemann     if (parent)
32117caa17caSRaphael Isemann       parent->DrawTreeForChild(window, this, reverse_depth + 1);
32127caa17caSRaphael Isemann 
32137caa17caSRaphael Isemann     if (&GetChildren().back() == child) {
32147caa17caSRaphael Isemann       // Last child
32157caa17caSRaphael Isemann       if (reverse_depth == 0) {
32167caa17caSRaphael Isemann         window.PutChar(ACS_LLCORNER);
32177caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
32187caa17caSRaphael Isemann       } else {
32197caa17caSRaphael Isemann         window.PutChar(' ');
32207caa17caSRaphael Isemann         window.PutChar(' ');
32217caa17caSRaphael Isemann       }
32227caa17caSRaphael Isemann     } else {
32237caa17caSRaphael Isemann       if (reverse_depth == 0) {
32247caa17caSRaphael Isemann         window.PutChar(ACS_LTEE);
32257caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
32267caa17caSRaphael Isemann       } else {
32277caa17caSRaphael Isemann         window.PutChar(ACS_VLINE);
32287caa17caSRaphael Isemann         window.PutChar(' ');
32297caa17caSRaphael Isemann       }
32307caa17caSRaphael Isemann     }
32317caa17caSRaphael Isemann   }
32327caa17caSRaphael Isemann };
32337caa17caSRaphael Isemann 
32347caa17caSRaphael Isemann struct DisplayOptions {
32357caa17caSRaphael Isemann   bool show_types;
32367caa17caSRaphael Isemann };
32377caa17caSRaphael Isemann 
32387caa17caSRaphael Isemann class TreeItem;
32397caa17caSRaphael Isemann 
32407caa17caSRaphael Isemann class TreeDelegate {
32417caa17caSRaphael Isemann public:
32427caa17caSRaphael Isemann   TreeDelegate() = default;
32437caa17caSRaphael Isemann   virtual ~TreeDelegate() = default;
32447caa17caSRaphael Isemann 
32457caa17caSRaphael Isemann   virtual void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) = 0;
32467caa17caSRaphael Isemann   virtual void TreeDelegateGenerateChildren(TreeItem &item) = 0;
32477caa17caSRaphael Isemann   virtual bool TreeDelegateItemSelected(
32487caa17caSRaphael Isemann       TreeItem &item) = 0; // Return true if we need to update views
32497caa17caSRaphael Isemann };
32507caa17caSRaphael Isemann 
32517caa17caSRaphael Isemann typedef std::shared_ptr<TreeDelegate> TreeDelegateSP;
32527caa17caSRaphael Isemann 
32537caa17caSRaphael Isemann class TreeItem {
32547caa17caSRaphael Isemann public:
32557caa17caSRaphael Isemann   TreeItem(TreeItem *parent, TreeDelegate &delegate, bool might_have_children)
32567caa17caSRaphael Isemann       : m_parent(parent), m_delegate(delegate), m_user_data(nullptr),
32577caa17caSRaphael Isemann         m_identifier(0), m_row_idx(-1), m_children(),
32587caa17caSRaphael Isemann         m_might_have_children(might_have_children), m_is_expanded(false) {}
32597caa17caSRaphael Isemann 
32607caa17caSRaphael Isemann   TreeItem &operator=(const TreeItem &rhs) {
32617caa17caSRaphael Isemann     if (this != &rhs) {
32627caa17caSRaphael Isemann       m_parent = rhs.m_parent;
32637caa17caSRaphael Isemann       m_delegate = rhs.m_delegate;
32647caa17caSRaphael Isemann       m_user_data = rhs.m_user_data;
32657caa17caSRaphael Isemann       m_identifier = rhs.m_identifier;
32667caa17caSRaphael Isemann       m_row_idx = rhs.m_row_idx;
32677caa17caSRaphael Isemann       m_children = rhs.m_children;
32687caa17caSRaphael Isemann       m_might_have_children = rhs.m_might_have_children;
32697caa17caSRaphael Isemann       m_is_expanded = rhs.m_is_expanded;
32707caa17caSRaphael Isemann     }
32717caa17caSRaphael Isemann     return *this;
32727caa17caSRaphael Isemann   }
32737caa17caSRaphael Isemann 
32745312139fSEric Christopher   TreeItem(const TreeItem &) = default;
32755312139fSEric Christopher 
32767caa17caSRaphael Isemann   size_t GetDepth() const {
32777caa17caSRaphael Isemann     if (m_parent)
32787caa17caSRaphael Isemann       return 1 + m_parent->GetDepth();
32797caa17caSRaphael Isemann     return 0;
32807caa17caSRaphael Isemann   }
32817caa17caSRaphael Isemann 
32827caa17caSRaphael Isemann   int GetRowIndex() const { return m_row_idx; }
32837caa17caSRaphael Isemann 
32847caa17caSRaphael Isemann   void ClearChildren() { m_children.clear(); }
32857caa17caSRaphael Isemann 
32867caa17caSRaphael Isemann   void Resize(size_t n, const TreeItem &t) { m_children.resize(n, t); }
32877caa17caSRaphael Isemann 
32887caa17caSRaphael Isemann   TreeItem &operator[](size_t i) { return m_children[i]; }
32897caa17caSRaphael Isemann 
32907caa17caSRaphael Isemann   void SetRowIndex(int row_idx) { m_row_idx = row_idx; }
32917caa17caSRaphael Isemann 
32927caa17caSRaphael Isemann   size_t GetNumChildren() {
32937caa17caSRaphael Isemann     m_delegate.TreeDelegateGenerateChildren(*this);
32947caa17caSRaphael Isemann     return m_children.size();
32957caa17caSRaphael Isemann   }
32967caa17caSRaphael Isemann 
32977caa17caSRaphael Isemann   void ItemWasSelected() { m_delegate.TreeDelegateItemSelected(*this); }
32987caa17caSRaphael Isemann 
32997caa17caSRaphael Isemann   void CalculateRowIndexes(int &row_idx) {
33007caa17caSRaphael Isemann     SetRowIndex(row_idx);
33017caa17caSRaphael Isemann     ++row_idx;
33027caa17caSRaphael Isemann 
33037caa17caSRaphael Isemann     const bool expanded = IsExpanded();
33047caa17caSRaphael Isemann 
33057caa17caSRaphael Isemann     // The root item must calculate its children, or we must calculate the
33067caa17caSRaphael Isemann     // number of children if the item is expanded
33077caa17caSRaphael Isemann     if (m_parent == nullptr || expanded)
33087caa17caSRaphael Isemann       GetNumChildren();
33097caa17caSRaphael Isemann 
33107caa17caSRaphael Isemann     for (auto &item : m_children) {
33117caa17caSRaphael Isemann       if (expanded)
33127caa17caSRaphael Isemann         item.CalculateRowIndexes(row_idx);
33137caa17caSRaphael Isemann       else
33147caa17caSRaphael Isemann         item.SetRowIndex(-1);
33157caa17caSRaphael Isemann     }
33167caa17caSRaphael Isemann   }
33177caa17caSRaphael Isemann 
33187caa17caSRaphael Isemann   TreeItem *GetParent() { return m_parent; }
33197caa17caSRaphael Isemann 
33207caa17caSRaphael Isemann   bool IsExpanded() const { return m_is_expanded; }
33217caa17caSRaphael Isemann 
33227caa17caSRaphael Isemann   void Expand() { m_is_expanded = true; }
33237caa17caSRaphael Isemann 
33247caa17caSRaphael Isemann   void Unexpand() { m_is_expanded = false; }
33257caa17caSRaphael Isemann 
33267caa17caSRaphael Isemann   bool Draw(Window &window, const int first_visible_row,
33277caa17caSRaphael Isemann             const uint32_t selected_row_idx, int &row_idx, int &num_rows_left) {
33287caa17caSRaphael Isemann     if (num_rows_left <= 0)
33297caa17caSRaphael Isemann       return false;
33307caa17caSRaphael Isemann 
33317caa17caSRaphael Isemann     if (m_row_idx >= first_visible_row) {
33327caa17caSRaphael Isemann       window.MoveCursor(2, row_idx + 1);
33337caa17caSRaphael Isemann 
33347caa17caSRaphael Isemann       if (m_parent)
33357caa17caSRaphael Isemann         m_parent->DrawTreeForChild(window, this, 0);
33367caa17caSRaphael Isemann 
33377caa17caSRaphael Isemann       if (m_might_have_children) {
33387caa17caSRaphael Isemann         // It we can get UTF8 characters to work we should try to use the
33397caa17caSRaphael Isemann         // "symbol" UTF8 string below
33407caa17caSRaphael Isemann         //            const char *symbol = "";
33417caa17caSRaphael Isemann         //            if (row.expanded)
33427caa17caSRaphael Isemann         //                symbol = "\xe2\x96\xbd ";
33437caa17caSRaphael Isemann         //            else
33447caa17caSRaphael Isemann         //                symbol = "\xe2\x96\xb7 ";
33457caa17caSRaphael Isemann         //            window.PutCString (symbol);
33467caa17caSRaphael Isemann 
33477caa17caSRaphael Isemann         // The ACS_DARROW and ACS_RARROW don't look very nice they are just a
33487caa17caSRaphael Isemann         // 'v' or '>' character...
33497caa17caSRaphael Isemann         //            if (expanded)
33507caa17caSRaphael Isemann         //                window.PutChar (ACS_DARROW);
33517caa17caSRaphael Isemann         //            else
33527caa17caSRaphael Isemann         //                window.PutChar (ACS_RARROW);
33537caa17caSRaphael Isemann         // Since we can't find any good looking right arrow/down arrow symbols,
33547caa17caSRaphael Isemann         // just use a diamond...
33557caa17caSRaphael Isemann         window.PutChar(ACS_DIAMOND);
33567caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
33577caa17caSRaphael Isemann       }
33587caa17caSRaphael Isemann       bool highlight = (selected_row_idx == static_cast<size_t>(m_row_idx)) &&
33597caa17caSRaphael Isemann                        window.IsActive();
33607caa17caSRaphael Isemann 
33617caa17caSRaphael Isemann       if (highlight)
33627caa17caSRaphael Isemann         window.AttributeOn(A_REVERSE);
33637caa17caSRaphael Isemann 
33647caa17caSRaphael Isemann       m_delegate.TreeDelegateDrawTreeItem(*this, window);
33657caa17caSRaphael Isemann 
33667caa17caSRaphael Isemann       if (highlight)
33677caa17caSRaphael Isemann         window.AttributeOff(A_REVERSE);
33687caa17caSRaphael Isemann       ++row_idx;
33697caa17caSRaphael Isemann       --num_rows_left;
33707caa17caSRaphael Isemann     }
33717caa17caSRaphael Isemann 
33727caa17caSRaphael Isemann     if (num_rows_left <= 0)
33737caa17caSRaphael Isemann       return false; // We are done drawing...
33747caa17caSRaphael Isemann 
33757caa17caSRaphael Isemann     if (IsExpanded()) {
33767caa17caSRaphael Isemann       for (auto &item : m_children) {
33777caa17caSRaphael Isemann         // If we displayed all the rows and item.Draw() returns false we are
33787caa17caSRaphael Isemann         // done drawing and can exit this for loop
33797caa17caSRaphael Isemann         if (!item.Draw(window, first_visible_row, selected_row_idx, row_idx,
33807caa17caSRaphael Isemann                        num_rows_left))
33817caa17caSRaphael Isemann           break;
33827caa17caSRaphael Isemann       }
33837caa17caSRaphael Isemann     }
33847caa17caSRaphael Isemann     return num_rows_left >= 0; // Return true if not done drawing yet
33857caa17caSRaphael Isemann   }
33867caa17caSRaphael Isemann 
33877caa17caSRaphael Isemann   void DrawTreeForChild(Window &window, TreeItem *child,
33887caa17caSRaphael Isemann                         uint32_t reverse_depth) {
33897caa17caSRaphael Isemann     if (m_parent)
33907caa17caSRaphael Isemann       m_parent->DrawTreeForChild(window, this, reverse_depth + 1);
33917caa17caSRaphael Isemann 
33927caa17caSRaphael Isemann     if (&m_children.back() == child) {
33937caa17caSRaphael Isemann       // Last child
33947caa17caSRaphael Isemann       if (reverse_depth == 0) {
33957caa17caSRaphael Isemann         window.PutChar(ACS_LLCORNER);
33967caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
33977caa17caSRaphael Isemann       } else {
33987caa17caSRaphael Isemann         window.PutChar(' ');
33997caa17caSRaphael Isemann         window.PutChar(' ');
34007caa17caSRaphael Isemann       }
34017caa17caSRaphael Isemann     } else {
34027caa17caSRaphael Isemann       if (reverse_depth == 0) {
34037caa17caSRaphael Isemann         window.PutChar(ACS_LTEE);
34047caa17caSRaphael Isemann         window.PutChar(ACS_HLINE);
34057caa17caSRaphael Isemann       } else {
34067caa17caSRaphael Isemann         window.PutChar(ACS_VLINE);
34077caa17caSRaphael Isemann         window.PutChar(' ');
34087caa17caSRaphael Isemann       }
34097caa17caSRaphael Isemann     }
34107caa17caSRaphael Isemann   }
34117caa17caSRaphael Isemann 
34127caa17caSRaphael Isemann   TreeItem *GetItemForRowIndex(uint32_t row_idx) {
34137caa17caSRaphael Isemann     if (static_cast<uint32_t>(m_row_idx) == row_idx)
34147caa17caSRaphael Isemann       return this;
34157caa17caSRaphael Isemann     if (m_children.empty())
34167caa17caSRaphael Isemann       return nullptr;
34177caa17caSRaphael Isemann     if (IsExpanded()) {
34187caa17caSRaphael Isemann       for (auto &item : m_children) {
34197caa17caSRaphael Isemann         TreeItem *selected_item_ptr = item.GetItemForRowIndex(row_idx);
34207caa17caSRaphael Isemann         if (selected_item_ptr)
34217caa17caSRaphael Isemann           return selected_item_ptr;
34227caa17caSRaphael Isemann       }
34237caa17caSRaphael Isemann     }
34247caa17caSRaphael Isemann     return nullptr;
34257caa17caSRaphael Isemann   }
34267caa17caSRaphael Isemann 
34277caa17caSRaphael Isemann   void *GetUserData() const { return m_user_data; }
34287caa17caSRaphael Isemann 
34297caa17caSRaphael Isemann   void SetUserData(void *user_data) { m_user_data = user_data; }
34307caa17caSRaphael Isemann 
34317caa17caSRaphael Isemann   uint64_t GetIdentifier() const { return m_identifier; }
34327caa17caSRaphael Isemann 
34337caa17caSRaphael Isemann   void SetIdentifier(uint64_t identifier) { m_identifier = identifier; }
34347caa17caSRaphael Isemann 
34357caa17caSRaphael Isemann   void SetMightHaveChildren(bool b) { m_might_have_children = b; }
34367caa17caSRaphael Isemann 
34377caa17caSRaphael Isemann protected:
34387caa17caSRaphael Isemann   TreeItem *m_parent;
34397caa17caSRaphael Isemann   TreeDelegate &m_delegate;
34407caa17caSRaphael Isemann   void *m_user_data;
34417caa17caSRaphael Isemann   uint64_t m_identifier;
34427caa17caSRaphael Isemann   int m_row_idx; // Zero based visible row index, -1 if not visible or for the
34437caa17caSRaphael Isemann                  // root item
34447caa17caSRaphael Isemann   std::vector<TreeItem> m_children;
34457caa17caSRaphael Isemann   bool m_might_have_children;
34467caa17caSRaphael Isemann   bool m_is_expanded;
34477caa17caSRaphael Isemann };
34487caa17caSRaphael Isemann 
34497caa17caSRaphael Isemann class TreeWindowDelegate : public WindowDelegate {
34507caa17caSRaphael Isemann public:
34517caa17caSRaphael Isemann   TreeWindowDelegate(Debugger &debugger, const TreeDelegateSP &delegate_sp)
34527caa17caSRaphael Isemann       : m_debugger(debugger), m_delegate_sp(delegate_sp),
34537caa17caSRaphael Isemann         m_root(nullptr, *delegate_sp, true), m_selected_item(nullptr),
34547caa17caSRaphael Isemann         m_num_rows(0), m_selected_row_idx(0), m_first_visible_row(0),
34557caa17caSRaphael Isemann         m_min_x(0), m_min_y(0), m_max_x(0), m_max_y(0) {}
34567caa17caSRaphael Isemann 
34577caa17caSRaphael Isemann   int NumVisibleRows() const { return m_max_y - m_min_y; }
34587caa17caSRaphael Isemann 
34597caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
34607caa17caSRaphael Isemann     ExecutionContext exe_ctx(
34617caa17caSRaphael Isemann         m_debugger.GetCommandInterpreter().GetExecutionContext());
34627caa17caSRaphael Isemann     Process *process = exe_ctx.GetProcessPtr();
34637caa17caSRaphael Isemann 
34647caa17caSRaphael Isemann     bool display_content = false;
34657caa17caSRaphael Isemann     if (process) {
34667caa17caSRaphael Isemann       StateType state = process->GetState();
34677caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
34687caa17caSRaphael Isemann         // We are stopped, so it is ok to
34697caa17caSRaphael Isemann         display_content = true;
34707caa17caSRaphael Isemann       } else if (StateIsRunningState(state)) {
34717caa17caSRaphael Isemann         return true; // Don't do any updating when we are running
34727caa17caSRaphael Isemann       }
34737caa17caSRaphael Isemann     }
34747caa17caSRaphael Isemann 
34757caa17caSRaphael Isemann     m_min_x = 2;
34767caa17caSRaphael Isemann     m_min_y = 1;
34777caa17caSRaphael Isemann     m_max_x = window.GetWidth() - 1;
34787caa17caSRaphael Isemann     m_max_y = window.GetHeight() - 1;
34797caa17caSRaphael Isemann 
34807caa17caSRaphael Isemann     window.Erase();
34817caa17caSRaphael Isemann     window.DrawTitleBox(window.GetName());
34827caa17caSRaphael Isemann 
34837caa17caSRaphael Isemann     if (display_content) {
34847caa17caSRaphael Isemann       const int num_visible_rows = NumVisibleRows();
34857caa17caSRaphael Isemann       m_num_rows = 0;
34867caa17caSRaphael Isemann       m_root.CalculateRowIndexes(m_num_rows);
34877caa17caSRaphael Isemann 
34887caa17caSRaphael Isemann       // If we unexpanded while having something selected our total number of
34897caa17caSRaphael Isemann       // rows is less than the num visible rows, then make sure we show all the
34907caa17caSRaphael Isemann       // rows by setting the first visible row accordingly.
34917caa17caSRaphael Isemann       if (m_first_visible_row > 0 && m_num_rows < num_visible_rows)
34927caa17caSRaphael Isemann         m_first_visible_row = 0;
34937caa17caSRaphael Isemann 
34947caa17caSRaphael Isemann       // Make sure the selected row is always visible
34957caa17caSRaphael Isemann       if (m_selected_row_idx < m_first_visible_row)
34967caa17caSRaphael Isemann         m_first_visible_row = m_selected_row_idx;
34977caa17caSRaphael Isemann       else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
34987caa17caSRaphael Isemann         m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
34997caa17caSRaphael Isemann 
35007caa17caSRaphael Isemann       int row_idx = 0;
35017caa17caSRaphael Isemann       int num_rows_left = num_visible_rows;
35027caa17caSRaphael Isemann       m_root.Draw(window, m_first_visible_row, m_selected_row_idx, row_idx,
35037caa17caSRaphael Isemann                   num_rows_left);
35047caa17caSRaphael Isemann       // Get the selected row
35057caa17caSRaphael Isemann       m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35067caa17caSRaphael Isemann     } else {
35077caa17caSRaphael Isemann       m_selected_item = nullptr;
35087caa17caSRaphael Isemann     }
35097caa17caSRaphael Isemann 
35107caa17caSRaphael Isemann     return true; // Drawing handled
35117caa17caSRaphael Isemann   }
35127caa17caSRaphael Isemann 
35137caa17caSRaphael Isemann   const char *WindowDelegateGetHelpText() override {
35147caa17caSRaphael Isemann     return "Thread window keyboard shortcuts:";
35157caa17caSRaphael Isemann   }
35167caa17caSRaphael Isemann 
35177caa17caSRaphael Isemann   KeyHelp *WindowDelegateGetKeyHelp() override {
35187caa17caSRaphael Isemann     static curses::KeyHelp g_source_view_key_help[] = {
35197caa17caSRaphael Isemann         {KEY_UP, "Select previous item"},
35207caa17caSRaphael Isemann         {KEY_DOWN, "Select next item"},
35217caa17caSRaphael Isemann         {KEY_RIGHT, "Expand the selected item"},
35227caa17caSRaphael Isemann         {KEY_LEFT,
35237caa17caSRaphael Isemann          "Unexpand the selected item or select parent if not expanded"},
35247caa17caSRaphael Isemann         {KEY_PPAGE, "Page up"},
35257caa17caSRaphael Isemann         {KEY_NPAGE, "Page down"},
35267caa17caSRaphael Isemann         {'h', "Show help dialog"},
35277caa17caSRaphael Isemann         {' ', "Toggle item expansion"},
35287caa17caSRaphael Isemann         {',', "Page up"},
35297caa17caSRaphael Isemann         {'.', "Page down"},
35307caa17caSRaphael Isemann         {'\0', nullptr}};
35317caa17caSRaphael Isemann     return g_source_view_key_help;
35327caa17caSRaphael Isemann   }
35337caa17caSRaphael Isemann 
35347caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
35357caa17caSRaphael Isemann     switch (c) {
35367caa17caSRaphael Isemann     case ',':
35377caa17caSRaphael Isemann     case KEY_PPAGE:
35387caa17caSRaphael Isemann       // Page up key
35397caa17caSRaphael Isemann       if (m_first_visible_row > 0) {
35407caa17caSRaphael Isemann         if (m_first_visible_row > m_max_y)
35417caa17caSRaphael Isemann           m_first_visible_row -= m_max_y;
35427caa17caSRaphael Isemann         else
35437caa17caSRaphael Isemann           m_first_visible_row = 0;
35447caa17caSRaphael Isemann         m_selected_row_idx = m_first_visible_row;
35457caa17caSRaphael Isemann         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35467caa17caSRaphael Isemann         if (m_selected_item)
35477caa17caSRaphael Isemann           m_selected_item->ItemWasSelected();
35487caa17caSRaphael Isemann       }
35497caa17caSRaphael Isemann       return eKeyHandled;
35507caa17caSRaphael Isemann 
35517caa17caSRaphael Isemann     case '.':
35527caa17caSRaphael Isemann     case KEY_NPAGE:
35537caa17caSRaphael Isemann       // Page down key
35547caa17caSRaphael Isemann       if (m_num_rows > m_max_y) {
35557caa17caSRaphael Isemann         if (m_first_visible_row + m_max_y < m_num_rows) {
35567caa17caSRaphael Isemann           m_first_visible_row += m_max_y;
35577caa17caSRaphael Isemann           m_selected_row_idx = m_first_visible_row;
35587caa17caSRaphael Isemann           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35597caa17caSRaphael Isemann           if (m_selected_item)
35607caa17caSRaphael Isemann             m_selected_item->ItemWasSelected();
35617caa17caSRaphael Isemann         }
35627caa17caSRaphael Isemann       }
35637caa17caSRaphael Isemann       return eKeyHandled;
35647caa17caSRaphael Isemann 
35657caa17caSRaphael Isemann     case KEY_UP:
35667caa17caSRaphael Isemann       if (m_selected_row_idx > 0) {
35677caa17caSRaphael Isemann         --m_selected_row_idx;
35687caa17caSRaphael Isemann         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35697caa17caSRaphael Isemann         if (m_selected_item)
35707caa17caSRaphael Isemann           m_selected_item->ItemWasSelected();
35717caa17caSRaphael Isemann       }
35727caa17caSRaphael Isemann       return eKeyHandled;
35737caa17caSRaphael Isemann 
35747caa17caSRaphael Isemann     case KEY_DOWN:
35757caa17caSRaphael Isemann       if (m_selected_row_idx + 1 < m_num_rows) {
35767caa17caSRaphael Isemann         ++m_selected_row_idx;
35777caa17caSRaphael Isemann         m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35787caa17caSRaphael Isemann         if (m_selected_item)
35797caa17caSRaphael Isemann           m_selected_item->ItemWasSelected();
35807caa17caSRaphael Isemann       }
35817caa17caSRaphael Isemann       return eKeyHandled;
35827caa17caSRaphael Isemann 
35837caa17caSRaphael Isemann     case KEY_RIGHT:
35847caa17caSRaphael Isemann       if (m_selected_item) {
35857caa17caSRaphael Isemann         if (!m_selected_item->IsExpanded())
35867caa17caSRaphael Isemann           m_selected_item->Expand();
35877caa17caSRaphael Isemann       }
35887caa17caSRaphael Isemann       return eKeyHandled;
35897caa17caSRaphael Isemann 
35907caa17caSRaphael Isemann     case KEY_LEFT:
35917caa17caSRaphael Isemann       if (m_selected_item) {
35927caa17caSRaphael Isemann         if (m_selected_item->IsExpanded())
35937caa17caSRaphael Isemann           m_selected_item->Unexpand();
35947caa17caSRaphael Isemann         else if (m_selected_item->GetParent()) {
35957caa17caSRaphael Isemann           m_selected_row_idx = m_selected_item->GetParent()->GetRowIndex();
35967caa17caSRaphael Isemann           m_selected_item = m_root.GetItemForRowIndex(m_selected_row_idx);
35977caa17caSRaphael Isemann           if (m_selected_item)
35987caa17caSRaphael Isemann             m_selected_item->ItemWasSelected();
35997caa17caSRaphael Isemann         }
36007caa17caSRaphael Isemann       }
36017caa17caSRaphael Isemann       return eKeyHandled;
36027caa17caSRaphael Isemann 
36037caa17caSRaphael Isemann     case ' ':
36047caa17caSRaphael Isemann       // Toggle expansion state when SPACE is pressed
36057caa17caSRaphael Isemann       if (m_selected_item) {
36067caa17caSRaphael Isemann         if (m_selected_item->IsExpanded())
36077caa17caSRaphael Isemann           m_selected_item->Unexpand();
36087caa17caSRaphael Isemann         else
36097caa17caSRaphael Isemann           m_selected_item->Expand();
36107caa17caSRaphael Isemann       }
36117caa17caSRaphael Isemann       return eKeyHandled;
36127caa17caSRaphael Isemann 
36137caa17caSRaphael Isemann     case 'h':
36147caa17caSRaphael Isemann       window.CreateHelpSubwindow();
36157caa17caSRaphael Isemann       return eKeyHandled;
36167caa17caSRaphael Isemann 
36177caa17caSRaphael Isemann     default:
36187caa17caSRaphael Isemann       break;
36197caa17caSRaphael Isemann     }
36207caa17caSRaphael Isemann     return eKeyNotHandled;
36217caa17caSRaphael Isemann   }
36227caa17caSRaphael Isemann 
36237caa17caSRaphael Isemann protected:
36247caa17caSRaphael Isemann   Debugger &m_debugger;
36257caa17caSRaphael Isemann   TreeDelegateSP m_delegate_sp;
36267caa17caSRaphael Isemann   TreeItem m_root;
36277caa17caSRaphael Isemann   TreeItem *m_selected_item;
36287caa17caSRaphael Isemann   int m_num_rows;
36297caa17caSRaphael Isemann   int m_selected_row_idx;
36307caa17caSRaphael Isemann   int m_first_visible_row;
36317caa17caSRaphael Isemann   int m_min_x;
36327caa17caSRaphael Isemann   int m_min_y;
36337caa17caSRaphael Isemann   int m_max_x;
36347caa17caSRaphael Isemann   int m_max_y;
36357caa17caSRaphael Isemann };
36367caa17caSRaphael Isemann 
36377caa17caSRaphael Isemann class FrameTreeDelegate : public TreeDelegate {
36387caa17caSRaphael Isemann public:
36397caa17caSRaphael Isemann   FrameTreeDelegate() : TreeDelegate() {
36407caa17caSRaphael Isemann     FormatEntity::Parse(
36417caa17caSRaphael Isemann         "frame #${frame.index}: {${function.name}${function.pc-offset}}}",
36427caa17caSRaphael Isemann         m_format);
36437caa17caSRaphael Isemann   }
36447caa17caSRaphael Isemann 
36457caa17caSRaphael Isemann   ~FrameTreeDelegate() override = default;
36467caa17caSRaphael Isemann 
36477caa17caSRaphael Isemann   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
36487caa17caSRaphael Isemann     Thread *thread = (Thread *)item.GetUserData();
36497caa17caSRaphael Isemann     if (thread) {
36507caa17caSRaphael Isemann       const uint64_t frame_idx = item.GetIdentifier();
36517caa17caSRaphael Isemann       StackFrameSP frame_sp = thread->GetStackFrameAtIndex(frame_idx);
36527caa17caSRaphael Isemann       if (frame_sp) {
36537caa17caSRaphael Isemann         StreamString strm;
36547caa17caSRaphael Isemann         const SymbolContext &sc =
36557caa17caSRaphael Isemann             frame_sp->GetSymbolContext(eSymbolContextEverything);
36567caa17caSRaphael Isemann         ExecutionContext exe_ctx(frame_sp);
36577caa17caSRaphael Isemann         if (FormatEntity::Format(m_format, strm, &sc, &exe_ctx, nullptr,
36587caa17caSRaphael Isemann                                  nullptr, false, false)) {
36597caa17caSRaphael Isemann           int right_pad = 1;
3660fc0e8fb7SLuboš Luňák           window.PutCStringTruncated(right_pad, strm.GetString().str().c_str());
36617caa17caSRaphael Isemann         }
36627caa17caSRaphael Isemann       }
36637caa17caSRaphael Isemann     }
36647caa17caSRaphael Isemann   }
36657caa17caSRaphael Isemann 
36667caa17caSRaphael Isemann   void TreeDelegateGenerateChildren(TreeItem &item) override {
36677caa17caSRaphael Isemann     // No children for frames yet...
36687caa17caSRaphael Isemann   }
36697caa17caSRaphael Isemann 
36707caa17caSRaphael Isemann   bool TreeDelegateItemSelected(TreeItem &item) override {
36717caa17caSRaphael Isemann     Thread *thread = (Thread *)item.GetUserData();
36727caa17caSRaphael Isemann     if (thread) {
36737caa17caSRaphael Isemann       thread->GetProcess()->GetThreadList().SetSelectedThreadByID(
36747caa17caSRaphael Isemann           thread->GetID());
36757caa17caSRaphael Isemann       const uint64_t frame_idx = item.GetIdentifier();
36767caa17caSRaphael Isemann       thread->SetSelectedFrameByIndex(frame_idx);
36777caa17caSRaphael Isemann       return true;
36787caa17caSRaphael Isemann     }
36797caa17caSRaphael Isemann     return false;
36807caa17caSRaphael Isemann   }
36817caa17caSRaphael Isemann 
36827caa17caSRaphael Isemann protected:
36837caa17caSRaphael Isemann   FormatEntity::Entry m_format;
36847caa17caSRaphael Isemann };
36857caa17caSRaphael Isemann 
36867caa17caSRaphael Isemann class ThreadTreeDelegate : public TreeDelegate {
36877caa17caSRaphael Isemann public:
36887caa17caSRaphael Isemann   ThreadTreeDelegate(Debugger &debugger)
36897caa17caSRaphael Isemann       : TreeDelegate(), m_debugger(debugger), m_tid(LLDB_INVALID_THREAD_ID),
36907caa17caSRaphael Isemann         m_stop_id(UINT32_MAX) {
36917caa17caSRaphael Isemann     FormatEntity::Parse("thread #${thread.index}: tid = ${thread.id}{, stop "
36927caa17caSRaphael Isemann                         "reason = ${thread.stop-reason}}",
36937caa17caSRaphael Isemann                         m_format);
36947caa17caSRaphael Isemann   }
36957caa17caSRaphael Isemann 
36967caa17caSRaphael Isemann   ~ThreadTreeDelegate() override = default;
36977caa17caSRaphael Isemann 
36987caa17caSRaphael Isemann   ProcessSP GetProcess() {
36997caa17caSRaphael Isemann     return m_debugger.GetCommandInterpreter()
37007caa17caSRaphael Isemann         .GetExecutionContext()
37017caa17caSRaphael Isemann         .GetProcessSP();
37027caa17caSRaphael Isemann   }
37037caa17caSRaphael Isemann 
37047caa17caSRaphael Isemann   ThreadSP GetThread(const TreeItem &item) {
37057caa17caSRaphael Isemann     ProcessSP process_sp = GetProcess();
37067caa17caSRaphael Isemann     if (process_sp)
37077caa17caSRaphael Isemann       return process_sp->GetThreadList().FindThreadByID(item.GetIdentifier());
37087caa17caSRaphael Isemann     return ThreadSP();
37097caa17caSRaphael Isemann   }
37107caa17caSRaphael Isemann 
37117caa17caSRaphael Isemann   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
37127caa17caSRaphael Isemann     ThreadSP thread_sp = GetThread(item);
37137caa17caSRaphael Isemann     if (thread_sp) {
37147caa17caSRaphael Isemann       StreamString strm;
37157caa17caSRaphael Isemann       ExecutionContext exe_ctx(thread_sp);
37167caa17caSRaphael Isemann       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
37177caa17caSRaphael Isemann                                nullptr, false, false)) {
37187caa17caSRaphael Isemann         int right_pad = 1;
3719fc0e8fb7SLuboš Luňák         window.PutCStringTruncated(right_pad, strm.GetString().str().c_str());
37207caa17caSRaphael Isemann       }
37217caa17caSRaphael Isemann     }
37227caa17caSRaphael Isemann   }
37237caa17caSRaphael Isemann 
37247caa17caSRaphael Isemann   void TreeDelegateGenerateChildren(TreeItem &item) override {
37257caa17caSRaphael Isemann     ProcessSP process_sp = GetProcess();
37267caa17caSRaphael Isemann     if (process_sp && process_sp->IsAlive()) {
37277caa17caSRaphael Isemann       StateType state = process_sp->GetState();
37287caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
37297caa17caSRaphael Isemann         ThreadSP thread_sp = GetThread(item);
37307caa17caSRaphael Isemann         if (thread_sp) {
37317caa17caSRaphael Isemann           if (m_stop_id == process_sp->GetStopID() &&
37327caa17caSRaphael Isemann               thread_sp->GetID() == m_tid)
37337caa17caSRaphael Isemann             return; // Children are already up to date
37347caa17caSRaphael Isemann           if (!m_frame_delegate_sp) {
37357caa17caSRaphael Isemann             // Always expand the thread item the first time we show it
37367caa17caSRaphael Isemann             m_frame_delegate_sp = std::make_shared<FrameTreeDelegate>();
37377caa17caSRaphael Isemann           }
37387caa17caSRaphael Isemann 
37397caa17caSRaphael Isemann           m_stop_id = process_sp->GetStopID();
37407caa17caSRaphael Isemann           m_tid = thread_sp->GetID();
37417caa17caSRaphael Isemann 
37427caa17caSRaphael Isemann           TreeItem t(&item, *m_frame_delegate_sp, false);
37437caa17caSRaphael Isemann           size_t num_frames = thread_sp->GetStackFrameCount();
37447caa17caSRaphael Isemann           item.Resize(num_frames, t);
37457caa17caSRaphael Isemann           for (size_t i = 0; i < num_frames; ++i) {
37467caa17caSRaphael Isemann             item[i].SetUserData(thread_sp.get());
37477caa17caSRaphael Isemann             item[i].SetIdentifier(i);
37487caa17caSRaphael Isemann           }
37497caa17caSRaphael Isemann         }
37507caa17caSRaphael Isemann         return;
37517caa17caSRaphael Isemann       }
37527caa17caSRaphael Isemann     }
37537caa17caSRaphael Isemann     item.ClearChildren();
37547caa17caSRaphael Isemann   }
37557caa17caSRaphael Isemann 
37567caa17caSRaphael Isemann   bool TreeDelegateItemSelected(TreeItem &item) override {
37577caa17caSRaphael Isemann     ProcessSP process_sp = GetProcess();
37587caa17caSRaphael Isemann     if (process_sp && process_sp->IsAlive()) {
37597caa17caSRaphael Isemann       StateType state = process_sp->GetState();
37607caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
37617caa17caSRaphael Isemann         ThreadSP thread_sp = GetThread(item);
37627caa17caSRaphael Isemann         if (thread_sp) {
37637caa17caSRaphael Isemann           ThreadList &thread_list = thread_sp->GetProcess()->GetThreadList();
37647caa17caSRaphael Isemann           std::lock_guard<std::recursive_mutex> guard(thread_list.GetMutex());
37657caa17caSRaphael Isemann           ThreadSP selected_thread_sp = thread_list.GetSelectedThread();
37667caa17caSRaphael Isemann           if (selected_thread_sp->GetID() != thread_sp->GetID()) {
37677caa17caSRaphael Isemann             thread_list.SetSelectedThreadByID(thread_sp->GetID());
37687caa17caSRaphael Isemann             return true;
37697caa17caSRaphael Isemann           }
37707caa17caSRaphael Isemann         }
37717caa17caSRaphael Isemann       }
37727caa17caSRaphael Isemann     }
37737caa17caSRaphael Isemann     return false;
37747caa17caSRaphael Isemann   }
37757caa17caSRaphael Isemann 
37767caa17caSRaphael Isemann protected:
37777caa17caSRaphael Isemann   Debugger &m_debugger;
37787caa17caSRaphael Isemann   std::shared_ptr<FrameTreeDelegate> m_frame_delegate_sp;
37797caa17caSRaphael Isemann   lldb::user_id_t m_tid;
37807caa17caSRaphael Isemann   uint32_t m_stop_id;
37817caa17caSRaphael Isemann   FormatEntity::Entry m_format;
37827caa17caSRaphael Isemann };
37837caa17caSRaphael Isemann 
37847caa17caSRaphael Isemann class ThreadsTreeDelegate : public TreeDelegate {
37857caa17caSRaphael Isemann public:
37867caa17caSRaphael Isemann   ThreadsTreeDelegate(Debugger &debugger)
37877caa17caSRaphael Isemann       : TreeDelegate(), m_thread_delegate_sp(), m_debugger(debugger),
37887caa17caSRaphael Isemann         m_stop_id(UINT32_MAX) {
37897caa17caSRaphael Isemann     FormatEntity::Parse("process ${process.id}{, name = ${process.name}}",
37907caa17caSRaphael Isemann                         m_format);
37917caa17caSRaphael Isemann   }
37927caa17caSRaphael Isemann 
37937caa17caSRaphael Isemann   ~ThreadsTreeDelegate() override = default;
37947caa17caSRaphael Isemann 
37957caa17caSRaphael Isemann   ProcessSP GetProcess() {
37967caa17caSRaphael Isemann     return m_debugger.GetCommandInterpreter()
37977caa17caSRaphael Isemann         .GetExecutionContext()
37987caa17caSRaphael Isemann         .GetProcessSP();
37997caa17caSRaphael Isemann   }
38007caa17caSRaphael Isemann 
38017caa17caSRaphael Isemann   void TreeDelegateDrawTreeItem(TreeItem &item, Window &window) override {
38027caa17caSRaphael Isemann     ProcessSP process_sp = GetProcess();
38037caa17caSRaphael Isemann     if (process_sp && process_sp->IsAlive()) {
38047caa17caSRaphael Isemann       StreamString strm;
38057caa17caSRaphael Isemann       ExecutionContext exe_ctx(process_sp);
38067caa17caSRaphael Isemann       if (FormatEntity::Format(m_format, strm, nullptr, &exe_ctx, nullptr,
38077caa17caSRaphael Isemann                                nullptr, false, false)) {
38087caa17caSRaphael Isemann         int right_pad = 1;
3809fc0e8fb7SLuboš Luňák         window.PutCStringTruncated(right_pad, strm.GetString().str().c_str());
38107caa17caSRaphael Isemann       }
38117caa17caSRaphael Isemann     }
38127caa17caSRaphael Isemann   }
38137caa17caSRaphael Isemann 
38147caa17caSRaphael Isemann   void TreeDelegateGenerateChildren(TreeItem &item) override {
38157caa17caSRaphael Isemann     ProcessSP process_sp = GetProcess();
38167caa17caSRaphael Isemann     if (process_sp && process_sp->IsAlive()) {
38177caa17caSRaphael Isemann       StateType state = process_sp->GetState();
38187caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
38197caa17caSRaphael Isemann         const uint32_t stop_id = process_sp->GetStopID();
38207caa17caSRaphael Isemann         if (m_stop_id == stop_id)
38217caa17caSRaphael Isemann           return; // Children are already up to date
38227caa17caSRaphael Isemann 
38237caa17caSRaphael Isemann         m_stop_id = stop_id;
38247caa17caSRaphael Isemann 
38257caa17caSRaphael Isemann         if (!m_thread_delegate_sp) {
38267caa17caSRaphael Isemann           // Always expand the thread item the first time we show it
38277caa17caSRaphael Isemann           // item.Expand();
38287caa17caSRaphael Isemann           m_thread_delegate_sp =
38297caa17caSRaphael Isemann               std::make_shared<ThreadTreeDelegate>(m_debugger);
38307caa17caSRaphael Isemann         }
38317caa17caSRaphael Isemann 
38327caa17caSRaphael Isemann         TreeItem t(&item, *m_thread_delegate_sp, false);
38337caa17caSRaphael Isemann         ThreadList &threads = process_sp->GetThreadList();
38347caa17caSRaphael Isemann         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
38357caa17caSRaphael Isemann         size_t num_threads = threads.GetSize();
38367caa17caSRaphael Isemann         item.Resize(num_threads, t);
38377caa17caSRaphael Isemann         for (size_t i = 0; i < num_threads; ++i) {
38387caa17caSRaphael Isemann           item[i].SetIdentifier(threads.GetThreadAtIndex(i)->GetID());
38397caa17caSRaphael Isemann           item[i].SetMightHaveChildren(true);
38407caa17caSRaphael Isemann         }
38417caa17caSRaphael Isemann         return;
38427caa17caSRaphael Isemann       }
38437caa17caSRaphael Isemann     }
38447caa17caSRaphael Isemann     item.ClearChildren();
38457caa17caSRaphael Isemann   }
38467caa17caSRaphael Isemann 
38477caa17caSRaphael Isemann   bool TreeDelegateItemSelected(TreeItem &item) override { return false; }
38487caa17caSRaphael Isemann 
38497caa17caSRaphael Isemann protected:
38507caa17caSRaphael Isemann   std::shared_ptr<ThreadTreeDelegate> m_thread_delegate_sp;
38517caa17caSRaphael Isemann   Debugger &m_debugger;
38527caa17caSRaphael Isemann   uint32_t m_stop_id;
38537caa17caSRaphael Isemann   FormatEntity::Entry m_format;
38547caa17caSRaphael Isemann };
38557caa17caSRaphael Isemann 
38567caa17caSRaphael Isemann class ValueObjectListDelegate : public WindowDelegate {
38577caa17caSRaphael Isemann public:
38589494c510SJonas Devlieghere   ValueObjectListDelegate() : m_rows() {}
38597caa17caSRaphael Isemann 
38607caa17caSRaphael Isemann   ValueObjectListDelegate(ValueObjectList &valobj_list)
38617caa17caSRaphael Isemann       : m_rows(), m_selected_row(nullptr), m_selected_row_idx(0),
38627caa17caSRaphael Isemann         m_first_visible_row(0), m_num_rows(0), m_max_x(0), m_max_y(0) {
38637caa17caSRaphael Isemann     SetValues(valobj_list);
38647caa17caSRaphael Isemann   }
38657caa17caSRaphael Isemann 
38667caa17caSRaphael Isemann   ~ValueObjectListDelegate() override = default;
38677caa17caSRaphael Isemann 
38687caa17caSRaphael Isemann   void SetValues(ValueObjectList &valobj_list) {
38697caa17caSRaphael Isemann     m_selected_row = nullptr;
38707caa17caSRaphael Isemann     m_selected_row_idx = 0;
38717caa17caSRaphael Isemann     m_first_visible_row = 0;
38727caa17caSRaphael Isemann     m_num_rows = 0;
38737caa17caSRaphael Isemann     m_rows.clear();
38747caa17caSRaphael Isemann     for (auto &valobj_sp : valobj_list.GetObjects())
38757caa17caSRaphael Isemann       m_rows.push_back(Row(valobj_sp, nullptr));
38767caa17caSRaphael Isemann   }
38777caa17caSRaphael Isemann 
38787caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
38797caa17caSRaphael Isemann     m_num_rows = 0;
38807caa17caSRaphael Isemann     m_min_x = 2;
38817caa17caSRaphael Isemann     m_min_y = 1;
38827caa17caSRaphael Isemann     m_max_x = window.GetWidth() - 1;
38837caa17caSRaphael Isemann     m_max_y = window.GetHeight() - 1;
38847caa17caSRaphael Isemann 
38857caa17caSRaphael Isemann     window.Erase();
38867caa17caSRaphael Isemann     window.DrawTitleBox(window.GetName());
38877caa17caSRaphael Isemann 
38887caa17caSRaphael Isemann     const int num_visible_rows = NumVisibleRows();
38897caa17caSRaphael Isemann     const int num_rows = CalculateTotalNumberRows(m_rows);
38907caa17caSRaphael Isemann 
38917caa17caSRaphael Isemann     // If we unexpanded while having something selected our total number of
38927caa17caSRaphael Isemann     // rows is less than the num visible rows, then make sure we show all the
38937caa17caSRaphael Isemann     // rows by setting the first visible row accordingly.
38947caa17caSRaphael Isemann     if (m_first_visible_row > 0 && num_rows < num_visible_rows)
38957caa17caSRaphael Isemann       m_first_visible_row = 0;
38967caa17caSRaphael Isemann 
38977caa17caSRaphael Isemann     // Make sure the selected row is always visible
38987caa17caSRaphael Isemann     if (m_selected_row_idx < m_first_visible_row)
38997caa17caSRaphael Isemann       m_first_visible_row = m_selected_row_idx;
39007caa17caSRaphael Isemann     else if (m_first_visible_row + num_visible_rows <= m_selected_row_idx)
39017caa17caSRaphael Isemann       m_first_visible_row = m_selected_row_idx - num_visible_rows + 1;
39027caa17caSRaphael Isemann 
39037caa17caSRaphael Isemann     DisplayRows(window, m_rows, g_options);
39047caa17caSRaphael Isemann 
39057caa17caSRaphael Isemann     // Get the selected row
39067caa17caSRaphael Isemann     m_selected_row = GetRowForRowIndex(m_selected_row_idx);
39077caa17caSRaphael Isemann     // Keep the cursor on the selected row so the highlight and the cursor are
39087caa17caSRaphael Isemann     // always on the same line
39097caa17caSRaphael Isemann     if (m_selected_row)
39107caa17caSRaphael Isemann       window.MoveCursor(m_selected_row->x, m_selected_row->y);
39117caa17caSRaphael Isemann 
39127caa17caSRaphael Isemann     return true; // Drawing handled
39137caa17caSRaphael Isemann   }
39147caa17caSRaphael Isemann 
39157caa17caSRaphael Isemann   KeyHelp *WindowDelegateGetKeyHelp() override {
39167caa17caSRaphael Isemann     static curses::KeyHelp g_source_view_key_help[] = {
39177caa17caSRaphael Isemann         {KEY_UP, "Select previous item"},
39187caa17caSRaphael Isemann         {KEY_DOWN, "Select next item"},
39197caa17caSRaphael Isemann         {KEY_RIGHT, "Expand selected item"},
39207caa17caSRaphael Isemann         {KEY_LEFT, "Unexpand selected item or select parent if not expanded"},
39217caa17caSRaphael Isemann         {KEY_PPAGE, "Page up"},
39227caa17caSRaphael Isemann         {KEY_NPAGE, "Page down"},
39237caa17caSRaphael Isemann         {'A', "Format as annotated address"},
39247caa17caSRaphael Isemann         {'b', "Format as binary"},
39257caa17caSRaphael Isemann         {'B', "Format as hex bytes with ASCII"},
39267caa17caSRaphael Isemann         {'c', "Format as character"},
39277caa17caSRaphael Isemann         {'d', "Format as a signed integer"},
39287caa17caSRaphael Isemann         {'D', "Format selected value using the default format for the type"},
39297caa17caSRaphael Isemann         {'f', "Format as float"},
39307caa17caSRaphael Isemann         {'h', "Show help dialog"},
39317caa17caSRaphael Isemann         {'i', "Format as instructions"},
39327caa17caSRaphael Isemann         {'o', "Format as octal"},
39337caa17caSRaphael Isemann         {'p', "Format as pointer"},
39347caa17caSRaphael Isemann         {'s', "Format as C string"},
39357caa17caSRaphael Isemann         {'t', "Toggle showing/hiding type names"},
39367caa17caSRaphael Isemann         {'u', "Format as an unsigned integer"},
39377caa17caSRaphael Isemann         {'x', "Format as hex"},
39387caa17caSRaphael Isemann         {'X', "Format as uppercase hex"},
39397caa17caSRaphael Isemann         {' ', "Toggle item expansion"},
39407caa17caSRaphael Isemann         {',', "Page up"},
39417caa17caSRaphael Isemann         {'.', "Page down"},
39427caa17caSRaphael Isemann         {'\0', nullptr}};
39437caa17caSRaphael Isemann     return g_source_view_key_help;
39447caa17caSRaphael Isemann   }
39457caa17caSRaphael Isemann 
39467caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
39477caa17caSRaphael Isemann     switch (c) {
39487caa17caSRaphael Isemann     case 'x':
39497caa17caSRaphael Isemann     case 'X':
39507caa17caSRaphael Isemann     case 'o':
39517caa17caSRaphael Isemann     case 's':
39527caa17caSRaphael Isemann     case 'u':
39537caa17caSRaphael Isemann     case 'd':
39547caa17caSRaphael Isemann     case 'D':
39557caa17caSRaphael Isemann     case 'i':
39567caa17caSRaphael Isemann     case 'A':
39577caa17caSRaphael Isemann     case 'p':
39587caa17caSRaphael Isemann     case 'c':
39597caa17caSRaphael Isemann     case 'b':
39607caa17caSRaphael Isemann     case 'B':
39617caa17caSRaphael Isemann     case 'f':
39627caa17caSRaphael Isemann       // Change the format for the currently selected item
39637caa17caSRaphael Isemann       if (m_selected_row) {
39647caa17caSRaphael Isemann         auto valobj_sp = m_selected_row->value.GetSP();
39657caa17caSRaphael Isemann         if (valobj_sp)
39667caa17caSRaphael Isemann           valobj_sp->SetFormat(FormatForChar(c));
39677caa17caSRaphael Isemann       }
39687caa17caSRaphael Isemann       return eKeyHandled;
39697caa17caSRaphael Isemann 
39707caa17caSRaphael Isemann     case 't':
39717caa17caSRaphael Isemann       // Toggle showing type names
39727caa17caSRaphael Isemann       g_options.show_types = !g_options.show_types;
39737caa17caSRaphael Isemann       return eKeyHandled;
39747caa17caSRaphael Isemann 
39757caa17caSRaphael Isemann     case ',':
39767caa17caSRaphael Isemann     case KEY_PPAGE:
39777caa17caSRaphael Isemann       // Page up key
39787caa17caSRaphael Isemann       if (m_first_visible_row > 0) {
39797caa17caSRaphael Isemann         if (static_cast<int>(m_first_visible_row) > m_max_y)
39807caa17caSRaphael Isemann           m_first_visible_row -= m_max_y;
39817caa17caSRaphael Isemann         else
39827caa17caSRaphael Isemann           m_first_visible_row = 0;
39837caa17caSRaphael Isemann         m_selected_row_idx = m_first_visible_row;
39847caa17caSRaphael Isemann       }
39857caa17caSRaphael Isemann       return eKeyHandled;
39867caa17caSRaphael Isemann 
39877caa17caSRaphael Isemann     case '.':
39887caa17caSRaphael Isemann     case KEY_NPAGE:
39897caa17caSRaphael Isemann       // Page down key
39907caa17caSRaphael Isemann       if (m_num_rows > static_cast<size_t>(m_max_y)) {
39917caa17caSRaphael Isemann         if (m_first_visible_row + m_max_y < m_num_rows) {
39927caa17caSRaphael Isemann           m_first_visible_row += m_max_y;
39937caa17caSRaphael Isemann           m_selected_row_idx = m_first_visible_row;
39947caa17caSRaphael Isemann         }
39957caa17caSRaphael Isemann       }
39967caa17caSRaphael Isemann       return eKeyHandled;
39977caa17caSRaphael Isemann 
39987caa17caSRaphael Isemann     case KEY_UP:
39997caa17caSRaphael Isemann       if (m_selected_row_idx > 0)
40007caa17caSRaphael Isemann         --m_selected_row_idx;
40017caa17caSRaphael Isemann       return eKeyHandled;
40027caa17caSRaphael Isemann 
40037caa17caSRaphael Isemann     case KEY_DOWN:
40047caa17caSRaphael Isemann       if (m_selected_row_idx + 1 < m_num_rows)
40057caa17caSRaphael Isemann         ++m_selected_row_idx;
40067caa17caSRaphael Isemann       return eKeyHandled;
40077caa17caSRaphael Isemann 
40087caa17caSRaphael Isemann     case KEY_RIGHT:
40097caa17caSRaphael Isemann       if (m_selected_row) {
40107caa17caSRaphael Isemann         if (!m_selected_row->expanded)
40117caa17caSRaphael Isemann           m_selected_row->Expand();
40127caa17caSRaphael Isemann       }
40137caa17caSRaphael Isemann       return eKeyHandled;
40147caa17caSRaphael Isemann 
40157caa17caSRaphael Isemann     case KEY_LEFT:
40167caa17caSRaphael Isemann       if (m_selected_row) {
40177caa17caSRaphael Isemann         if (m_selected_row->expanded)
40187caa17caSRaphael Isemann           m_selected_row->Unexpand();
40197caa17caSRaphael Isemann         else if (m_selected_row->parent)
40207caa17caSRaphael Isemann           m_selected_row_idx = m_selected_row->parent->row_idx;
40217caa17caSRaphael Isemann       }
40227caa17caSRaphael Isemann       return eKeyHandled;
40237caa17caSRaphael Isemann 
40247caa17caSRaphael Isemann     case ' ':
40257caa17caSRaphael Isemann       // Toggle expansion state when SPACE is pressed
40267caa17caSRaphael Isemann       if (m_selected_row) {
40277caa17caSRaphael Isemann         if (m_selected_row->expanded)
40287caa17caSRaphael Isemann           m_selected_row->Unexpand();
40297caa17caSRaphael Isemann         else
40307caa17caSRaphael Isemann           m_selected_row->Expand();
40317caa17caSRaphael Isemann       }
40327caa17caSRaphael Isemann       return eKeyHandled;
40337caa17caSRaphael Isemann 
40347caa17caSRaphael Isemann     case 'h':
40357caa17caSRaphael Isemann       window.CreateHelpSubwindow();
40367caa17caSRaphael Isemann       return eKeyHandled;
40377caa17caSRaphael Isemann 
40387caa17caSRaphael Isemann     default:
40397caa17caSRaphael Isemann       break;
40407caa17caSRaphael Isemann     }
40417caa17caSRaphael Isemann     return eKeyNotHandled;
40427caa17caSRaphael Isemann   }
40437caa17caSRaphael Isemann 
40447caa17caSRaphael Isemann protected:
40457caa17caSRaphael Isemann   std::vector<Row> m_rows;
40469494c510SJonas Devlieghere   Row *m_selected_row = nullptr;
40479494c510SJonas Devlieghere   uint32_t m_selected_row_idx = 0;
40489494c510SJonas Devlieghere   uint32_t m_first_visible_row = 0;
40499494c510SJonas Devlieghere   uint32_t m_num_rows = 0;
40507caa17caSRaphael Isemann   int m_min_x;
40517caa17caSRaphael Isemann   int m_min_y;
40529494c510SJonas Devlieghere   int m_max_x = 0;
40539494c510SJonas Devlieghere   int m_max_y = 0;
40547caa17caSRaphael Isemann 
40557caa17caSRaphael Isemann   static Format FormatForChar(int c) {
40567caa17caSRaphael Isemann     switch (c) {
40577caa17caSRaphael Isemann     case 'x':
40587caa17caSRaphael Isemann       return eFormatHex;
40597caa17caSRaphael Isemann     case 'X':
40607caa17caSRaphael Isemann       return eFormatHexUppercase;
40617caa17caSRaphael Isemann     case 'o':
40627caa17caSRaphael Isemann       return eFormatOctal;
40637caa17caSRaphael Isemann     case 's':
40647caa17caSRaphael Isemann       return eFormatCString;
40657caa17caSRaphael Isemann     case 'u':
40667caa17caSRaphael Isemann       return eFormatUnsigned;
40677caa17caSRaphael Isemann     case 'd':
40687caa17caSRaphael Isemann       return eFormatDecimal;
40697caa17caSRaphael Isemann     case 'D':
40707caa17caSRaphael Isemann       return eFormatDefault;
40717caa17caSRaphael Isemann     case 'i':
40727caa17caSRaphael Isemann       return eFormatInstruction;
40737caa17caSRaphael Isemann     case 'A':
40747caa17caSRaphael Isemann       return eFormatAddressInfo;
40757caa17caSRaphael Isemann     case 'p':
40767caa17caSRaphael Isemann       return eFormatPointer;
40777caa17caSRaphael Isemann     case 'c':
40787caa17caSRaphael Isemann       return eFormatChar;
40797caa17caSRaphael Isemann     case 'b':
40807caa17caSRaphael Isemann       return eFormatBinary;
40817caa17caSRaphael Isemann     case 'B':
40827caa17caSRaphael Isemann       return eFormatBytesWithASCII;
40837caa17caSRaphael Isemann     case 'f':
40847caa17caSRaphael Isemann       return eFormatFloat;
40857caa17caSRaphael Isemann     }
40867caa17caSRaphael Isemann     return eFormatDefault;
40877caa17caSRaphael Isemann   }
40887caa17caSRaphael Isemann 
40897caa17caSRaphael Isemann   bool DisplayRowObject(Window &window, Row &row, DisplayOptions &options,
40907caa17caSRaphael Isemann                         bool highlight, bool last_child) {
40917caa17caSRaphael Isemann     ValueObject *valobj = row.value.GetSP().get();
40927caa17caSRaphael Isemann 
40937caa17caSRaphael Isemann     if (valobj == nullptr)
40947caa17caSRaphael Isemann       return false;
40957caa17caSRaphael Isemann 
40967caa17caSRaphael Isemann     const char *type_name =
40977caa17caSRaphael Isemann         options.show_types ? valobj->GetTypeName().GetCString() : nullptr;
40987caa17caSRaphael Isemann     const char *name = valobj->GetName().GetCString();
40997caa17caSRaphael Isemann     const char *value = valobj->GetValueAsCString();
41007caa17caSRaphael Isemann     const char *summary = valobj->GetSummaryAsCString();
41017caa17caSRaphael Isemann 
41027caa17caSRaphael Isemann     window.MoveCursor(row.x, row.y);
41037caa17caSRaphael Isemann 
41047caa17caSRaphael Isemann     row.DrawTree(window);
41057caa17caSRaphael Isemann 
41067caa17caSRaphael Isemann     if (highlight)
41077caa17caSRaphael Isemann       window.AttributeOn(A_REVERSE);
41087caa17caSRaphael Isemann 
41097caa17caSRaphael Isemann     if (type_name && type_name[0])
4110fc0e8fb7SLuboš Luňák       window.PrintfTruncated(1, "(%s) ", type_name);
41117caa17caSRaphael Isemann 
41127caa17caSRaphael Isemann     if (name && name[0])
4113fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, name);
41147caa17caSRaphael Isemann 
41157caa17caSRaphael Isemann     attr_t changd_attr = 0;
41167caa17caSRaphael Isemann     if (valobj->GetValueDidChange())
41174a8e4b5cSLuboš Luňák       changd_attr = COLOR_PAIR(RedOnBlack) | A_BOLD;
41187caa17caSRaphael Isemann 
41197caa17caSRaphael Isemann     if (value && value[0]) {
4120fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, " = ");
41217caa17caSRaphael Isemann       if (changd_attr)
41227caa17caSRaphael Isemann         window.AttributeOn(changd_attr);
4123fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, value);
41247caa17caSRaphael Isemann       if (changd_attr)
41257caa17caSRaphael Isemann         window.AttributeOff(changd_attr);
41267caa17caSRaphael Isemann     }
41277caa17caSRaphael Isemann 
41287caa17caSRaphael Isemann     if (summary && summary[0]) {
4129fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, " ");
41307caa17caSRaphael Isemann       if (changd_attr)
41317caa17caSRaphael Isemann         window.AttributeOn(changd_attr);
4132fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, summary);
41337caa17caSRaphael Isemann       if (changd_attr)
41347caa17caSRaphael Isemann         window.AttributeOff(changd_attr);
41357caa17caSRaphael Isemann     }
41367caa17caSRaphael Isemann 
41377caa17caSRaphael Isemann     if (highlight)
41387caa17caSRaphael Isemann       window.AttributeOff(A_REVERSE);
41397caa17caSRaphael Isemann 
41407caa17caSRaphael Isemann     return true;
41417caa17caSRaphael Isemann   }
41427caa17caSRaphael Isemann 
41437caa17caSRaphael Isemann   void DisplayRows(Window &window, std::vector<Row> &rows,
41447caa17caSRaphael Isemann                    DisplayOptions &options) {
41457caa17caSRaphael Isemann     // >   0x25B7
41467caa17caSRaphael Isemann     // \/  0x25BD
41477caa17caSRaphael Isemann 
41487caa17caSRaphael Isemann     bool window_is_active = window.IsActive();
41497caa17caSRaphael Isemann     for (auto &row : rows) {
41507caa17caSRaphael Isemann       const bool last_child = row.parent && &rows[rows.size() - 1] == &row;
41517caa17caSRaphael Isemann       // Save the row index in each Row structure
41527caa17caSRaphael Isemann       row.row_idx = m_num_rows;
41537caa17caSRaphael Isemann       if ((m_num_rows >= m_first_visible_row) &&
41547caa17caSRaphael Isemann           ((m_num_rows - m_first_visible_row) <
41557caa17caSRaphael Isemann            static_cast<size_t>(NumVisibleRows()))) {
41567caa17caSRaphael Isemann         row.x = m_min_x;
41577caa17caSRaphael Isemann         row.y = m_num_rows - m_first_visible_row + 1;
41587caa17caSRaphael Isemann         if (DisplayRowObject(window, row, options,
41597caa17caSRaphael Isemann                              window_is_active &&
41607caa17caSRaphael Isemann                                  m_num_rows == m_selected_row_idx,
41617caa17caSRaphael Isemann                              last_child)) {
41627caa17caSRaphael Isemann           ++m_num_rows;
41637caa17caSRaphael Isemann         } else {
41647caa17caSRaphael Isemann           row.x = 0;
41657caa17caSRaphael Isemann           row.y = 0;
41667caa17caSRaphael Isemann         }
41677caa17caSRaphael Isemann       } else {
41687caa17caSRaphael Isemann         row.x = 0;
41697caa17caSRaphael Isemann         row.y = 0;
41707caa17caSRaphael Isemann         ++m_num_rows;
41717caa17caSRaphael Isemann       }
41727caa17caSRaphael Isemann 
41737caa17caSRaphael Isemann       auto &children = row.GetChildren();
41747caa17caSRaphael Isemann       if (row.expanded && !children.empty()) {
41757caa17caSRaphael Isemann         DisplayRows(window, children, options);
41767caa17caSRaphael Isemann       }
41777caa17caSRaphael Isemann     }
41787caa17caSRaphael Isemann   }
41797caa17caSRaphael Isemann 
41807caa17caSRaphael Isemann   int CalculateTotalNumberRows(std::vector<Row> &rows) {
41817caa17caSRaphael Isemann     int row_count = 0;
41827caa17caSRaphael Isemann     for (auto &row : rows) {
41837caa17caSRaphael Isemann       ++row_count;
41847caa17caSRaphael Isemann       if (row.expanded)
41857caa17caSRaphael Isemann         row_count += CalculateTotalNumberRows(row.GetChildren());
41867caa17caSRaphael Isemann     }
41877caa17caSRaphael Isemann     return row_count;
41887caa17caSRaphael Isemann   }
41897caa17caSRaphael Isemann 
41907caa17caSRaphael Isemann   static Row *GetRowForRowIndexImpl(std::vector<Row> &rows, size_t &row_index) {
41917caa17caSRaphael Isemann     for (auto &row : rows) {
41927caa17caSRaphael Isemann       if (row_index == 0)
41937caa17caSRaphael Isemann         return &row;
41947caa17caSRaphael Isemann       else {
41957caa17caSRaphael Isemann         --row_index;
41967caa17caSRaphael Isemann         auto &children = row.GetChildren();
41977caa17caSRaphael Isemann         if (row.expanded && !children.empty()) {
41987caa17caSRaphael Isemann           Row *result = GetRowForRowIndexImpl(children, row_index);
41997caa17caSRaphael Isemann           if (result)
42007caa17caSRaphael Isemann             return result;
42017caa17caSRaphael Isemann         }
42027caa17caSRaphael Isemann       }
42037caa17caSRaphael Isemann     }
42047caa17caSRaphael Isemann     return nullptr;
42057caa17caSRaphael Isemann   }
42067caa17caSRaphael Isemann 
42077caa17caSRaphael Isemann   Row *GetRowForRowIndex(size_t row_index) {
42087caa17caSRaphael Isemann     return GetRowForRowIndexImpl(m_rows, row_index);
42097caa17caSRaphael Isemann   }
42107caa17caSRaphael Isemann 
42117caa17caSRaphael Isemann   int NumVisibleRows() const { return m_max_y - m_min_y; }
42127caa17caSRaphael Isemann 
42137caa17caSRaphael Isemann   static DisplayOptions g_options;
42147caa17caSRaphael Isemann };
42157caa17caSRaphael Isemann 
42167caa17caSRaphael Isemann class FrameVariablesWindowDelegate : public ValueObjectListDelegate {
42177caa17caSRaphael Isemann public:
42187caa17caSRaphael Isemann   FrameVariablesWindowDelegate(Debugger &debugger)
42197caa17caSRaphael Isemann       : ValueObjectListDelegate(), m_debugger(debugger),
42207caa17caSRaphael Isemann         m_frame_block(nullptr) {}
42217caa17caSRaphael Isemann 
42227caa17caSRaphael Isemann   ~FrameVariablesWindowDelegate() override = default;
42237caa17caSRaphael Isemann 
42247caa17caSRaphael Isemann   const char *WindowDelegateGetHelpText() override {
42257caa17caSRaphael Isemann     return "Frame variable window keyboard shortcuts:";
42267caa17caSRaphael Isemann   }
42277caa17caSRaphael Isemann 
42287caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
42297caa17caSRaphael Isemann     ExecutionContext exe_ctx(
42307caa17caSRaphael Isemann         m_debugger.GetCommandInterpreter().GetExecutionContext());
42317caa17caSRaphael Isemann     Process *process = exe_ctx.GetProcessPtr();
42327caa17caSRaphael Isemann     Block *frame_block = nullptr;
42337caa17caSRaphael Isemann     StackFrame *frame = nullptr;
42347caa17caSRaphael Isemann 
42357caa17caSRaphael Isemann     if (process) {
42367caa17caSRaphael Isemann       StateType state = process->GetState();
42377caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
42387caa17caSRaphael Isemann         frame = exe_ctx.GetFramePtr();
42397caa17caSRaphael Isemann         if (frame)
42407caa17caSRaphael Isemann           frame_block = frame->GetFrameBlock();
42417caa17caSRaphael Isemann       } else if (StateIsRunningState(state)) {
42427caa17caSRaphael Isemann         return true; // Don't do any updating when we are running
42437caa17caSRaphael Isemann       }
42447caa17caSRaphael Isemann     }
42457caa17caSRaphael Isemann 
42467caa17caSRaphael Isemann     ValueObjectList local_values;
42477caa17caSRaphael Isemann     if (frame_block) {
42487caa17caSRaphael Isemann       // Only update the variables if they have changed
42497caa17caSRaphael Isemann       if (m_frame_block != frame_block) {
42507caa17caSRaphael Isemann         m_frame_block = frame_block;
42517caa17caSRaphael Isemann 
42527caa17caSRaphael Isemann         VariableList *locals = frame->GetVariableList(true);
42537caa17caSRaphael Isemann         if (locals) {
42547caa17caSRaphael Isemann           const DynamicValueType use_dynamic = eDynamicDontRunTarget;
42557caa17caSRaphael Isemann           for (const VariableSP &local_sp : *locals) {
42567caa17caSRaphael Isemann             ValueObjectSP value_sp =
42577caa17caSRaphael Isemann                 frame->GetValueObjectForFrameVariable(local_sp, use_dynamic);
42587caa17caSRaphael Isemann             if (value_sp) {
42597caa17caSRaphael Isemann               ValueObjectSP synthetic_value_sp = value_sp->GetSyntheticValue();
42607caa17caSRaphael Isemann               if (synthetic_value_sp)
42617caa17caSRaphael Isemann                 local_values.Append(synthetic_value_sp);
42627caa17caSRaphael Isemann               else
42637caa17caSRaphael Isemann                 local_values.Append(value_sp);
42647caa17caSRaphael Isemann             }
42657caa17caSRaphael Isemann           }
42667caa17caSRaphael Isemann           // Update the values
42677caa17caSRaphael Isemann           SetValues(local_values);
42687caa17caSRaphael Isemann         }
42697caa17caSRaphael Isemann       }
42707caa17caSRaphael Isemann     } else {
42717caa17caSRaphael Isemann       m_frame_block = nullptr;
42727caa17caSRaphael Isemann       // Update the values with an empty list if there is no frame
42737caa17caSRaphael Isemann       SetValues(local_values);
42747caa17caSRaphael Isemann     }
42757caa17caSRaphael Isemann 
42767caa17caSRaphael Isemann     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
42777caa17caSRaphael Isemann   }
42787caa17caSRaphael Isemann 
42797caa17caSRaphael Isemann protected:
42807caa17caSRaphael Isemann   Debugger &m_debugger;
42817caa17caSRaphael Isemann   Block *m_frame_block;
42827caa17caSRaphael Isemann };
42837caa17caSRaphael Isemann 
42847caa17caSRaphael Isemann class RegistersWindowDelegate : public ValueObjectListDelegate {
42857caa17caSRaphael Isemann public:
42867caa17caSRaphael Isemann   RegistersWindowDelegate(Debugger &debugger)
42877caa17caSRaphael Isemann       : ValueObjectListDelegate(), m_debugger(debugger) {}
42887caa17caSRaphael Isemann 
42897caa17caSRaphael Isemann   ~RegistersWindowDelegate() override = default;
42907caa17caSRaphael Isemann 
42917caa17caSRaphael Isemann   const char *WindowDelegateGetHelpText() override {
42927caa17caSRaphael Isemann     return "Register window keyboard shortcuts:";
42937caa17caSRaphael Isemann   }
42947caa17caSRaphael Isemann 
42957caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
42967caa17caSRaphael Isemann     ExecutionContext exe_ctx(
42977caa17caSRaphael Isemann         m_debugger.GetCommandInterpreter().GetExecutionContext());
42987caa17caSRaphael Isemann     StackFrame *frame = exe_ctx.GetFramePtr();
42997caa17caSRaphael Isemann 
43007caa17caSRaphael Isemann     ValueObjectList value_list;
43017caa17caSRaphael Isemann     if (frame) {
43027caa17caSRaphael Isemann       if (frame->GetStackID() != m_stack_id) {
43037caa17caSRaphael Isemann         m_stack_id = frame->GetStackID();
43047caa17caSRaphael Isemann         RegisterContextSP reg_ctx(frame->GetRegisterContext());
43057caa17caSRaphael Isemann         if (reg_ctx) {
43067caa17caSRaphael Isemann           const uint32_t num_sets = reg_ctx->GetRegisterSetCount();
43077caa17caSRaphael Isemann           for (uint32_t set_idx = 0; set_idx < num_sets; ++set_idx) {
43087caa17caSRaphael Isemann             value_list.Append(
43097caa17caSRaphael Isemann                 ValueObjectRegisterSet::Create(frame, reg_ctx, set_idx));
43107caa17caSRaphael Isemann           }
43117caa17caSRaphael Isemann         }
43127caa17caSRaphael Isemann         SetValues(value_list);
43137caa17caSRaphael Isemann       }
43147caa17caSRaphael Isemann     } else {
43157caa17caSRaphael Isemann       Process *process = exe_ctx.GetProcessPtr();
43167caa17caSRaphael Isemann       if (process && process->IsAlive())
43177caa17caSRaphael Isemann         return true; // Don't do any updating if we are running
43187caa17caSRaphael Isemann       else {
43197caa17caSRaphael Isemann         // Update the values with an empty list if there is no process or the
43207caa17caSRaphael Isemann         // process isn't alive anymore
43217caa17caSRaphael Isemann         SetValues(value_list);
43227caa17caSRaphael Isemann       }
43237caa17caSRaphael Isemann     }
43247caa17caSRaphael Isemann     return ValueObjectListDelegate::WindowDelegateDraw(window, force);
43257caa17caSRaphael Isemann   }
43267caa17caSRaphael Isemann 
43277caa17caSRaphael Isemann protected:
43287caa17caSRaphael Isemann   Debugger &m_debugger;
43297caa17caSRaphael Isemann   StackID m_stack_id;
43307caa17caSRaphael Isemann };
43317caa17caSRaphael Isemann 
43327caa17caSRaphael Isemann static const char *CursesKeyToCString(int ch) {
43337caa17caSRaphael Isemann   static char g_desc[32];
43347caa17caSRaphael Isemann   if (ch >= KEY_F0 && ch < KEY_F0 + 64) {
43357caa17caSRaphael Isemann     snprintf(g_desc, sizeof(g_desc), "F%u", ch - KEY_F0);
43367caa17caSRaphael Isemann     return g_desc;
43377caa17caSRaphael Isemann   }
43387caa17caSRaphael Isemann   switch (ch) {
43397caa17caSRaphael Isemann   case KEY_DOWN:
43407caa17caSRaphael Isemann     return "down";
43417caa17caSRaphael Isemann   case KEY_UP:
43427caa17caSRaphael Isemann     return "up";
43437caa17caSRaphael Isemann   case KEY_LEFT:
43447caa17caSRaphael Isemann     return "left";
43457caa17caSRaphael Isemann   case KEY_RIGHT:
43467caa17caSRaphael Isemann     return "right";
43477caa17caSRaphael Isemann   case KEY_HOME:
43487caa17caSRaphael Isemann     return "home";
43497caa17caSRaphael Isemann   case KEY_BACKSPACE:
43507caa17caSRaphael Isemann     return "backspace";
43517caa17caSRaphael Isemann   case KEY_DL:
43527caa17caSRaphael Isemann     return "delete-line";
43537caa17caSRaphael Isemann   case KEY_IL:
43547caa17caSRaphael Isemann     return "insert-line";
43557caa17caSRaphael Isemann   case KEY_DC:
43567caa17caSRaphael Isemann     return "delete-char";
43577caa17caSRaphael Isemann   case KEY_IC:
43587caa17caSRaphael Isemann     return "insert-char";
43597caa17caSRaphael Isemann   case KEY_CLEAR:
43607caa17caSRaphael Isemann     return "clear";
43617caa17caSRaphael Isemann   case KEY_EOS:
43627caa17caSRaphael Isemann     return "clear-to-eos";
43637caa17caSRaphael Isemann   case KEY_EOL:
43647caa17caSRaphael Isemann     return "clear-to-eol";
43657caa17caSRaphael Isemann   case KEY_SF:
43667caa17caSRaphael Isemann     return "scroll-forward";
43677caa17caSRaphael Isemann   case KEY_SR:
43687caa17caSRaphael Isemann     return "scroll-backward";
43697caa17caSRaphael Isemann   case KEY_NPAGE:
43707caa17caSRaphael Isemann     return "page-down";
43717caa17caSRaphael Isemann   case KEY_PPAGE:
43727caa17caSRaphael Isemann     return "page-up";
43737caa17caSRaphael Isemann   case KEY_STAB:
43747caa17caSRaphael Isemann     return "set-tab";
43757caa17caSRaphael Isemann   case KEY_CTAB:
43767caa17caSRaphael Isemann     return "clear-tab";
43777caa17caSRaphael Isemann   case KEY_CATAB:
43787caa17caSRaphael Isemann     return "clear-all-tabs";
43797caa17caSRaphael Isemann   case KEY_ENTER:
43807caa17caSRaphael Isemann     return "enter";
43817caa17caSRaphael Isemann   case KEY_PRINT:
43827caa17caSRaphael Isemann     return "print";
43837caa17caSRaphael Isemann   case KEY_LL:
43847caa17caSRaphael Isemann     return "lower-left key";
43857caa17caSRaphael Isemann   case KEY_A1:
43867caa17caSRaphael Isemann     return "upper left of keypad";
43877caa17caSRaphael Isemann   case KEY_A3:
43887caa17caSRaphael Isemann     return "upper right of keypad";
43897caa17caSRaphael Isemann   case KEY_B2:
43907caa17caSRaphael Isemann     return "center of keypad";
43917caa17caSRaphael Isemann   case KEY_C1:
43927caa17caSRaphael Isemann     return "lower left of keypad";
43937caa17caSRaphael Isemann   case KEY_C3:
43947caa17caSRaphael Isemann     return "lower right of keypad";
43957caa17caSRaphael Isemann   case KEY_BTAB:
43967caa17caSRaphael Isemann     return "back-tab key";
43977caa17caSRaphael Isemann   case KEY_BEG:
43987caa17caSRaphael Isemann     return "begin key";
43997caa17caSRaphael Isemann   case KEY_CANCEL:
44007caa17caSRaphael Isemann     return "cancel key";
44017caa17caSRaphael Isemann   case KEY_CLOSE:
44027caa17caSRaphael Isemann     return "close key";
44037caa17caSRaphael Isemann   case KEY_COMMAND:
44047caa17caSRaphael Isemann     return "command key";
44057caa17caSRaphael Isemann   case KEY_COPY:
44067caa17caSRaphael Isemann     return "copy key";
44077caa17caSRaphael Isemann   case KEY_CREATE:
44087caa17caSRaphael Isemann     return "create key";
44097caa17caSRaphael Isemann   case KEY_END:
44107caa17caSRaphael Isemann     return "end key";
44117caa17caSRaphael Isemann   case KEY_EXIT:
44127caa17caSRaphael Isemann     return "exit key";
44137caa17caSRaphael Isemann   case KEY_FIND:
44147caa17caSRaphael Isemann     return "find key";
44157caa17caSRaphael Isemann   case KEY_HELP:
44167caa17caSRaphael Isemann     return "help key";
44177caa17caSRaphael Isemann   case KEY_MARK:
44187caa17caSRaphael Isemann     return "mark key";
44197caa17caSRaphael Isemann   case KEY_MESSAGE:
44207caa17caSRaphael Isemann     return "message key";
44217caa17caSRaphael Isemann   case KEY_MOVE:
44227caa17caSRaphael Isemann     return "move key";
44237caa17caSRaphael Isemann   case KEY_NEXT:
44247caa17caSRaphael Isemann     return "next key";
44257caa17caSRaphael Isemann   case KEY_OPEN:
44267caa17caSRaphael Isemann     return "open key";
44277caa17caSRaphael Isemann   case KEY_OPTIONS:
44287caa17caSRaphael Isemann     return "options key";
44297caa17caSRaphael Isemann   case KEY_PREVIOUS:
44307caa17caSRaphael Isemann     return "previous key";
44317caa17caSRaphael Isemann   case KEY_REDO:
44327caa17caSRaphael Isemann     return "redo key";
44337caa17caSRaphael Isemann   case KEY_REFERENCE:
44347caa17caSRaphael Isemann     return "reference key";
44357caa17caSRaphael Isemann   case KEY_REFRESH:
44367caa17caSRaphael Isemann     return "refresh key";
44377caa17caSRaphael Isemann   case KEY_REPLACE:
44387caa17caSRaphael Isemann     return "replace key";
44397caa17caSRaphael Isemann   case KEY_RESTART:
44407caa17caSRaphael Isemann     return "restart key";
44417caa17caSRaphael Isemann   case KEY_RESUME:
44427caa17caSRaphael Isemann     return "resume key";
44437caa17caSRaphael Isemann   case KEY_SAVE:
44447caa17caSRaphael Isemann     return "save key";
44457caa17caSRaphael Isemann   case KEY_SBEG:
44467caa17caSRaphael Isemann     return "shifted begin key";
44477caa17caSRaphael Isemann   case KEY_SCANCEL:
44487caa17caSRaphael Isemann     return "shifted cancel key";
44497caa17caSRaphael Isemann   case KEY_SCOMMAND:
44507caa17caSRaphael Isemann     return "shifted command key";
44517caa17caSRaphael Isemann   case KEY_SCOPY:
44527caa17caSRaphael Isemann     return "shifted copy key";
44537caa17caSRaphael Isemann   case KEY_SCREATE:
44547caa17caSRaphael Isemann     return "shifted create key";
44557caa17caSRaphael Isemann   case KEY_SDC:
44567caa17caSRaphael Isemann     return "shifted delete-character key";
44577caa17caSRaphael Isemann   case KEY_SDL:
44587caa17caSRaphael Isemann     return "shifted delete-line key";
44597caa17caSRaphael Isemann   case KEY_SELECT:
44607caa17caSRaphael Isemann     return "select key";
44617caa17caSRaphael Isemann   case KEY_SEND:
44627caa17caSRaphael Isemann     return "shifted end key";
44637caa17caSRaphael Isemann   case KEY_SEOL:
44647caa17caSRaphael Isemann     return "shifted clear-to-end-of-line key";
44657caa17caSRaphael Isemann   case KEY_SEXIT:
44667caa17caSRaphael Isemann     return "shifted exit key";
44677caa17caSRaphael Isemann   case KEY_SFIND:
44687caa17caSRaphael Isemann     return "shifted find key";
44697caa17caSRaphael Isemann   case KEY_SHELP:
44707caa17caSRaphael Isemann     return "shifted help key";
44717caa17caSRaphael Isemann   case KEY_SHOME:
44727caa17caSRaphael Isemann     return "shifted home key";
44737caa17caSRaphael Isemann   case KEY_SIC:
44747caa17caSRaphael Isemann     return "shifted insert-character key";
44757caa17caSRaphael Isemann   case KEY_SLEFT:
44767caa17caSRaphael Isemann     return "shifted left-arrow key";
44777caa17caSRaphael Isemann   case KEY_SMESSAGE:
44787caa17caSRaphael Isemann     return "shifted message key";
44797caa17caSRaphael Isemann   case KEY_SMOVE:
44807caa17caSRaphael Isemann     return "shifted move key";
44817caa17caSRaphael Isemann   case KEY_SNEXT:
44827caa17caSRaphael Isemann     return "shifted next key";
44837caa17caSRaphael Isemann   case KEY_SOPTIONS:
44847caa17caSRaphael Isemann     return "shifted options key";
44857caa17caSRaphael Isemann   case KEY_SPREVIOUS:
44867caa17caSRaphael Isemann     return "shifted previous key";
44877caa17caSRaphael Isemann   case KEY_SPRINT:
44887caa17caSRaphael Isemann     return "shifted print key";
44897caa17caSRaphael Isemann   case KEY_SREDO:
44907caa17caSRaphael Isemann     return "shifted redo key";
44917caa17caSRaphael Isemann   case KEY_SREPLACE:
44927caa17caSRaphael Isemann     return "shifted replace key";
44937caa17caSRaphael Isemann   case KEY_SRIGHT:
44947caa17caSRaphael Isemann     return "shifted right-arrow key";
44957caa17caSRaphael Isemann   case KEY_SRSUME:
44967caa17caSRaphael Isemann     return "shifted resume key";
44977caa17caSRaphael Isemann   case KEY_SSAVE:
44987caa17caSRaphael Isemann     return "shifted save key";
44997caa17caSRaphael Isemann   case KEY_SSUSPEND:
45007caa17caSRaphael Isemann     return "shifted suspend key";
45017caa17caSRaphael Isemann   case KEY_SUNDO:
45027caa17caSRaphael Isemann     return "shifted undo key";
45037caa17caSRaphael Isemann   case KEY_SUSPEND:
45047caa17caSRaphael Isemann     return "suspend key";
45057caa17caSRaphael Isemann   case KEY_UNDO:
45067caa17caSRaphael Isemann     return "undo key";
45077caa17caSRaphael Isemann   case KEY_MOUSE:
45087caa17caSRaphael Isemann     return "Mouse event has occurred";
45097caa17caSRaphael Isemann   case KEY_RESIZE:
45107caa17caSRaphael Isemann     return "Terminal resize event";
45117caa17caSRaphael Isemann #ifdef KEY_EVENT
45127caa17caSRaphael Isemann   case KEY_EVENT:
45137caa17caSRaphael Isemann     return "We were interrupted by an event";
45147caa17caSRaphael Isemann #endif
45157caa17caSRaphael Isemann   case KEY_RETURN:
45167caa17caSRaphael Isemann     return "return";
45177caa17caSRaphael Isemann   case ' ':
45187caa17caSRaphael Isemann     return "space";
45197caa17caSRaphael Isemann   case '\t':
45207caa17caSRaphael Isemann     return "tab";
45217caa17caSRaphael Isemann   case KEY_ESCAPE:
45227caa17caSRaphael Isemann     return "escape";
45237caa17caSRaphael Isemann   default:
4524f5eaa2afSRaphael Isemann     if (llvm::isPrint(ch))
45257caa17caSRaphael Isemann       snprintf(g_desc, sizeof(g_desc), "%c", ch);
45267caa17caSRaphael Isemann     else
45277caa17caSRaphael Isemann       snprintf(g_desc, sizeof(g_desc), "\\x%2.2x", ch);
45287caa17caSRaphael Isemann     return g_desc;
45297caa17caSRaphael Isemann   }
45307caa17caSRaphael Isemann   return nullptr;
45317caa17caSRaphael Isemann }
45327caa17caSRaphael Isemann 
45337caa17caSRaphael Isemann HelpDialogDelegate::HelpDialogDelegate(const char *text,
45347caa17caSRaphael Isemann                                        KeyHelp *key_help_array)
45357caa17caSRaphael Isemann     : m_text(), m_first_visible_line(0) {
45367caa17caSRaphael Isemann   if (text && text[0]) {
45377caa17caSRaphael Isemann     m_text.SplitIntoLines(text);
45387caa17caSRaphael Isemann     m_text.AppendString("");
45397caa17caSRaphael Isemann   }
45407caa17caSRaphael Isemann   if (key_help_array) {
45417caa17caSRaphael Isemann     for (KeyHelp *key = key_help_array; key->ch; ++key) {
45427caa17caSRaphael Isemann       StreamString key_description;
45437caa17caSRaphael Isemann       key_description.Printf("%10s - %s", CursesKeyToCString(key->ch),
45447caa17caSRaphael Isemann                              key->description);
45457caa17caSRaphael Isemann       m_text.AppendString(key_description.GetString());
45467caa17caSRaphael Isemann     }
45477caa17caSRaphael Isemann   }
45487caa17caSRaphael Isemann }
45497caa17caSRaphael Isemann 
45507caa17caSRaphael Isemann HelpDialogDelegate::~HelpDialogDelegate() = default;
45517caa17caSRaphael Isemann 
45527caa17caSRaphael Isemann bool HelpDialogDelegate::WindowDelegateDraw(Window &window, bool force) {
45537caa17caSRaphael Isemann   window.Erase();
45547caa17caSRaphael Isemann   const int window_height = window.GetHeight();
45557caa17caSRaphael Isemann   int x = 2;
45567caa17caSRaphael Isemann   int y = 1;
45577caa17caSRaphael Isemann   const int min_y = y;
45587caa17caSRaphael Isemann   const int max_y = window_height - 1 - y;
45597caa17caSRaphael Isemann   const size_t num_visible_lines = max_y - min_y + 1;
45607caa17caSRaphael Isemann   const size_t num_lines = m_text.GetSize();
45617caa17caSRaphael Isemann   const char *bottom_message;
45627caa17caSRaphael Isemann   if (num_lines <= num_visible_lines)
45637caa17caSRaphael Isemann     bottom_message = "Press any key to exit";
45647caa17caSRaphael Isemann   else
45657caa17caSRaphael Isemann     bottom_message = "Use arrows to scroll, any other key to exit";
45667caa17caSRaphael Isemann   window.DrawTitleBox(window.GetName(), bottom_message);
45677caa17caSRaphael Isemann   while (y <= max_y) {
45687caa17caSRaphael Isemann     window.MoveCursor(x, y);
45697caa17caSRaphael Isemann     window.PutCStringTruncated(
4570fc0e8fb7SLuboš Luňák         1, m_text.GetStringAtIndex(m_first_visible_line + y - min_y));
45717caa17caSRaphael Isemann     ++y;
45727caa17caSRaphael Isemann   }
45737caa17caSRaphael Isemann   return true;
45747caa17caSRaphael Isemann }
45757caa17caSRaphael Isemann 
45767caa17caSRaphael Isemann HandleCharResult HelpDialogDelegate::WindowDelegateHandleChar(Window &window,
45777caa17caSRaphael Isemann                                                               int key) {
45787caa17caSRaphael Isemann   bool done = false;
45797caa17caSRaphael Isemann   const size_t num_lines = m_text.GetSize();
45807caa17caSRaphael Isemann   const size_t num_visible_lines = window.GetHeight() - 2;
45817caa17caSRaphael Isemann 
45827caa17caSRaphael Isemann   if (num_lines <= num_visible_lines) {
45837caa17caSRaphael Isemann     done = true;
45847caa17caSRaphael Isemann     // If we have all lines visible and don't need scrolling, then any key
45857caa17caSRaphael Isemann     // press will cause us to exit
45867caa17caSRaphael Isemann   } else {
45877caa17caSRaphael Isemann     switch (key) {
45887caa17caSRaphael Isemann     case KEY_UP:
45897caa17caSRaphael Isemann       if (m_first_visible_line > 0)
45907caa17caSRaphael Isemann         --m_first_visible_line;
45917caa17caSRaphael Isemann       break;
45927caa17caSRaphael Isemann 
45937caa17caSRaphael Isemann     case KEY_DOWN:
45947caa17caSRaphael Isemann       if (m_first_visible_line + num_visible_lines < num_lines)
45957caa17caSRaphael Isemann         ++m_first_visible_line;
45967caa17caSRaphael Isemann       break;
45977caa17caSRaphael Isemann 
45987caa17caSRaphael Isemann     case KEY_PPAGE:
45997caa17caSRaphael Isemann     case ',':
46007caa17caSRaphael Isemann       if (m_first_visible_line > 0) {
46017caa17caSRaphael Isemann         if (static_cast<size_t>(m_first_visible_line) >= num_visible_lines)
46027caa17caSRaphael Isemann           m_first_visible_line -= num_visible_lines;
46037caa17caSRaphael Isemann         else
46047caa17caSRaphael Isemann           m_first_visible_line = 0;
46057caa17caSRaphael Isemann       }
46067caa17caSRaphael Isemann       break;
46077caa17caSRaphael Isemann 
46087caa17caSRaphael Isemann     case KEY_NPAGE:
46097caa17caSRaphael Isemann     case '.':
46107caa17caSRaphael Isemann       if (m_first_visible_line + num_visible_lines < num_lines) {
46117caa17caSRaphael Isemann         m_first_visible_line += num_visible_lines;
46127caa17caSRaphael Isemann         if (static_cast<size_t>(m_first_visible_line) > num_lines)
46137caa17caSRaphael Isemann           m_first_visible_line = num_lines - num_visible_lines;
46147caa17caSRaphael Isemann       }
46157caa17caSRaphael Isemann       break;
46167caa17caSRaphael Isemann 
46177caa17caSRaphael Isemann     default:
46187caa17caSRaphael Isemann       done = true;
46197caa17caSRaphael Isemann       break;
46207caa17caSRaphael Isemann     }
46217caa17caSRaphael Isemann   }
46227caa17caSRaphael Isemann   if (done)
46237caa17caSRaphael Isemann     window.GetParent()->RemoveSubWindow(&window);
46247caa17caSRaphael Isemann   return eKeyHandled;
46257caa17caSRaphael Isemann }
46267caa17caSRaphael Isemann 
46277caa17caSRaphael Isemann class ApplicationDelegate : public WindowDelegate, public MenuDelegate {
46287caa17caSRaphael Isemann public:
46297caa17caSRaphael Isemann   enum {
46307caa17caSRaphael Isemann     eMenuID_LLDB = 1,
46317caa17caSRaphael Isemann     eMenuID_LLDBAbout,
46327caa17caSRaphael Isemann     eMenuID_LLDBExit,
46337caa17caSRaphael Isemann 
46347caa17caSRaphael Isemann     eMenuID_Target,
46357caa17caSRaphael Isemann     eMenuID_TargetCreate,
46367caa17caSRaphael Isemann     eMenuID_TargetDelete,
46377caa17caSRaphael Isemann 
46387caa17caSRaphael Isemann     eMenuID_Process,
46397caa17caSRaphael Isemann     eMenuID_ProcessAttach,
464047d2c7cdSLuboš Luňák     eMenuID_ProcessDetachResume,
464147d2c7cdSLuboš Luňák     eMenuID_ProcessDetachSuspended,
46427caa17caSRaphael Isemann     eMenuID_ProcessLaunch,
46437caa17caSRaphael Isemann     eMenuID_ProcessContinue,
46447caa17caSRaphael Isemann     eMenuID_ProcessHalt,
46457caa17caSRaphael Isemann     eMenuID_ProcessKill,
46467caa17caSRaphael Isemann 
46477caa17caSRaphael Isemann     eMenuID_Thread,
46487caa17caSRaphael Isemann     eMenuID_ThreadStepIn,
46497caa17caSRaphael Isemann     eMenuID_ThreadStepOver,
46507caa17caSRaphael Isemann     eMenuID_ThreadStepOut,
46517caa17caSRaphael Isemann 
46527caa17caSRaphael Isemann     eMenuID_View,
46537caa17caSRaphael Isemann     eMenuID_ViewBacktrace,
46547caa17caSRaphael Isemann     eMenuID_ViewRegisters,
46557caa17caSRaphael Isemann     eMenuID_ViewSource,
46567caa17caSRaphael Isemann     eMenuID_ViewVariables,
46577caa17caSRaphael Isemann 
46587caa17caSRaphael Isemann     eMenuID_Help,
46597caa17caSRaphael Isemann     eMenuID_HelpGUIHelp
46607caa17caSRaphael Isemann   };
46617caa17caSRaphael Isemann 
46627caa17caSRaphael Isemann   ApplicationDelegate(Application &app, Debugger &debugger)
46637caa17caSRaphael Isemann       : WindowDelegate(), MenuDelegate(), m_app(app), m_debugger(debugger) {}
46647caa17caSRaphael Isemann 
46657caa17caSRaphael Isemann   ~ApplicationDelegate() override = default;
46667caa17caSRaphael Isemann 
46677caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
46687caa17caSRaphael Isemann     return false; // Drawing not handled, let standard window drawing happen
46697caa17caSRaphael Isemann   }
46707caa17caSRaphael Isemann 
46717caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int key) override {
46727caa17caSRaphael Isemann     switch (key) {
46737caa17caSRaphael Isemann     case '\t':
46747caa17caSRaphael Isemann       window.SelectNextWindowAsActive();
46757caa17caSRaphael Isemann       return eKeyHandled;
46767caa17caSRaphael Isemann 
467729cc50e1SOmar Emara     case KEY_SHIFT_TAB:
46787a63dc53SLuboš Luňák       window.SelectPreviousWindowAsActive();
46797a63dc53SLuboš Luňák       return eKeyHandled;
46807a63dc53SLuboš Luňák 
46817caa17caSRaphael Isemann     case 'h':
46827caa17caSRaphael Isemann       window.CreateHelpSubwindow();
46837caa17caSRaphael Isemann       return eKeyHandled;
46847caa17caSRaphael Isemann 
46857caa17caSRaphael Isemann     case KEY_ESCAPE:
46867caa17caSRaphael Isemann       return eQuitApplication;
46877caa17caSRaphael Isemann 
46887caa17caSRaphael Isemann     default:
46897caa17caSRaphael Isemann       break;
46907caa17caSRaphael Isemann     }
46917caa17caSRaphael Isemann     return eKeyNotHandled;
46927caa17caSRaphael Isemann   }
46937caa17caSRaphael Isemann 
46947caa17caSRaphael Isemann   const char *WindowDelegateGetHelpText() override {
46957caa17caSRaphael Isemann     return "Welcome to the LLDB curses GUI.\n\n"
46967caa17caSRaphael Isemann            "Press the TAB key to change the selected view.\n"
46977caa17caSRaphael Isemann            "Each view has its own keyboard shortcuts, press 'h' to open a "
46987caa17caSRaphael Isemann            "dialog to display them.\n\n"
46997caa17caSRaphael Isemann            "Common key bindings for all views:";
47007caa17caSRaphael Isemann   }
47017caa17caSRaphael Isemann 
47027caa17caSRaphael Isemann   KeyHelp *WindowDelegateGetKeyHelp() override {
47037caa17caSRaphael Isemann     static curses::KeyHelp g_source_view_key_help[] = {
47047caa17caSRaphael Isemann         {'\t', "Select next view"},
47057a63dc53SLuboš Luňák         {KEY_BTAB, "Select previous view"},
47067caa17caSRaphael Isemann         {'h', "Show help dialog with view specific key bindings"},
47077caa17caSRaphael Isemann         {',', "Page up"},
47087caa17caSRaphael Isemann         {'.', "Page down"},
47097caa17caSRaphael Isemann         {KEY_UP, "Select previous"},
47107caa17caSRaphael Isemann         {KEY_DOWN, "Select next"},
47117caa17caSRaphael Isemann         {KEY_LEFT, "Unexpand or select parent"},
47127caa17caSRaphael Isemann         {KEY_RIGHT, "Expand"},
47137caa17caSRaphael Isemann         {KEY_PPAGE, "Page up"},
47147caa17caSRaphael Isemann         {KEY_NPAGE, "Page down"},
47157caa17caSRaphael Isemann         {'\0', nullptr}};
47167caa17caSRaphael Isemann     return g_source_view_key_help;
47177caa17caSRaphael Isemann   }
47187caa17caSRaphael Isemann 
47197caa17caSRaphael Isemann   MenuActionResult MenuDelegateAction(Menu &menu) override {
47207caa17caSRaphael Isemann     switch (menu.GetIdentifier()) {
47217caa17caSRaphael Isemann     case eMenuID_ThreadStepIn: {
47227caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47237caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47247caa17caSRaphael Isemann       if (exe_ctx.HasThreadScope()) {
47257caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47267caa17caSRaphael Isemann         if (process && process->IsAlive() &&
47277caa17caSRaphael Isemann             StateIsStoppedState(process->GetState(), true))
47287caa17caSRaphael Isemann           exe_ctx.GetThreadRef().StepIn(true);
47297caa17caSRaphael Isemann       }
47307caa17caSRaphael Isemann     }
47317caa17caSRaphael Isemann       return MenuActionResult::Handled;
47327caa17caSRaphael Isemann 
47337caa17caSRaphael Isemann     case eMenuID_ThreadStepOut: {
47347caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47357caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47367caa17caSRaphael Isemann       if (exe_ctx.HasThreadScope()) {
47377caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47387caa17caSRaphael Isemann         if (process && process->IsAlive() &&
47397caa17caSRaphael Isemann             StateIsStoppedState(process->GetState(), true))
47407caa17caSRaphael Isemann           exe_ctx.GetThreadRef().StepOut();
47417caa17caSRaphael Isemann       }
47427caa17caSRaphael Isemann     }
47437caa17caSRaphael Isemann       return MenuActionResult::Handled;
47447caa17caSRaphael Isemann 
47457caa17caSRaphael Isemann     case eMenuID_ThreadStepOver: {
47467caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47477caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47487caa17caSRaphael Isemann       if (exe_ctx.HasThreadScope()) {
47497caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47507caa17caSRaphael Isemann         if (process && process->IsAlive() &&
47517caa17caSRaphael Isemann             StateIsStoppedState(process->GetState(), true))
47527caa17caSRaphael Isemann           exe_ctx.GetThreadRef().StepOver(true);
47537caa17caSRaphael Isemann       }
47547caa17caSRaphael Isemann     }
47557caa17caSRaphael Isemann       return MenuActionResult::Handled;
47567caa17caSRaphael Isemann 
4757*0321dbc8SOmar Emara     case eMenuID_ProcessAttach: {
4758*0321dbc8SOmar Emara       WindowSP main_window_sp = m_app.GetMainWindow();
4759*0321dbc8SOmar Emara       FormDelegateSP form_delegate_sp = FormDelegateSP(
4760*0321dbc8SOmar Emara           new ProcessAttachFormDelegate(m_debugger, main_window_sp));
4761*0321dbc8SOmar Emara       Rect bounds = main_window_sp->GetCenteredRect(80, 22);
4762*0321dbc8SOmar Emara       WindowSP form_window_sp = main_window_sp->CreateSubWindow(
4763*0321dbc8SOmar Emara           form_delegate_sp->GetName().c_str(), bounds, true);
4764*0321dbc8SOmar Emara       WindowDelegateSP window_delegate_sp =
4765*0321dbc8SOmar Emara           WindowDelegateSP(new FormWindowDelegate(form_delegate_sp));
4766*0321dbc8SOmar Emara       form_window_sp->SetDelegate(window_delegate_sp);
4767*0321dbc8SOmar Emara       return MenuActionResult::Handled;
4768*0321dbc8SOmar Emara     }
4769*0321dbc8SOmar Emara 
47707caa17caSRaphael Isemann     case eMenuID_ProcessContinue: {
47717caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47727caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47737caa17caSRaphael Isemann       if (exe_ctx.HasProcessScope()) {
47747caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47757caa17caSRaphael Isemann         if (process && process->IsAlive() &&
47767caa17caSRaphael Isemann             StateIsStoppedState(process->GetState(), true))
47777caa17caSRaphael Isemann           process->Resume();
47787caa17caSRaphael Isemann       }
47797caa17caSRaphael Isemann     }
47807caa17caSRaphael Isemann       return MenuActionResult::Handled;
47817caa17caSRaphael Isemann 
47827caa17caSRaphael Isemann     case eMenuID_ProcessKill: {
47837caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47847caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47857caa17caSRaphael Isemann       if (exe_ctx.HasProcessScope()) {
47867caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47877caa17caSRaphael Isemann         if (process && process->IsAlive())
47887caa17caSRaphael Isemann           process->Destroy(false);
47897caa17caSRaphael Isemann       }
47907caa17caSRaphael Isemann     }
47917caa17caSRaphael Isemann       return MenuActionResult::Handled;
47927caa17caSRaphael Isemann 
47937caa17caSRaphael Isemann     case eMenuID_ProcessHalt: {
47947caa17caSRaphael Isemann       ExecutionContext exe_ctx =
47957caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
47967caa17caSRaphael Isemann       if (exe_ctx.HasProcessScope()) {
47977caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
47987caa17caSRaphael Isemann         if (process && process->IsAlive())
47997caa17caSRaphael Isemann           process->Halt();
48007caa17caSRaphael Isemann       }
48017caa17caSRaphael Isemann     }
48027caa17caSRaphael Isemann       return MenuActionResult::Handled;
48037caa17caSRaphael Isemann 
480447d2c7cdSLuboš Luňák     case eMenuID_ProcessDetachResume:
480547d2c7cdSLuboš Luňák     case eMenuID_ProcessDetachSuspended: {
48067caa17caSRaphael Isemann       ExecutionContext exe_ctx =
48077caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
48087caa17caSRaphael Isemann       if (exe_ctx.HasProcessScope()) {
48097caa17caSRaphael Isemann         Process *process = exe_ctx.GetProcessPtr();
48107caa17caSRaphael Isemann         if (process && process->IsAlive())
481147d2c7cdSLuboš Luňák           process->Detach(menu.GetIdentifier() ==
481247d2c7cdSLuboš Luňák                           eMenuID_ProcessDetachSuspended);
48137caa17caSRaphael Isemann       }
48147caa17caSRaphael Isemann     }
48157caa17caSRaphael Isemann       return MenuActionResult::Handled;
48167caa17caSRaphael Isemann 
48177caa17caSRaphael Isemann     case eMenuID_Process: {
48187caa17caSRaphael Isemann       // Populate the menu with all of the threads if the process is stopped
48197caa17caSRaphael Isemann       // when the Process menu gets selected and is about to display its
48207caa17caSRaphael Isemann       // submenu.
48217caa17caSRaphael Isemann       Menus &submenus = menu.GetSubmenus();
48227caa17caSRaphael Isemann       ExecutionContext exe_ctx =
48237caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
48247caa17caSRaphael Isemann       Process *process = exe_ctx.GetProcessPtr();
48257caa17caSRaphael Isemann       if (process && process->IsAlive() &&
48267caa17caSRaphael Isemann           StateIsStoppedState(process->GetState(), true)) {
48277caa17caSRaphael Isemann         if (submenus.size() == 7)
48287caa17caSRaphael Isemann           menu.AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
48297caa17caSRaphael Isemann         else if (submenus.size() > 8)
48307caa17caSRaphael Isemann           submenus.erase(submenus.begin() + 8, submenus.end());
48317caa17caSRaphael Isemann 
48327caa17caSRaphael Isemann         ThreadList &threads = process->GetThreadList();
48337caa17caSRaphael Isemann         std::lock_guard<std::recursive_mutex> guard(threads.GetMutex());
48347caa17caSRaphael Isemann         size_t num_threads = threads.GetSize();
48357caa17caSRaphael Isemann         for (size_t i = 0; i < num_threads; ++i) {
48367caa17caSRaphael Isemann           ThreadSP thread_sp = threads.GetThreadAtIndex(i);
48377caa17caSRaphael Isemann           char menu_char = '\0';
48387caa17caSRaphael Isemann           if (i < 9)
48397caa17caSRaphael Isemann             menu_char = '1' + i;
48407caa17caSRaphael Isemann           StreamString thread_menu_title;
48417caa17caSRaphael Isemann           thread_menu_title.Printf("Thread %u", thread_sp->GetIndexID());
48427caa17caSRaphael Isemann           const char *thread_name = thread_sp->GetName();
48437caa17caSRaphael Isemann           if (thread_name && thread_name[0])
48447caa17caSRaphael Isemann             thread_menu_title.Printf(" %s", thread_name);
48457caa17caSRaphael Isemann           else {
48467caa17caSRaphael Isemann             const char *queue_name = thread_sp->GetQueueName();
48477caa17caSRaphael Isemann             if (queue_name && queue_name[0])
48487caa17caSRaphael Isemann               thread_menu_title.Printf(" %s", queue_name);
48497caa17caSRaphael Isemann           }
48507caa17caSRaphael Isemann           menu.AddSubmenu(
48517caa17caSRaphael Isemann               MenuSP(new Menu(thread_menu_title.GetString().str().c_str(),
48527caa17caSRaphael Isemann                               nullptr, menu_char, thread_sp->GetID())));
48537caa17caSRaphael Isemann         }
48547caa17caSRaphael Isemann       } else if (submenus.size() > 7) {
48557caa17caSRaphael Isemann         // Remove the separator and any other thread submenu items that were
48567caa17caSRaphael Isemann         // previously added
48577caa17caSRaphael Isemann         submenus.erase(submenus.begin() + 7, submenus.end());
48587caa17caSRaphael Isemann       }
48597caa17caSRaphael Isemann       // Since we are adding and removing items we need to recalculate the name
48607caa17caSRaphael Isemann       // lengths
48617caa17caSRaphael Isemann       menu.RecalculateNameLengths();
48627caa17caSRaphael Isemann     }
48637caa17caSRaphael Isemann       return MenuActionResult::Handled;
48647caa17caSRaphael Isemann 
48657caa17caSRaphael Isemann     case eMenuID_ViewVariables: {
48667caa17caSRaphael Isemann       WindowSP main_window_sp = m_app.GetMainWindow();
48677caa17caSRaphael Isemann       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
48687caa17caSRaphael Isemann       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
48697caa17caSRaphael Isemann       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
48707caa17caSRaphael Isemann       const Rect source_bounds = source_window_sp->GetBounds();
48717caa17caSRaphael Isemann 
48727caa17caSRaphael Isemann       if (variables_window_sp) {
48737caa17caSRaphael Isemann         const Rect variables_bounds = variables_window_sp->GetBounds();
48747caa17caSRaphael Isemann 
48757caa17caSRaphael Isemann         main_window_sp->RemoveSubWindow(variables_window_sp.get());
48767caa17caSRaphael Isemann 
48777caa17caSRaphael Isemann         if (registers_window_sp) {
48787caa17caSRaphael Isemann           // We have a registers window, so give all the area back to the
48797caa17caSRaphael Isemann           // registers window
48807caa17caSRaphael Isemann           Rect registers_bounds = variables_bounds;
48817caa17caSRaphael Isemann           registers_bounds.size.width = source_bounds.size.width;
48827caa17caSRaphael Isemann           registers_window_sp->SetBounds(registers_bounds);
48837caa17caSRaphael Isemann         } else {
48847caa17caSRaphael Isemann           // We have no registers window showing so give the bottom area back
48857caa17caSRaphael Isemann           // to the source view
48867caa17caSRaphael Isemann           source_window_sp->Resize(source_bounds.size.width,
48877caa17caSRaphael Isemann                                    source_bounds.size.height +
48887caa17caSRaphael Isemann                                        variables_bounds.size.height);
48897caa17caSRaphael Isemann         }
48907caa17caSRaphael Isemann       } else {
48917caa17caSRaphael Isemann         Rect new_variables_rect;
48927caa17caSRaphael Isemann         if (registers_window_sp) {
48937caa17caSRaphael Isemann           // We have a registers window so split the area of the registers
48947caa17caSRaphael Isemann           // window into two columns where the left hand side will be the
48957caa17caSRaphael Isemann           // variables and the right hand side will be the registers
48967caa17caSRaphael Isemann           const Rect variables_bounds = registers_window_sp->GetBounds();
48977caa17caSRaphael Isemann           Rect new_registers_rect;
48987caa17caSRaphael Isemann           variables_bounds.VerticalSplitPercentage(0.50, new_variables_rect,
48997caa17caSRaphael Isemann                                                    new_registers_rect);
49007caa17caSRaphael Isemann           registers_window_sp->SetBounds(new_registers_rect);
49017caa17caSRaphael Isemann         } else {
49022f1b24b7SLuboš Luňák           // No registers window, grab the bottom part of the source window
49037caa17caSRaphael Isemann           Rect new_source_rect;
49047caa17caSRaphael Isemann           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
49057caa17caSRaphael Isemann                                                   new_variables_rect);
49067caa17caSRaphael Isemann           source_window_sp->SetBounds(new_source_rect);
49077caa17caSRaphael Isemann         }
49087caa17caSRaphael Isemann         WindowSP new_window_sp = main_window_sp->CreateSubWindow(
49097caa17caSRaphael Isemann             "Variables", new_variables_rect, false);
49107caa17caSRaphael Isemann         new_window_sp->SetDelegate(
49117caa17caSRaphael Isemann             WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
49127caa17caSRaphael Isemann       }
49137caa17caSRaphael Isemann       touchwin(stdscr);
49147caa17caSRaphael Isemann     }
49157caa17caSRaphael Isemann       return MenuActionResult::Handled;
49167caa17caSRaphael Isemann 
49177caa17caSRaphael Isemann     case eMenuID_ViewRegisters: {
49187caa17caSRaphael Isemann       WindowSP main_window_sp = m_app.GetMainWindow();
49197caa17caSRaphael Isemann       WindowSP source_window_sp = main_window_sp->FindSubWindow("Source");
49207caa17caSRaphael Isemann       WindowSP variables_window_sp = main_window_sp->FindSubWindow("Variables");
49217caa17caSRaphael Isemann       WindowSP registers_window_sp = main_window_sp->FindSubWindow("Registers");
49227caa17caSRaphael Isemann       const Rect source_bounds = source_window_sp->GetBounds();
49237caa17caSRaphael Isemann 
49247caa17caSRaphael Isemann       if (registers_window_sp) {
49257caa17caSRaphael Isemann         if (variables_window_sp) {
49267caa17caSRaphael Isemann           const Rect variables_bounds = variables_window_sp->GetBounds();
49277caa17caSRaphael Isemann 
49287caa17caSRaphael Isemann           // We have a variables window, so give all the area back to the
49297caa17caSRaphael Isemann           // variables window
49307caa17caSRaphael Isemann           variables_window_sp->Resize(variables_bounds.size.width +
49317caa17caSRaphael Isemann                                           registers_window_sp->GetWidth(),
49327caa17caSRaphael Isemann                                       variables_bounds.size.height);
49337caa17caSRaphael Isemann         } else {
49347caa17caSRaphael Isemann           // We have no variables window showing so give the bottom area back
49357caa17caSRaphael Isemann           // to the source view
49367caa17caSRaphael Isemann           source_window_sp->Resize(source_bounds.size.width,
49377caa17caSRaphael Isemann                                    source_bounds.size.height +
49387caa17caSRaphael Isemann                                        registers_window_sp->GetHeight());
49397caa17caSRaphael Isemann         }
49407caa17caSRaphael Isemann         main_window_sp->RemoveSubWindow(registers_window_sp.get());
49417caa17caSRaphael Isemann       } else {
49427caa17caSRaphael Isemann         Rect new_regs_rect;
49437caa17caSRaphael Isemann         if (variables_window_sp) {
49447caa17caSRaphael Isemann           // We have a variables window, split it into two columns where the
49457caa17caSRaphael Isemann           // left hand side will be the variables and the right hand side will
49467caa17caSRaphael Isemann           // be the registers
49477caa17caSRaphael Isemann           const Rect variables_bounds = variables_window_sp->GetBounds();
49487caa17caSRaphael Isemann           Rect new_vars_rect;
49497caa17caSRaphael Isemann           variables_bounds.VerticalSplitPercentage(0.50, new_vars_rect,
49507caa17caSRaphael Isemann                                                    new_regs_rect);
49517caa17caSRaphael Isemann           variables_window_sp->SetBounds(new_vars_rect);
49527caa17caSRaphael Isemann         } else {
49532f1b24b7SLuboš Luňák           // No variables window, grab the bottom part of the source window
49547caa17caSRaphael Isemann           Rect new_source_rect;
49557caa17caSRaphael Isemann           source_bounds.HorizontalSplitPercentage(0.70, new_source_rect,
49567caa17caSRaphael Isemann                                                   new_regs_rect);
49577caa17caSRaphael Isemann           source_window_sp->SetBounds(new_source_rect);
49587caa17caSRaphael Isemann         }
49597caa17caSRaphael Isemann         WindowSP new_window_sp =
49607caa17caSRaphael Isemann             main_window_sp->CreateSubWindow("Registers", new_regs_rect, false);
49617caa17caSRaphael Isemann         new_window_sp->SetDelegate(
49627caa17caSRaphael Isemann             WindowDelegateSP(new RegistersWindowDelegate(m_debugger)));
49637caa17caSRaphael Isemann       }
49647caa17caSRaphael Isemann       touchwin(stdscr);
49657caa17caSRaphael Isemann     }
49667caa17caSRaphael Isemann       return MenuActionResult::Handled;
49677caa17caSRaphael Isemann 
49687caa17caSRaphael Isemann     case eMenuID_HelpGUIHelp:
49697caa17caSRaphael Isemann       m_app.GetMainWindow()->CreateHelpSubwindow();
49707caa17caSRaphael Isemann       return MenuActionResult::Handled;
49717caa17caSRaphael Isemann 
49727caa17caSRaphael Isemann     default:
49737caa17caSRaphael Isemann       break;
49747caa17caSRaphael Isemann     }
49757caa17caSRaphael Isemann 
49767caa17caSRaphael Isemann     return MenuActionResult::NotHandled;
49777caa17caSRaphael Isemann   }
49787caa17caSRaphael Isemann 
49797caa17caSRaphael Isemann protected:
49807caa17caSRaphael Isemann   Application &m_app;
49817caa17caSRaphael Isemann   Debugger &m_debugger;
49827caa17caSRaphael Isemann };
49837caa17caSRaphael Isemann 
49847caa17caSRaphael Isemann class StatusBarWindowDelegate : public WindowDelegate {
49857caa17caSRaphael Isemann public:
49867caa17caSRaphael Isemann   StatusBarWindowDelegate(Debugger &debugger) : m_debugger(debugger) {
49877caa17caSRaphael Isemann     FormatEntity::Parse("Thread: ${thread.id%tid}", m_format);
49887caa17caSRaphael Isemann   }
49897caa17caSRaphael Isemann 
49907caa17caSRaphael Isemann   ~StatusBarWindowDelegate() override = default;
49917caa17caSRaphael Isemann 
49927caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
49937caa17caSRaphael Isemann     ExecutionContext exe_ctx =
49947caa17caSRaphael Isemann         m_debugger.GetCommandInterpreter().GetExecutionContext();
49957caa17caSRaphael Isemann     Process *process = exe_ctx.GetProcessPtr();
49967caa17caSRaphael Isemann     Thread *thread = exe_ctx.GetThreadPtr();
49977caa17caSRaphael Isemann     StackFrame *frame = exe_ctx.GetFramePtr();
49987caa17caSRaphael Isemann     window.Erase();
49994a8e4b5cSLuboš Luňák     window.SetBackground(BlackOnWhite);
50007caa17caSRaphael Isemann     window.MoveCursor(0, 0);
50017caa17caSRaphael Isemann     if (process) {
50027caa17caSRaphael Isemann       const StateType state = process->GetState();
50037caa17caSRaphael Isemann       window.Printf("Process: %5" PRIu64 " %10s", process->GetID(),
50047caa17caSRaphael Isemann                     StateAsCString(state));
50057caa17caSRaphael Isemann 
50067caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
50077caa17caSRaphael Isemann         StreamString strm;
50087caa17caSRaphael Isemann         if (thread && FormatEntity::Format(m_format, strm, nullptr, &exe_ctx,
50097caa17caSRaphael Isemann                                            nullptr, nullptr, false, false)) {
50107caa17caSRaphael Isemann           window.MoveCursor(40, 0);
5011fc0e8fb7SLuboš Luňák           window.PutCStringTruncated(1, strm.GetString().str().c_str());
50127caa17caSRaphael Isemann         }
50137caa17caSRaphael Isemann 
50147caa17caSRaphael Isemann         window.MoveCursor(60, 0);
50157caa17caSRaphael Isemann         if (frame)
50167caa17caSRaphael Isemann           window.Printf("Frame: %3u  PC = 0x%16.16" PRIx64,
50177caa17caSRaphael Isemann                         frame->GetFrameIndex(),
50187caa17caSRaphael Isemann                         frame->GetFrameCodeAddress().GetOpcodeLoadAddress(
50197caa17caSRaphael Isemann                             exe_ctx.GetTargetPtr()));
50207caa17caSRaphael Isemann       } else if (state == eStateExited) {
50217caa17caSRaphael Isemann         const char *exit_desc = process->GetExitDescription();
50227caa17caSRaphael Isemann         const int exit_status = process->GetExitStatus();
50237caa17caSRaphael Isemann         if (exit_desc && exit_desc[0])
50247caa17caSRaphael Isemann           window.Printf(" with status = %i (%s)", exit_status, exit_desc);
50257caa17caSRaphael Isemann         else
50267caa17caSRaphael Isemann           window.Printf(" with status = %i", exit_status);
50277caa17caSRaphael Isemann       }
50287caa17caSRaphael Isemann     }
50297caa17caSRaphael Isemann     return true;
50307caa17caSRaphael Isemann   }
50317caa17caSRaphael Isemann 
50327caa17caSRaphael Isemann protected:
50337caa17caSRaphael Isemann   Debugger &m_debugger;
50347caa17caSRaphael Isemann   FormatEntity::Entry m_format;
50357caa17caSRaphael Isemann };
50367caa17caSRaphael Isemann 
50377caa17caSRaphael Isemann class SourceFileWindowDelegate : public WindowDelegate {
50387caa17caSRaphael Isemann public:
50397caa17caSRaphael Isemann   SourceFileWindowDelegate(Debugger &debugger)
50407caa17caSRaphael Isemann       : WindowDelegate(), m_debugger(debugger), m_sc(), m_file_sp(),
50417caa17caSRaphael Isemann         m_disassembly_scope(nullptr), m_disassembly_sp(), m_disassembly_range(),
5042b4825a6dSJordan Rupprecht         m_title(), m_tid(LLDB_INVALID_THREAD_ID), m_line_width(4),
5043b4825a6dSJordan Rupprecht         m_selected_line(0), m_pc_line(0), m_stop_id(0), m_frame_idx(UINT32_MAX),
5044b4825a6dSJordan Rupprecht         m_first_visible_line(0), m_first_visible_column(0), m_min_x(0),
5045b4825a6dSJordan Rupprecht         m_min_y(0), m_max_x(0), m_max_y(0) {}
50467caa17caSRaphael Isemann 
50477caa17caSRaphael Isemann   ~SourceFileWindowDelegate() override = default;
50487caa17caSRaphael Isemann 
50497caa17caSRaphael Isemann   void Update(const SymbolContext &sc) { m_sc = sc; }
50507caa17caSRaphael Isemann 
50517caa17caSRaphael Isemann   uint32_t NumVisibleLines() const { return m_max_y - m_min_y; }
50527caa17caSRaphael Isemann 
50537caa17caSRaphael Isemann   const char *WindowDelegateGetHelpText() override {
50547caa17caSRaphael Isemann     return "Source/Disassembly window keyboard shortcuts:";
50557caa17caSRaphael Isemann   }
50567caa17caSRaphael Isemann 
50577caa17caSRaphael Isemann   KeyHelp *WindowDelegateGetKeyHelp() override {
50587caa17caSRaphael Isemann     static curses::KeyHelp g_source_view_key_help[] = {
50597caa17caSRaphael Isemann         {KEY_RETURN, "Run to selected line with one shot breakpoint"},
50607caa17caSRaphael Isemann         {KEY_UP, "Select previous source line"},
50617caa17caSRaphael Isemann         {KEY_DOWN, "Select next source line"},
5062dcd4589aSLuboš Luňák         {KEY_LEFT, "Scroll to the left"},
5063dcd4589aSLuboš Luňák         {KEY_RIGHT, "Scroll to the right"},
50647caa17caSRaphael Isemann         {KEY_PPAGE, "Page up"},
50657caa17caSRaphael Isemann         {KEY_NPAGE, "Page down"},
50667caa17caSRaphael Isemann         {'b', "Set breakpoint on selected source/disassembly line"},
50677caa17caSRaphael Isemann         {'c', "Continue process"},
50687caa17caSRaphael Isemann         {'D', "Detach with process suspended"},
50697caa17caSRaphael Isemann         {'h', "Show help dialog"},
50707caa17caSRaphael Isemann         {'n', "Step over (source line)"},
50717caa17caSRaphael Isemann         {'N', "Step over (single instruction)"},
5072336c7029SLuboš Luňák         {'f', "Step out (finish)"},
50737caa17caSRaphael Isemann         {'s', "Step in (source line)"},
50747caa17caSRaphael Isemann         {'S', "Step in (single instruction)"},
507513978643SLuboš Luňák         {'u', "Frame up"},
507613978643SLuboš Luňák         {'d', "Frame down"},
50777caa17caSRaphael Isemann         {',', "Page up"},
50787caa17caSRaphael Isemann         {'.', "Page down"},
50797caa17caSRaphael Isemann         {'\0', nullptr}};
50807caa17caSRaphael Isemann     return g_source_view_key_help;
50817caa17caSRaphael Isemann   }
50827caa17caSRaphael Isemann 
50837caa17caSRaphael Isemann   bool WindowDelegateDraw(Window &window, bool force) override {
50847caa17caSRaphael Isemann     ExecutionContext exe_ctx =
50857caa17caSRaphael Isemann         m_debugger.GetCommandInterpreter().GetExecutionContext();
50867caa17caSRaphael Isemann     Process *process = exe_ctx.GetProcessPtr();
50877caa17caSRaphael Isemann     Thread *thread = nullptr;
50887caa17caSRaphael Isemann 
50897caa17caSRaphael Isemann     bool update_location = false;
50907caa17caSRaphael Isemann     if (process) {
50917caa17caSRaphael Isemann       StateType state = process->GetState();
50927caa17caSRaphael Isemann       if (StateIsStoppedState(state, true)) {
50937caa17caSRaphael Isemann         // We are stopped, so it is ok to
50947caa17caSRaphael Isemann         update_location = true;
50957caa17caSRaphael Isemann       }
50967caa17caSRaphael Isemann     }
50977caa17caSRaphael Isemann 
50987caa17caSRaphael Isemann     m_min_x = 1;
50997caa17caSRaphael Isemann     m_min_y = 2;
51007caa17caSRaphael Isemann     m_max_x = window.GetMaxX() - 1;
51017caa17caSRaphael Isemann     m_max_y = window.GetMaxY() - 1;
51027caa17caSRaphael Isemann 
51037caa17caSRaphael Isemann     const uint32_t num_visible_lines = NumVisibleLines();
51047caa17caSRaphael Isemann     StackFrameSP frame_sp;
51057caa17caSRaphael Isemann     bool set_selected_line_to_pc = false;
51067caa17caSRaphael Isemann 
51077caa17caSRaphael Isemann     if (update_location) {
51087caa17caSRaphael Isemann       const bool process_alive = process ? process->IsAlive() : false;
51097caa17caSRaphael Isemann       bool thread_changed = false;
51107caa17caSRaphael Isemann       if (process_alive) {
51117caa17caSRaphael Isemann         thread = exe_ctx.GetThreadPtr();
51127caa17caSRaphael Isemann         if (thread) {
51137caa17caSRaphael Isemann           frame_sp = thread->GetSelectedFrame();
51147caa17caSRaphael Isemann           auto tid = thread->GetID();
51157caa17caSRaphael Isemann           thread_changed = tid != m_tid;
51167caa17caSRaphael Isemann           m_tid = tid;
51177caa17caSRaphael Isemann         } else {
51187caa17caSRaphael Isemann           if (m_tid != LLDB_INVALID_THREAD_ID) {
51197caa17caSRaphael Isemann             thread_changed = true;
51207caa17caSRaphael Isemann             m_tid = LLDB_INVALID_THREAD_ID;
51217caa17caSRaphael Isemann           }
51227caa17caSRaphael Isemann         }
51237caa17caSRaphael Isemann       }
51247caa17caSRaphael Isemann       const uint32_t stop_id = process ? process->GetStopID() : 0;
51257caa17caSRaphael Isemann       const bool stop_id_changed = stop_id != m_stop_id;
51267caa17caSRaphael Isemann       bool frame_changed = false;
51277caa17caSRaphael Isemann       m_stop_id = stop_id;
51287caa17caSRaphael Isemann       m_title.Clear();
51297caa17caSRaphael Isemann       if (frame_sp) {
51307caa17caSRaphael Isemann         m_sc = frame_sp->GetSymbolContext(eSymbolContextEverything);
51317caa17caSRaphael Isemann         if (m_sc.module_sp) {
51327caa17caSRaphael Isemann           m_title.Printf(
51337caa17caSRaphael Isemann               "%s", m_sc.module_sp->GetFileSpec().GetFilename().GetCString());
51347caa17caSRaphael Isemann           ConstString func_name = m_sc.GetFunctionName();
51357caa17caSRaphael Isemann           if (func_name)
51367caa17caSRaphael Isemann             m_title.Printf("`%s", func_name.GetCString());
51377caa17caSRaphael Isemann         }
51387caa17caSRaphael Isemann         const uint32_t frame_idx = frame_sp->GetFrameIndex();
51397caa17caSRaphael Isemann         frame_changed = frame_idx != m_frame_idx;
51407caa17caSRaphael Isemann         m_frame_idx = frame_idx;
51417caa17caSRaphael Isemann       } else {
51427caa17caSRaphael Isemann         m_sc.Clear(true);
51437caa17caSRaphael Isemann         frame_changed = m_frame_idx != UINT32_MAX;
51447caa17caSRaphael Isemann         m_frame_idx = UINT32_MAX;
51457caa17caSRaphael Isemann       }
51467caa17caSRaphael Isemann 
51477caa17caSRaphael Isemann       const bool context_changed =
51487caa17caSRaphael Isemann           thread_changed || frame_changed || stop_id_changed;
51497caa17caSRaphael Isemann 
51507caa17caSRaphael Isemann       if (process_alive) {
51517caa17caSRaphael Isemann         if (m_sc.line_entry.IsValid()) {
51527caa17caSRaphael Isemann           m_pc_line = m_sc.line_entry.line;
51537caa17caSRaphael Isemann           if (m_pc_line != UINT32_MAX)
51547caa17caSRaphael Isemann             --m_pc_line; // Convert to zero based line number...
51557caa17caSRaphael Isemann           // Update the selected line if the stop ID changed...
51567caa17caSRaphael Isemann           if (context_changed)
51577caa17caSRaphael Isemann             m_selected_line = m_pc_line;
51587caa17caSRaphael Isemann 
5159532290e6SPavel Labath           if (m_file_sp && m_file_sp->GetFileSpec() == m_sc.line_entry.file) {
51607caa17caSRaphael Isemann             // Same file, nothing to do, we should either have the lines or not
51617caa17caSRaphael Isemann             // (source file missing)
51627caa17caSRaphael Isemann             if (m_selected_line >= static_cast<size_t>(m_first_visible_line)) {
51637caa17caSRaphael Isemann               if (m_selected_line >= m_first_visible_line + num_visible_lines)
51647caa17caSRaphael Isemann                 m_first_visible_line = m_selected_line - 10;
51657caa17caSRaphael Isemann             } else {
51667caa17caSRaphael Isemann               if (m_selected_line > 10)
51677caa17caSRaphael Isemann                 m_first_visible_line = m_selected_line - 10;
51687caa17caSRaphael Isemann               else
51697caa17caSRaphael Isemann                 m_first_visible_line = 0;
51707caa17caSRaphael Isemann             }
51717caa17caSRaphael Isemann           } else {
51727caa17caSRaphael Isemann             // File changed, set selected line to the line with the PC
51737caa17caSRaphael Isemann             m_selected_line = m_pc_line;
51747caa17caSRaphael Isemann             m_file_sp =
51757caa17caSRaphael Isemann                 m_debugger.GetSourceManager().GetFile(m_sc.line_entry.file);
51767caa17caSRaphael Isemann             if (m_file_sp) {
51777caa17caSRaphael Isemann               const size_t num_lines = m_file_sp->GetNumLines();
51787caa17caSRaphael Isemann               m_line_width = 1;
51797caa17caSRaphael Isemann               for (size_t n = num_lines; n >= 10; n = n / 10)
51807caa17caSRaphael Isemann                 ++m_line_width;
51817caa17caSRaphael Isemann 
51827caa17caSRaphael Isemann               if (num_lines < num_visible_lines ||
51837caa17caSRaphael Isemann                   m_selected_line < num_visible_lines)
51847caa17caSRaphael Isemann                 m_first_visible_line = 0;
51857caa17caSRaphael Isemann               else
51867caa17caSRaphael Isemann                 m_first_visible_line = m_selected_line - 10;
51877caa17caSRaphael Isemann             }
51887caa17caSRaphael Isemann           }
51897caa17caSRaphael Isemann         } else {
51907caa17caSRaphael Isemann           m_file_sp.reset();
51917caa17caSRaphael Isemann         }
51927caa17caSRaphael Isemann 
51937caa17caSRaphael Isemann         if (!m_file_sp || m_file_sp->GetNumLines() == 0) {
51947caa17caSRaphael Isemann           // Show disassembly
51957caa17caSRaphael Isemann           bool prefer_file_cache = false;
51967caa17caSRaphael Isemann           if (m_sc.function) {
51977caa17caSRaphael Isemann             if (m_disassembly_scope != m_sc.function) {
51987caa17caSRaphael Isemann               m_disassembly_scope = m_sc.function;
51997caa17caSRaphael Isemann               m_disassembly_sp = m_sc.function->GetInstructions(
5200e9fe788dSJason Molenda                   exe_ctx, nullptr, !prefer_file_cache);
52017caa17caSRaphael Isemann               if (m_disassembly_sp) {
52027caa17caSRaphael Isemann                 set_selected_line_to_pc = true;
52037caa17caSRaphael Isemann                 m_disassembly_range = m_sc.function->GetAddressRange();
52047caa17caSRaphael Isemann               } else {
52057caa17caSRaphael Isemann                 m_disassembly_range.Clear();
52067caa17caSRaphael Isemann               }
52077caa17caSRaphael Isemann             } else {
52087caa17caSRaphael Isemann               set_selected_line_to_pc = context_changed;
52097caa17caSRaphael Isemann             }
52107caa17caSRaphael Isemann           } else if (m_sc.symbol) {
52117caa17caSRaphael Isemann             if (m_disassembly_scope != m_sc.symbol) {
52127caa17caSRaphael Isemann               m_disassembly_scope = m_sc.symbol;
52137caa17caSRaphael Isemann               m_disassembly_sp = m_sc.symbol->GetInstructions(
52147caa17caSRaphael Isemann                   exe_ctx, nullptr, prefer_file_cache);
52157caa17caSRaphael Isemann               if (m_disassembly_sp) {
52167caa17caSRaphael Isemann                 set_selected_line_to_pc = true;
52177caa17caSRaphael Isemann                 m_disassembly_range.GetBaseAddress() =
52187caa17caSRaphael Isemann                     m_sc.symbol->GetAddress();
52197caa17caSRaphael Isemann                 m_disassembly_range.SetByteSize(m_sc.symbol->GetByteSize());
52207caa17caSRaphael Isemann               } else {
52217caa17caSRaphael Isemann                 m_disassembly_range.Clear();
52227caa17caSRaphael Isemann               }
52237caa17caSRaphael Isemann             } else {
52247caa17caSRaphael Isemann               set_selected_line_to_pc = context_changed;
52257caa17caSRaphael Isemann             }
52267caa17caSRaphael Isemann           }
52277caa17caSRaphael Isemann         }
52287caa17caSRaphael Isemann       } else {
52297caa17caSRaphael Isemann         m_pc_line = UINT32_MAX;
52307caa17caSRaphael Isemann       }
52317caa17caSRaphael Isemann     }
52327caa17caSRaphael Isemann 
52337caa17caSRaphael Isemann     const int window_width = window.GetWidth();
52347caa17caSRaphael Isemann     window.Erase();
52357caa17caSRaphael Isemann     window.DrawTitleBox("Sources");
52367caa17caSRaphael Isemann     if (!m_title.GetString().empty()) {
52377caa17caSRaphael Isemann       window.AttributeOn(A_REVERSE);
52387caa17caSRaphael Isemann       window.MoveCursor(1, 1);
52397caa17caSRaphael Isemann       window.PutChar(' ');
5240fc0e8fb7SLuboš Luňák       window.PutCStringTruncated(1, m_title.GetString().str().c_str());
52417caa17caSRaphael Isemann       int x = window.GetCursorX();
52427caa17caSRaphael Isemann       if (x < window_width - 1) {
52437caa17caSRaphael Isemann         window.Printf("%*s", window_width - x - 1, "");
52447caa17caSRaphael Isemann       }
52457caa17caSRaphael Isemann       window.AttributeOff(A_REVERSE);
52467caa17caSRaphael Isemann     }
52477caa17caSRaphael Isemann 
52487caa17caSRaphael Isemann     Target *target = exe_ctx.GetTargetPtr();
52497caa17caSRaphael Isemann     const size_t num_source_lines = GetNumSourceLines();
52507caa17caSRaphael Isemann     if (num_source_lines > 0) {
52517caa17caSRaphael Isemann       // Display source
52527caa17caSRaphael Isemann       BreakpointLines bp_lines;
52537caa17caSRaphael Isemann       if (target) {
52547caa17caSRaphael Isemann         BreakpointList &bp_list = target->GetBreakpointList();
52557caa17caSRaphael Isemann         const size_t num_bps = bp_list.GetSize();
52567caa17caSRaphael Isemann         for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
52577caa17caSRaphael Isemann           BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
52587caa17caSRaphael Isemann           const size_t num_bps_locs = bp_sp->GetNumLocations();
52597caa17caSRaphael Isemann           for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
52607caa17caSRaphael Isemann             BreakpointLocationSP bp_loc_sp =
52617caa17caSRaphael Isemann                 bp_sp->GetLocationAtIndex(bp_loc_idx);
52627caa17caSRaphael Isemann             LineEntry bp_loc_line_entry;
52637caa17caSRaphael Isemann             if (bp_loc_sp->GetAddress().CalculateSymbolContextLineEntry(
52647caa17caSRaphael Isemann                     bp_loc_line_entry)) {
52657caa17caSRaphael Isemann               if (m_file_sp->GetFileSpec() == bp_loc_line_entry.file) {
52667caa17caSRaphael Isemann                 bp_lines.insert(bp_loc_line_entry.line);
52677caa17caSRaphael Isemann               }
52687caa17caSRaphael Isemann             }
52697caa17caSRaphael Isemann           }
52707caa17caSRaphael Isemann         }
52717caa17caSRaphael Isemann       }
52727caa17caSRaphael Isemann 
52737caa17caSRaphael Isemann       const attr_t selected_highlight_attr = A_REVERSE;
52744a8e4b5cSLuboš Luňák       const attr_t pc_highlight_attr = COLOR_PAIR(BlackOnBlue);
52757caa17caSRaphael Isemann 
52767caa17caSRaphael Isemann       for (size_t i = 0; i < num_visible_lines; ++i) {
52777caa17caSRaphael Isemann         const uint32_t curr_line = m_first_visible_line + i;
52787caa17caSRaphael Isemann         if (curr_line < num_source_lines) {
52797caa17caSRaphael Isemann           const int line_y = m_min_y + i;
52807caa17caSRaphael Isemann           window.MoveCursor(1, line_y);
52817caa17caSRaphael Isemann           const bool is_pc_line = curr_line == m_pc_line;
52827caa17caSRaphael Isemann           const bool line_is_selected = m_selected_line == curr_line;
52837caa17caSRaphael Isemann           // Highlight the line as the PC line first, then if the selected line
52847caa17caSRaphael Isemann           // isn't the same as the PC line, highlight it differently
52857caa17caSRaphael Isemann           attr_t highlight_attr = 0;
52867caa17caSRaphael Isemann           attr_t bp_attr = 0;
52877caa17caSRaphael Isemann           if (is_pc_line)
52887caa17caSRaphael Isemann             highlight_attr = pc_highlight_attr;
52897caa17caSRaphael Isemann           else if (line_is_selected)
52907caa17caSRaphael Isemann             highlight_attr = selected_highlight_attr;
52917caa17caSRaphael Isemann 
52927caa17caSRaphael Isemann           if (bp_lines.find(curr_line + 1) != bp_lines.end())
52934a8e4b5cSLuboš Luňák             bp_attr = COLOR_PAIR(BlackOnWhite);
52947caa17caSRaphael Isemann 
52957caa17caSRaphael Isemann           if (bp_attr)
52967caa17caSRaphael Isemann             window.AttributeOn(bp_attr);
52977caa17caSRaphael Isemann 
52987caa17caSRaphael Isemann           window.Printf(" %*u ", m_line_width, curr_line + 1);
52997caa17caSRaphael Isemann 
53007caa17caSRaphael Isemann           if (bp_attr)
53017caa17caSRaphael Isemann             window.AttributeOff(bp_attr);
53027caa17caSRaphael Isemann 
53037caa17caSRaphael Isemann           window.PutChar(ACS_VLINE);
53047caa17caSRaphael Isemann           // Mark the line with the PC with a diamond
53057caa17caSRaphael Isemann           if (is_pc_line)
53067caa17caSRaphael Isemann             window.PutChar(ACS_DIAMOND);
53077caa17caSRaphael Isemann           else
53087caa17caSRaphael Isemann             window.PutChar(' ');
53097caa17caSRaphael Isemann 
53107caa17caSRaphael Isemann           if (highlight_attr)
53117caa17caSRaphael Isemann             window.AttributeOn(highlight_attr);
531214406ca0SLuboš Luňák 
531314406ca0SLuboš Luňák           StreamString lineStream;
531414406ca0SLuboš Luňák           m_file_sp->DisplaySourceLines(curr_line + 1, {}, 0, 0, &lineStream);
531514406ca0SLuboš Luňák           StringRef line = lineStream.GetString();
531614406ca0SLuboš Luňák           if (line.endswith("\n"))
531714406ca0SLuboš Luňák             line = line.drop_back();
5318dcd4589aSLuboš Luňák           bool wasWritten = window.OutputColoredStringTruncated(
5319dcd4589aSLuboš Luňák               1, line, m_first_visible_column, line_is_selected);
5320dcd4589aSLuboš Luňák           if (line_is_selected && !wasWritten) {
5321dcd4589aSLuboš Luňák             // Draw an empty space to show the selected line if empty,
5322dcd4589aSLuboš Luňák             // or draw '<' if nothing is visible because of scrolling too much
5323dcd4589aSLuboš Luňák             // to the right.
5324dcd4589aSLuboš Luňák             window.PutCStringTruncated(
5325dcd4589aSLuboš Luňák                 1, line.empty() && m_first_visible_column == 0 ? " " : "<");
5326dcd4589aSLuboš Luňák           }
53277caa17caSRaphael Isemann 
53287caa17caSRaphael Isemann           if (is_pc_line && frame_sp &&
53297caa17caSRaphael Isemann               frame_sp->GetConcreteFrameIndex() == 0) {
53307caa17caSRaphael Isemann             StopInfoSP stop_info_sp;
53317caa17caSRaphael Isemann             if (thread)
53327caa17caSRaphael Isemann               stop_info_sp = thread->GetStopInfo();
53337caa17caSRaphael Isemann             if (stop_info_sp) {
53347caa17caSRaphael Isemann               const char *stop_description = stop_info_sp->GetDescription();
53357caa17caSRaphael Isemann               if (stop_description && stop_description[0]) {
53367caa17caSRaphael Isemann                 size_t stop_description_len = strlen(stop_description);
53377caa17caSRaphael Isemann                 int desc_x = window_width - stop_description_len - 16;
5338fc0e8fb7SLuboš Luňák                 if (desc_x - window.GetCursorX() > 0)
53397caa17caSRaphael Isemann                   window.Printf("%*s", desc_x - window.GetCursorX(), "");
534014406ca0SLuboš Luňák                 window.MoveCursor(window_width - stop_description_len - 16,
5341fc0e8fb7SLuboš Luňák                                   line_y);
53424a8e4b5cSLuboš Luňák                 const attr_t stop_reason_attr = COLOR_PAIR(WhiteOnBlue);
534314406ca0SLuboš Luňák                 window.AttributeOn(stop_reason_attr);
5344fc0e8fb7SLuboš Luňák                 window.PrintfTruncated(1, " <<< Thread %u: %s ",
5345fc0e8fb7SLuboš Luňák                                        thread->GetIndexID(), stop_description);
534614406ca0SLuboš Luňák                 window.AttributeOff(stop_reason_attr);
53477caa17caSRaphael Isemann               }
53487caa17caSRaphael Isemann             } else {
53497caa17caSRaphael Isemann               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
53507caa17caSRaphael Isemann             }
53517caa17caSRaphael Isemann           }
53527caa17caSRaphael Isemann           if (highlight_attr)
53537caa17caSRaphael Isemann             window.AttributeOff(highlight_attr);
53547caa17caSRaphael Isemann         } else {
53557caa17caSRaphael Isemann           break;
53567caa17caSRaphael Isemann         }
53577caa17caSRaphael Isemann       }
53587caa17caSRaphael Isemann     } else {
53597caa17caSRaphael Isemann       size_t num_disassembly_lines = GetNumDisassemblyLines();
53607caa17caSRaphael Isemann       if (num_disassembly_lines > 0) {
53617caa17caSRaphael Isemann         // Display disassembly
53627caa17caSRaphael Isemann         BreakpointAddrs bp_file_addrs;
53637caa17caSRaphael Isemann         Target *target = exe_ctx.GetTargetPtr();
53647caa17caSRaphael Isemann         if (target) {
53657caa17caSRaphael Isemann           BreakpointList &bp_list = target->GetBreakpointList();
53667caa17caSRaphael Isemann           const size_t num_bps = bp_list.GetSize();
53677caa17caSRaphael Isemann           for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
53687caa17caSRaphael Isemann             BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
53697caa17caSRaphael Isemann             const size_t num_bps_locs = bp_sp->GetNumLocations();
53707caa17caSRaphael Isemann             for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs;
53717caa17caSRaphael Isemann                  ++bp_loc_idx) {
53727caa17caSRaphael Isemann               BreakpointLocationSP bp_loc_sp =
53737caa17caSRaphael Isemann                   bp_sp->GetLocationAtIndex(bp_loc_idx);
53747caa17caSRaphael Isemann               LineEntry bp_loc_line_entry;
53757caa17caSRaphael Isemann               const lldb::addr_t file_addr =
53767caa17caSRaphael Isemann                   bp_loc_sp->GetAddress().GetFileAddress();
53777caa17caSRaphael Isemann               if (file_addr != LLDB_INVALID_ADDRESS) {
53787caa17caSRaphael Isemann                 if (m_disassembly_range.ContainsFileAddress(file_addr))
53797caa17caSRaphael Isemann                   bp_file_addrs.insert(file_addr);
53807caa17caSRaphael Isemann               }
53817caa17caSRaphael Isemann             }
53827caa17caSRaphael Isemann           }
53837caa17caSRaphael Isemann         }
53847caa17caSRaphael Isemann 
53857caa17caSRaphael Isemann         const attr_t selected_highlight_attr = A_REVERSE;
53864a8e4b5cSLuboš Luňák         const attr_t pc_highlight_attr = COLOR_PAIR(WhiteOnBlue);
53877caa17caSRaphael Isemann 
53887caa17caSRaphael Isemann         StreamString strm;
53897caa17caSRaphael Isemann 
53907caa17caSRaphael Isemann         InstructionList &insts = m_disassembly_sp->GetInstructionList();
53917caa17caSRaphael Isemann         Address pc_address;
53927caa17caSRaphael Isemann 
53937caa17caSRaphael Isemann         if (frame_sp)
53947caa17caSRaphael Isemann           pc_address = frame_sp->GetFrameCodeAddress();
53957caa17caSRaphael Isemann         const uint32_t pc_idx =
53967caa17caSRaphael Isemann             pc_address.IsValid()
53977caa17caSRaphael Isemann                 ? insts.GetIndexOfInstructionAtAddress(pc_address)
53987caa17caSRaphael Isemann                 : UINT32_MAX;
53997caa17caSRaphael Isemann         if (set_selected_line_to_pc) {
54007caa17caSRaphael Isemann           m_selected_line = pc_idx;
54017caa17caSRaphael Isemann         }
54027caa17caSRaphael Isemann 
54037caa17caSRaphael Isemann         const uint32_t non_visible_pc_offset = (num_visible_lines / 5);
54047caa17caSRaphael Isemann         if (static_cast<size_t>(m_first_visible_line) >= num_disassembly_lines)
54057caa17caSRaphael Isemann           m_first_visible_line = 0;
54067caa17caSRaphael Isemann 
54077caa17caSRaphael Isemann         if (pc_idx < num_disassembly_lines) {
54087caa17caSRaphael Isemann           if (pc_idx < static_cast<uint32_t>(m_first_visible_line) ||
54097caa17caSRaphael Isemann               pc_idx >= m_first_visible_line + num_visible_lines)
54107caa17caSRaphael Isemann             m_first_visible_line = pc_idx - non_visible_pc_offset;
54117caa17caSRaphael Isemann         }
54127caa17caSRaphael Isemann 
54137caa17caSRaphael Isemann         for (size_t i = 0; i < num_visible_lines; ++i) {
54147caa17caSRaphael Isemann           const uint32_t inst_idx = m_first_visible_line + i;
54157caa17caSRaphael Isemann           Instruction *inst = insts.GetInstructionAtIndex(inst_idx).get();
54167caa17caSRaphael Isemann           if (!inst)
54177caa17caSRaphael Isemann             break;
54187caa17caSRaphael Isemann 
54197caa17caSRaphael Isemann           const int line_y = m_min_y + i;
54207caa17caSRaphael Isemann           window.MoveCursor(1, line_y);
54217caa17caSRaphael Isemann           const bool is_pc_line = frame_sp && inst_idx == pc_idx;
54227caa17caSRaphael Isemann           const bool line_is_selected = m_selected_line == inst_idx;
54237caa17caSRaphael Isemann           // Highlight the line as the PC line first, then if the selected line
54247caa17caSRaphael Isemann           // isn't the same as the PC line, highlight it differently
54257caa17caSRaphael Isemann           attr_t highlight_attr = 0;
54267caa17caSRaphael Isemann           attr_t bp_attr = 0;
54277caa17caSRaphael Isemann           if (is_pc_line)
54287caa17caSRaphael Isemann             highlight_attr = pc_highlight_attr;
54297caa17caSRaphael Isemann           else if (line_is_selected)
54307caa17caSRaphael Isemann             highlight_attr = selected_highlight_attr;
54317caa17caSRaphael Isemann 
54327caa17caSRaphael Isemann           if (bp_file_addrs.find(inst->GetAddress().GetFileAddress()) !=
54337caa17caSRaphael Isemann               bp_file_addrs.end())
54344a8e4b5cSLuboš Luňák             bp_attr = COLOR_PAIR(BlackOnWhite);
54357caa17caSRaphael Isemann 
54367caa17caSRaphael Isemann           if (bp_attr)
54377caa17caSRaphael Isemann             window.AttributeOn(bp_attr);
54387caa17caSRaphael Isemann 
54397caa17caSRaphael Isemann           window.Printf(" 0x%16.16llx ",
54407caa17caSRaphael Isemann                         static_cast<unsigned long long>(
54417caa17caSRaphael Isemann                             inst->GetAddress().GetLoadAddress(target)));
54427caa17caSRaphael Isemann 
54437caa17caSRaphael Isemann           if (bp_attr)
54447caa17caSRaphael Isemann             window.AttributeOff(bp_attr);
54457caa17caSRaphael Isemann 
54467caa17caSRaphael Isemann           window.PutChar(ACS_VLINE);
54477caa17caSRaphael Isemann           // Mark the line with the PC with a diamond
54487caa17caSRaphael Isemann           if (is_pc_line)
54497caa17caSRaphael Isemann             window.PutChar(ACS_DIAMOND);
54507caa17caSRaphael Isemann           else
54517caa17caSRaphael Isemann             window.PutChar(' ');
54527caa17caSRaphael Isemann 
54537caa17caSRaphael Isemann           if (highlight_attr)
54547caa17caSRaphael Isemann             window.AttributeOn(highlight_attr);
54557caa17caSRaphael Isemann 
54567caa17caSRaphael Isemann           const char *mnemonic = inst->GetMnemonic(&exe_ctx);
54577caa17caSRaphael Isemann           const char *operands = inst->GetOperands(&exe_ctx);
54587caa17caSRaphael Isemann           const char *comment = inst->GetComment(&exe_ctx);
54597caa17caSRaphael Isemann 
54607caa17caSRaphael Isemann           if (mnemonic != nullptr && mnemonic[0] == '\0')
54617caa17caSRaphael Isemann             mnemonic = nullptr;
54627caa17caSRaphael Isemann           if (operands != nullptr && operands[0] == '\0')
54637caa17caSRaphael Isemann             operands = nullptr;
54647caa17caSRaphael Isemann           if (comment != nullptr && comment[0] == '\0')
54657caa17caSRaphael Isemann             comment = nullptr;
54667caa17caSRaphael Isemann 
54677caa17caSRaphael Isemann           strm.Clear();
54687caa17caSRaphael Isemann 
54697caa17caSRaphael Isemann           if (mnemonic != nullptr && operands != nullptr && comment != nullptr)
54707caa17caSRaphael Isemann             strm.Printf("%-8s %-25s ; %s", mnemonic, operands, comment);
54717caa17caSRaphael Isemann           else if (mnemonic != nullptr && operands != nullptr)
54727caa17caSRaphael Isemann             strm.Printf("%-8s %s", mnemonic, operands);
54737caa17caSRaphael Isemann           else if (mnemonic != nullptr)
54747caa17caSRaphael Isemann             strm.Printf("%s", mnemonic);
54757caa17caSRaphael Isemann 
54767caa17caSRaphael Isemann           int right_pad = 1;
5477dcd4589aSLuboš Luňák           window.PutCStringTruncated(
5478dcd4589aSLuboš Luňák               right_pad,
5479dcd4589aSLuboš Luňák               strm.GetString().substr(m_first_visible_column).data());
54807caa17caSRaphael Isemann 
54817caa17caSRaphael Isemann           if (is_pc_line && frame_sp &&
54827caa17caSRaphael Isemann               frame_sp->GetConcreteFrameIndex() == 0) {
54837caa17caSRaphael Isemann             StopInfoSP stop_info_sp;
54847caa17caSRaphael Isemann             if (thread)
54857caa17caSRaphael Isemann               stop_info_sp = thread->GetStopInfo();
54867caa17caSRaphael Isemann             if (stop_info_sp) {
54877caa17caSRaphael Isemann               const char *stop_description = stop_info_sp->GetDescription();
54887caa17caSRaphael Isemann               if (stop_description && stop_description[0]) {
54897caa17caSRaphael Isemann                 size_t stop_description_len = strlen(stop_description);
54907caa17caSRaphael Isemann                 int desc_x = window_width - stop_description_len - 16;
5491fc0e8fb7SLuboš Luňák                 if (desc_x - window.GetCursorX() > 0)
54927caa17caSRaphael Isemann                   window.Printf("%*s", desc_x - window.GetCursorX(), "");
5493fc0e8fb7SLuboš Luňák                 window.MoveCursor(window_width - stop_description_len - 15,
5494fc0e8fb7SLuboš Luňák                                   line_y);
5495fc0e8fb7SLuboš Luňák                 window.PrintfTruncated(1, "<<< Thread %u: %s ",
5496fc0e8fb7SLuboš Luňák                                        thread->GetIndexID(), stop_description);
54977caa17caSRaphael Isemann               }
54987caa17caSRaphael Isemann             } else {
54997caa17caSRaphael Isemann               window.Printf("%*s", window_width - window.GetCursorX() - 1, "");
55007caa17caSRaphael Isemann             }
55017caa17caSRaphael Isemann           }
55027caa17caSRaphael Isemann           if (highlight_attr)
55037caa17caSRaphael Isemann             window.AttributeOff(highlight_attr);
55047caa17caSRaphael Isemann         }
55057caa17caSRaphael Isemann       }
55067caa17caSRaphael Isemann     }
55077caa17caSRaphael Isemann     return true; // Drawing handled
55087caa17caSRaphael Isemann   }
55097caa17caSRaphael Isemann 
55107caa17caSRaphael Isemann   size_t GetNumLines() {
55117caa17caSRaphael Isemann     size_t num_lines = GetNumSourceLines();
55127caa17caSRaphael Isemann     if (num_lines == 0)
55137caa17caSRaphael Isemann       num_lines = GetNumDisassemblyLines();
55147caa17caSRaphael Isemann     return num_lines;
55157caa17caSRaphael Isemann   }
55167caa17caSRaphael Isemann 
55177caa17caSRaphael Isemann   size_t GetNumSourceLines() const {
55187caa17caSRaphael Isemann     if (m_file_sp)
55197caa17caSRaphael Isemann       return m_file_sp->GetNumLines();
55207caa17caSRaphael Isemann     return 0;
55217caa17caSRaphael Isemann   }
55227caa17caSRaphael Isemann 
55237caa17caSRaphael Isemann   size_t GetNumDisassemblyLines() const {
55247caa17caSRaphael Isemann     if (m_disassembly_sp)
55257caa17caSRaphael Isemann       return m_disassembly_sp->GetInstructionList().GetSize();
55267caa17caSRaphael Isemann     return 0;
55277caa17caSRaphael Isemann   }
55287caa17caSRaphael Isemann 
55297caa17caSRaphael Isemann   HandleCharResult WindowDelegateHandleChar(Window &window, int c) override {
55307caa17caSRaphael Isemann     const uint32_t num_visible_lines = NumVisibleLines();
55317caa17caSRaphael Isemann     const size_t num_lines = GetNumLines();
55327caa17caSRaphael Isemann 
55337caa17caSRaphael Isemann     switch (c) {
55347caa17caSRaphael Isemann     case ',':
55357caa17caSRaphael Isemann     case KEY_PPAGE:
55367caa17caSRaphael Isemann       // Page up key
55377caa17caSRaphael Isemann       if (static_cast<uint32_t>(m_first_visible_line) > num_visible_lines)
55387caa17caSRaphael Isemann         m_first_visible_line -= num_visible_lines;
55397caa17caSRaphael Isemann       else
55407caa17caSRaphael Isemann         m_first_visible_line = 0;
55417caa17caSRaphael Isemann       m_selected_line = m_first_visible_line;
55427caa17caSRaphael Isemann       return eKeyHandled;
55437caa17caSRaphael Isemann 
55447caa17caSRaphael Isemann     case '.':
55457caa17caSRaphael Isemann     case KEY_NPAGE:
55467caa17caSRaphael Isemann       // Page down key
55477caa17caSRaphael Isemann       {
55487caa17caSRaphael Isemann         if (m_first_visible_line + num_visible_lines < num_lines)
55497caa17caSRaphael Isemann           m_first_visible_line += num_visible_lines;
55507caa17caSRaphael Isemann         else if (num_lines < num_visible_lines)
55517caa17caSRaphael Isemann           m_first_visible_line = 0;
55527caa17caSRaphael Isemann         else
55537caa17caSRaphael Isemann           m_first_visible_line = num_lines - num_visible_lines;
55547caa17caSRaphael Isemann         m_selected_line = m_first_visible_line;
55557caa17caSRaphael Isemann       }
55567caa17caSRaphael Isemann       return eKeyHandled;
55577caa17caSRaphael Isemann 
55587caa17caSRaphael Isemann     case KEY_UP:
55597caa17caSRaphael Isemann       if (m_selected_line > 0) {
55607caa17caSRaphael Isemann         m_selected_line--;
55617caa17caSRaphael Isemann         if (static_cast<size_t>(m_first_visible_line) > m_selected_line)
55627caa17caSRaphael Isemann           m_first_visible_line = m_selected_line;
55637caa17caSRaphael Isemann       }
55647caa17caSRaphael Isemann       return eKeyHandled;
55657caa17caSRaphael Isemann 
55667caa17caSRaphael Isemann     case KEY_DOWN:
55677caa17caSRaphael Isemann       if (m_selected_line + 1 < num_lines) {
55687caa17caSRaphael Isemann         m_selected_line++;
55697caa17caSRaphael Isemann         if (m_first_visible_line + num_visible_lines < m_selected_line)
55707caa17caSRaphael Isemann           m_first_visible_line++;
55717caa17caSRaphael Isemann       }
55727caa17caSRaphael Isemann       return eKeyHandled;
55737caa17caSRaphael Isemann 
5574dcd4589aSLuboš Luňák     case KEY_LEFT:
5575dcd4589aSLuboš Luňák       if (m_first_visible_column > 0)
5576dcd4589aSLuboš Luňák         --m_first_visible_column;
5577dcd4589aSLuboš Luňák       return eKeyHandled;
5578dcd4589aSLuboš Luňák 
5579dcd4589aSLuboš Luňák     case KEY_RIGHT:
5580dcd4589aSLuboš Luňák       ++m_first_visible_column;
5581dcd4589aSLuboš Luňák       return eKeyHandled;
5582dcd4589aSLuboš Luňák 
55837caa17caSRaphael Isemann     case '\r':
55847caa17caSRaphael Isemann     case '\n':
55857caa17caSRaphael Isemann     case KEY_ENTER:
55867caa17caSRaphael Isemann       // Set a breakpoint and run to the line using a one shot breakpoint
55877caa17caSRaphael Isemann       if (GetNumSourceLines() > 0) {
55887caa17caSRaphael Isemann         ExecutionContext exe_ctx =
55897caa17caSRaphael Isemann             m_debugger.GetCommandInterpreter().GetExecutionContext();
55907caa17caSRaphael Isemann         if (exe_ctx.HasProcessScope() && exe_ctx.GetProcessRef().IsAlive()) {
55917caa17caSRaphael Isemann           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
55927caa17caSRaphael Isemann               nullptr, // Don't limit the breakpoint to certain modules
55937caa17caSRaphael Isemann               m_file_sp->GetFileSpec(), // Source file
55947caa17caSRaphael Isemann               m_selected_line +
55957caa17caSRaphael Isemann                   1, // Source line number (m_selected_line is zero based)
55967caa17caSRaphael Isemann               0,     // Unspecified column.
55977caa17caSRaphael Isemann               0,     // No offset
55987caa17caSRaphael Isemann               eLazyBoolCalculate,  // Check inlines using global setting
55997caa17caSRaphael Isemann               eLazyBoolCalculate,  // Skip prologue using global setting,
56007caa17caSRaphael Isemann               false,               // internal
56017caa17caSRaphael Isemann               false,               // request_hardware
56027caa17caSRaphael Isemann               eLazyBoolCalculate); // move_to_nearest_code
56037caa17caSRaphael Isemann           // Make breakpoint one shot
5604cfb96d84SJim Ingham           bp_sp->GetOptions().SetOneShot(true);
56057caa17caSRaphael Isemann           exe_ctx.GetProcessRef().Resume();
56067caa17caSRaphael Isemann         }
56077caa17caSRaphael Isemann       } else if (m_selected_line < GetNumDisassemblyLines()) {
56087caa17caSRaphael Isemann         const Instruction *inst = m_disassembly_sp->GetInstructionList()
56097caa17caSRaphael Isemann                                       .GetInstructionAtIndex(m_selected_line)
56107caa17caSRaphael Isemann                                       .get();
56117caa17caSRaphael Isemann         ExecutionContext exe_ctx =
56127caa17caSRaphael Isemann             m_debugger.GetCommandInterpreter().GetExecutionContext();
56137caa17caSRaphael Isemann         if (exe_ctx.HasTargetScope()) {
56147caa17caSRaphael Isemann           Address addr = inst->GetAddress();
56157caa17caSRaphael Isemann           BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
56167caa17caSRaphael Isemann               addr,   // lldb_private::Address
56177caa17caSRaphael Isemann               false,  // internal
56187caa17caSRaphael Isemann               false); // request_hardware
56197caa17caSRaphael Isemann           // Make breakpoint one shot
5620cfb96d84SJim Ingham           bp_sp->GetOptions().SetOneShot(true);
56217caa17caSRaphael Isemann           exe_ctx.GetProcessRef().Resume();
56227caa17caSRaphael Isemann         }
56237caa17caSRaphael Isemann       }
56247caa17caSRaphael Isemann       return eKeyHandled;
56257caa17caSRaphael Isemann 
56267caa17caSRaphael Isemann     case 'b': // 'b' == toggle breakpoint on currently selected line
5627d6868d9cSLuboš Luňák       ToggleBreakpointOnSelectedLine();
56287caa17caSRaphael Isemann       return eKeyHandled;
56297caa17caSRaphael Isemann 
56307caa17caSRaphael Isemann     case 'D': // 'D' == detach and keep stopped
56317caa17caSRaphael Isemann     {
56327caa17caSRaphael Isemann       ExecutionContext exe_ctx =
56337caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
56347caa17caSRaphael Isemann       if (exe_ctx.HasProcessScope())
563547d2c7cdSLuboš Luňák         exe_ctx.GetProcessRef().Detach(true);
56367caa17caSRaphael Isemann     }
56377caa17caSRaphael Isemann       return eKeyHandled;
56387caa17caSRaphael Isemann 
56397caa17caSRaphael Isemann     case 'c':
56407caa17caSRaphael Isemann       // 'c' == continue
56417caa17caSRaphael Isemann       {
56427caa17caSRaphael Isemann         ExecutionContext exe_ctx =
56437caa17caSRaphael Isemann             m_debugger.GetCommandInterpreter().GetExecutionContext();
56447caa17caSRaphael Isemann         if (exe_ctx.HasProcessScope())
56457caa17caSRaphael Isemann           exe_ctx.GetProcessRef().Resume();
56467caa17caSRaphael Isemann       }
56477caa17caSRaphael Isemann       return eKeyHandled;
56487caa17caSRaphael Isemann 
5649336c7029SLuboš Luňák     case 'f':
5650336c7029SLuboš Luňák       // 'f' == step out (finish)
56517caa17caSRaphael Isemann       {
56527caa17caSRaphael Isemann         ExecutionContext exe_ctx =
56537caa17caSRaphael Isemann             m_debugger.GetCommandInterpreter().GetExecutionContext();
56547caa17caSRaphael Isemann         if (exe_ctx.HasThreadScope() &&
56557caa17caSRaphael Isemann             StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
56567caa17caSRaphael Isemann           exe_ctx.GetThreadRef().StepOut();
56577caa17caSRaphael Isemann         }
56587caa17caSRaphael Isemann       }
56597caa17caSRaphael Isemann       return eKeyHandled;
56607caa17caSRaphael Isemann 
56617caa17caSRaphael Isemann     case 'n': // 'n' == step over
56627caa17caSRaphael Isemann     case 'N': // 'N' == step over instruction
56637caa17caSRaphael Isemann     {
56647caa17caSRaphael Isemann       ExecutionContext exe_ctx =
56657caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
56667caa17caSRaphael Isemann       if (exe_ctx.HasThreadScope() &&
56677caa17caSRaphael Isemann           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
56687caa17caSRaphael Isemann         bool source_step = (c == 'n');
56697caa17caSRaphael Isemann         exe_ctx.GetThreadRef().StepOver(source_step);
56707caa17caSRaphael Isemann       }
56717caa17caSRaphael Isemann     }
56727caa17caSRaphael Isemann       return eKeyHandled;
56737caa17caSRaphael Isemann 
56747caa17caSRaphael Isemann     case 's': // 's' == step into
56757caa17caSRaphael Isemann     case 'S': // 'S' == step into instruction
56767caa17caSRaphael Isemann     {
56777caa17caSRaphael Isemann       ExecutionContext exe_ctx =
56787caa17caSRaphael Isemann           m_debugger.GetCommandInterpreter().GetExecutionContext();
56797caa17caSRaphael Isemann       if (exe_ctx.HasThreadScope() &&
56807caa17caSRaphael Isemann           StateIsStoppedState(exe_ctx.GetProcessRef().GetState(), true)) {
56817caa17caSRaphael Isemann         bool source_step = (c == 's');
56827caa17caSRaphael Isemann         exe_ctx.GetThreadRef().StepIn(source_step);
56837caa17caSRaphael Isemann       }
56847caa17caSRaphael Isemann     }
56857caa17caSRaphael Isemann       return eKeyHandled;
56867caa17caSRaphael Isemann 
568713978643SLuboš Luňák     case 'u': // 'u' == frame up
568813978643SLuboš Luňák     case 'd': // 'd' == frame down
568913978643SLuboš Luňák     {
569013978643SLuboš Luňák       ExecutionContext exe_ctx =
569113978643SLuboš Luňák           m_debugger.GetCommandInterpreter().GetExecutionContext();
569213978643SLuboš Luňák       if (exe_ctx.HasThreadScope()) {
569313978643SLuboš Luňák         Thread *thread = exe_ctx.GetThreadPtr();
569413978643SLuboš Luňák         uint32_t frame_idx = thread->GetSelectedFrameIndex();
569513978643SLuboš Luňák         if (frame_idx == UINT32_MAX)
569613978643SLuboš Luňák           frame_idx = 0;
569713978643SLuboš Luňák         if (c == 'u' && frame_idx + 1 < thread->GetStackFrameCount())
569813978643SLuboš Luňák           ++frame_idx;
569913978643SLuboš Luňák         else if (c == 'd' && frame_idx > 0)
570013978643SLuboš Luňák           --frame_idx;
570113978643SLuboš Luňák         if (thread->SetSelectedFrameByIndex(frame_idx, true))
570213978643SLuboš Luňák           exe_ctx.SetFrameSP(thread->GetSelectedFrame());
570313978643SLuboš Luňák       }
570413978643SLuboš Luňák     }
570513978643SLuboš Luňák       return eKeyHandled;
570613978643SLuboš Luňák 
57077caa17caSRaphael Isemann     case 'h':
57087caa17caSRaphael Isemann       window.CreateHelpSubwindow();
57097caa17caSRaphael Isemann       return eKeyHandled;
57107caa17caSRaphael Isemann 
57117caa17caSRaphael Isemann     default:
57127caa17caSRaphael Isemann       break;
57137caa17caSRaphael Isemann     }
57147caa17caSRaphael Isemann     return eKeyNotHandled;
57157caa17caSRaphael Isemann   }
57167caa17caSRaphael Isemann 
5717d6868d9cSLuboš Luňák   void ToggleBreakpointOnSelectedLine() {
5718d6868d9cSLuboš Luňák     ExecutionContext exe_ctx =
5719d6868d9cSLuboš Luňák         m_debugger.GetCommandInterpreter().GetExecutionContext();
5720d6868d9cSLuboš Luňák     if (!exe_ctx.HasTargetScope())
5721d6868d9cSLuboš Luňák       return;
5722d6868d9cSLuboš Luňák     if (GetNumSourceLines() > 0) {
5723d6868d9cSLuboš Luňák       // Source file breakpoint.
5724d6868d9cSLuboš Luňák       BreakpointList &bp_list = exe_ctx.GetTargetRef().GetBreakpointList();
5725d6868d9cSLuboš Luňák       const size_t num_bps = bp_list.GetSize();
5726d6868d9cSLuboš Luňák       for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
5727d6868d9cSLuboš Luňák         BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
5728d6868d9cSLuboš Luňák         const size_t num_bps_locs = bp_sp->GetNumLocations();
5729d6868d9cSLuboš Luňák         for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
5730d6868d9cSLuboš Luňák           BreakpointLocationSP bp_loc_sp =
5731d6868d9cSLuboš Luňák               bp_sp->GetLocationAtIndex(bp_loc_idx);
5732d6868d9cSLuboš Luňák           LineEntry bp_loc_line_entry;
5733d6868d9cSLuboš Luňák           if (bp_loc_sp->GetAddress().CalculateSymbolContextLineEntry(
5734d6868d9cSLuboš Luňák                   bp_loc_line_entry)) {
5735d6868d9cSLuboš Luňák             if (m_file_sp->GetFileSpec() == bp_loc_line_entry.file &&
5736d6868d9cSLuboš Luňák                 m_selected_line + 1 == bp_loc_line_entry.line) {
5737d6868d9cSLuboš Luňák               bool removed =
5738d6868d9cSLuboš Luňák                   exe_ctx.GetTargetRef().RemoveBreakpointByID(bp_sp->GetID());
5739d6868d9cSLuboš Luňák               assert(removed);
5740d6868d9cSLuboš Luňák               UNUSED_IF_ASSERT_DISABLED(removed);
5741d6868d9cSLuboš Luňák               return; // Existing breakpoint removed.
5742d6868d9cSLuboš Luňák             }
5743d6868d9cSLuboš Luňák           }
5744d6868d9cSLuboš Luňák         }
5745d6868d9cSLuboš Luňák       }
5746d6868d9cSLuboš Luňák       // No breakpoint found on the location, add it.
5747d6868d9cSLuboš Luňák       BreakpointSP bp_sp = exe_ctx.GetTargetRef().CreateBreakpoint(
5748d6868d9cSLuboš Luňák           nullptr, // Don't limit the breakpoint to certain modules
5749d6868d9cSLuboš Luňák           m_file_sp->GetFileSpec(), // Source file
5750d6868d9cSLuboš Luňák           m_selected_line +
5751d6868d9cSLuboš Luňák               1, // Source line number (m_selected_line is zero based)
5752d6868d9cSLuboš Luňák           0,     // No column specified.
5753d6868d9cSLuboš Luňák           0,     // No offset
5754d6868d9cSLuboš Luňák           eLazyBoolCalculate,  // Check inlines using global setting
5755d6868d9cSLuboš Luňák           eLazyBoolCalculate,  // Skip prologue using global setting,
5756d6868d9cSLuboš Luňák           false,               // internal
5757d6868d9cSLuboš Luňák           false,               // request_hardware
5758d6868d9cSLuboš Luňák           eLazyBoolCalculate); // move_to_nearest_code
5759d6868d9cSLuboš Luňák     } else {
5760d6868d9cSLuboš Luňák       // Disassembly breakpoint.
5761d6868d9cSLuboš Luňák       assert(GetNumDisassemblyLines() > 0);
5762d6868d9cSLuboš Luňák       assert(m_selected_line < GetNumDisassemblyLines());
5763d6868d9cSLuboš Luňák       const Instruction *inst = m_disassembly_sp->GetInstructionList()
5764d6868d9cSLuboš Luňák                                     .GetInstructionAtIndex(m_selected_line)
5765d6868d9cSLuboš Luňák                                     .get();
5766d6868d9cSLuboš Luňák       Address addr = inst->GetAddress();
5767d6868d9cSLuboš Luňák       // Try to find it.
5768d6868d9cSLuboš Luňák       BreakpointList &bp_list = exe_ctx.GetTargetRef().GetBreakpointList();
5769d6868d9cSLuboš Luňák       const size_t num_bps = bp_list.GetSize();
5770d6868d9cSLuboš Luňák       for (size_t bp_idx = 0; bp_idx < num_bps; ++bp_idx) {
5771d6868d9cSLuboš Luňák         BreakpointSP bp_sp = bp_list.GetBreakpointAtIndex(bp_idx);
5772d6868d9cSLuboš Luňák         const size_t num_bps_locs = bp_sp->GetNumLocations();
5773d6868d9cSLuboš Luňák         for (size_t bp_loc_idx = 0; bp_loc_idx < num_bps_locs; ++bp_loc_idx) {
5774d6868d9cSLuboš Luňák           BreakpointLocationSP bp_loc_sp =
5775d6868d9cSLuboš Luňák               bp_sp->GetLocationAtIndex(bp_loc_idx);
5776d6868d9cSLuboš Luňák           LineEntry bp_loc_line_entry;
5777d6868d9cSLuboš Luňák           const lldb::addr_t file_addr =
5778d6868d9cSLuboš Luňák               bp_loc_sp->GetAddress().GetFileAddress();
5779d6868d9cSLuboš Luňák           if (file_addr == addr.GetFileAddress()) {
5780d6868d9cSLuboš Luňák             bool removed =
5781d6868d9cSLuboš Luňák                 exe_ctx.GetTargetRef().RemoveBreakpointByID(bp_sp->GetID());
5782d6868d9cSLuboš Luňák             assert(removed);
5783d6868d9cSLuboš Luňák             UNUSED_IF_ASSERT_DISABLED(removed);
5784d6868d9cSLuboš Luňák             return; // Existing breakpoint removed.
5785d6868d9cSLuboš Luňák           }
5786d6868d9cSLuboš Luňák         }
5787d6868d9cSLuboš Luňák       }
5788d6868d9cSLuboš Luňák       // No breakpoint found on the address, add it.
5789d6868d9cSLuboš Luňák       BreakpointSP bp_sp =
5790d6868d9cSLuboš Luňák           exe_ctx.GetTargetRef().CreateBreakpoint(addr, // lldb_private::Address
5791d6868d9cSLuboš Luňák                                                   false,  // internal
5792d6868d9cSLuboš Luňák                                                   false); // request_hardware
5793d6868d9cSLuboš Luňák     }
5794d6868d9cSLuboš Luňák   }
5795d6868d9cSLuboš Luňák 
57967caa17caSRaphael Isemann protected:
57977caa17caSRaphael Isemann   typedef std::set<uint32_t> BreakpointLines;
57987caa17caSRaphael Isemann   typedef std::set<lldb::addr_t> BreakpointAddrs;
57997caa17caSRaphael Isemann 
58007caa17caSRaphael Isemann   Debugger &m_debugger;
58017caa17caSRaphael Isemann   SymbolContext m_sc;
58027caa17caSRaphael Isemann   SourceManager::FileSP m_file_sp;
58037caa17caSRaphael Isemann   SymbolContextScope *m_disassembly_scope;
58047caa17caSRaphael Isemann   lldb::DisassemblerSP m_disassembly_sp;
58057caa17caSRaphael Isemann   AddressRange m_disassembly_range;
58067caa17caSRaphael Isemann   StreamString m_title;
58077caa17caSRaphael Isemann   lldb::user_id_t m_tid;
58087caa17caSRaphael Isemann   int m_line_width;
58097caa17caSRaphael Isemann   uint32_t m_selected_line; // The selected line
58107caa17caSRaphael Isemann   uint32_t m_pc_line;       // The line with the PC
58117caa17caSRaphael Isemann   uint32_t m_stop_id;
58127caa17caSRaphael Isemann   uint32_t m_frame_idx;
58137caa17caSRaphael Isemann   int m_first_visible_line;
5814dcd4589aSLuboš Luňák   int m_first_visible_column;
58157caa17caSRaphael Isemann   int m_min_x;
58167caa17caSRaphael Isemann   int m_min_y;
58177caa17caSRaphael Isemann   int m_max_x;
58187caa17caSRaphael Isemann   int m_max_y;
58197caa17caSRaphael Isemann };
58207caa17caSRaphael Isemann 
58217caa17caSRaphael Isemann DisplayOptions ValueObjectListDelegate::g_options = {true};
58227caa17caSRaphael Isemann 
58237caa17caSRaphael Isemann IOHandlerCursesGUI::IOHandlerCursesGUI(Debugger &debugger)
58247caa17caSRaphael Isemann     : IOHandler(debugger, IOHandler::Type::Curses) {}
58257caa17caSRaphael Isemann 
58267caa17caSRaphael Isemann void IOHandlerCursesGUI::Activate() {
58277caa17caSRaphael Isemann   IOHandler::Activate();
58287caa17caSRaphael Isemann   if (!m_app_ap) {
582906412daeSJonas Devlieghere     m_app_ap = std::make_unique<Application>(GetInputFILE(), GetOutputFILE());
58307caa17caSRaphael Isemann 
58317caa17caSRaphael Isemann     // This is both a window and a menu delegate
58327caa17caSRaphael Isemann     std::shared_ptr<ApplicationDelegate> app_delegate_sp(
58337caa17caSRaphael Isemann         new ApplicationDelegate(*m_app_ap, m_debugger));
58347caa17caSRaphael Isemann 
58357caa17caSRaphael Isemann     MenuDelegateSP app_menu_delegate_sp =
58367caa17caSRaphael Isemann         std::static_pointer_cast<MenuDelegate>(app_delegate_sp);
58377caa17caSRaphael Isemann     MenuSP lldb_menu_sp(
58387caa17caSRaphael Isemann         new Menu("LLDB", "F1", KEY_F(1), ApplicationDelegate::eMenuID_LLDB));
58397caa17caSRaphael Isemann     MenuSP exit_menuitem_sp(
58407caa17caSRaphael Isemann         new Menu("Exit", nullptr, 'x', ApplicationDelegate::eMenuID_LLDBExit));
58417caa17caSRaphael Isemann     exit_menuitem_sp->SetCannedResult(MenuActionResult::Quit);
58427caa17caSRaphael Isemann     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(
58437caa17caSRaphael Isemann         "About LLDB", nullptr, 'a', ApplicationDelegate::eMenuID_LLDBAbout)));
58447caa17caSRaphael Isemann     lldb_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
58457caa17caSRaphael Isemann     lldb_menu_sp->AddSubmenu(exit_menuitem_sp);
58467caa17caSRaphael Isemann 
58477caa17caSRaphael Isemann     MenuSP target_menu_sp(new Menu("Target", "F2", KEY_F(2),
58487caa17caSRaphael Isemann                                    ApplicationDelegate::eMenuID_Target));
58497caa17caSRaphael Isemann     target_menu_sp->AddSubmenu(MenuSP(new Menu(
58507caa17caSRaphael Isemann         "Create", nullptr, 'c', ApplicationDelegate::eMenuID_TargetCreate)));
58517caa17caSRaphael Isemann     target_menu_sp->AddSubmenu(MenuSP(new Menu(
58527caa17caSRaphael Isemann         "Delete", nullptr, 'd', ApplicationDelegate::eMenuID_TargetDelete)));
58537caa17caSRaphael Isemann 
58547caa17caSRaphael Isemann     MenuSP process_menu_sp(new Menu("Process", "F3", KEY_F(3),
58557caa17caSRaphael Isemann                                     ApplicationDelegate::eMenuID_Process));
58567caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(MenuSP(new Menu(
58577caa17caSRaphael Isemann         "Attach", nullptr, 'a', ApplicationDelegate::eMenuID_ProcessAttach)));
585847d2c7cdSLuboš Luňák     process_menu_sp->AddSubmenu(
585947d2c7cdSLuboš Luňák         MenuSP(new Menu("Detach and resume", nullptr, 'd',
586047d2c7cdSLuboš Luňák                         ApplicationDelegate::eMenuID_ProcessDetachResume)));
586147d2c7cdSLuboš Luňák     process_menu_sp->AddSubmenu(
586247d2c7cdSLuboš Luňák         MenuSP(new Menu("Detach suspended", nullptr, 's',
586347d2c7cdSLuboš Luňák                         ApplicationDelegate::eMenuID_ProcessDetachSuspended)));
58647caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(MenuSP(new Menu(
58657caa17caSRaphael Isemann         "Launch", nullptr, 'l', ApplicationDelegate::eMenuID_ProcessLaunch)));
58667caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(MenuSP(new Menu(Menu::Type::Separator)));
58677caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(
58687caa17caSRaphael Isemann         MenuSP(new Menu("Continue", nullptr, 'c',
58697caa17caSRaphael Isemann                         ApplicationDelegate::eMenuID_ProcessContinue)));
58707caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(MenuSP(new Menu(
58717caa17caSRaphael Isemann         "Halt", nullptr, 'h', ApplicationDelegate::eMenuID_ProcessHalt)));
58727caa17caSRaphael Isemann     process_menu_sp->AddSubmenu(MenuSP(new Menu(
58737caa17caSRaphael Isemann         "Kill", nullptr, 'k', ApplicationDelegate::eMenuID_ProcessKill)));
58747caa17caSRaphael Isemann 
58757caa17caSRaphael Isemann     MenuSP thread_menu_sp(new Menu("Thread", "F4", KEY_F(4),
58767caa17caSRaphael Isemann                                    ApplicationDelegate::eMenuID_Thread));
58777caa17caSRaphael Isemann     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
58787caa17caSRaphael Isemann         "Step In", nullptr, 'i', ApplicationDelegate::eMenuID_ThreadStepIn)));
58797caa17caSRaphael Isemann     thread_menu_sp->AddSubmenu(
58807caa17caSRaphael Isemann         MenuSP(new Menu("Step Over", nullptr, 'v',
58817caa17caSRaphael Isemann                         ApplicationDelegate::eMenuID_ThreadStepOver)));
58827caa17caSRaphael Isemann     thread_menu_sp->AddSubmenu(MenuSP(new Menu(
58837caa17caSRaphael Isemann         "Step Out", nullptr, 'o', ApplicationDelegate::eMenuID_ThreadStepOut)));
58847caa17caSRaphael Isemann 
58857caa17caSRaphael Isemann     MenuSP view_menu_sp(
58867caa17caSRaphael Isemann         new Menu("View", "F5", KEY_F(5), ApplicationDelegate::eMenuID_View));
58877caa17caSRaphael Isemann     view_menu_sp->AddSubmenu(
58887caa17caSRaphael Isemann         MenuSP(new Menu("Backtrace", nullptr, 'b',
58897caa17caSRaphael Isemann                         ApplicationDelegate::eMenuID_ViewBacktrace)));
58907caa17caSRaphael Isemann     view_menu_sp->AddSubmenu(
58917caa17caSRaphael Isemann         MenuSP(new Menu("Registers", nullptr, 'r',
58927caa17caSRaphael Isemann                         ApplicationDelegate::eMenuID_ViewRegisters)));
58937caa17caSRaphael Isemann     view_menu_sp->AddSubmenu(MenuSP(new Menu(
58947caa17caSRaphael Isemann         "Source", nullptr, 's', ApplicationDelegate::eMenuID_ViewSource)));
58957caa17caSRaphael Isemann     view_menu_sp->AddSubmenu(
58967caa17caSRaphael Isemann         MenuSP(new Menu("Variables", nullptr, 'v',
58977caa17caSRaphael Isemann                         ApplicationDelegate::eMenuID_ViewVariables)));
58987caa17caSRaphael Isemann 
58997caa17caSRaphael Isemann     MenuSP help_menu_sp(
59007caa17caSRaphael Isemann         new Menu("Help", "F6", KEY_F(6), ApplicationDelegate::eMenuID_Help));
59017caa17caSRaphael Isemann     help_menu_sp->AddSubmenu(MenuSP(new Menu(
59027caa17caSRaphael Isemann         "GUI Help", nullptr, 'g', ApplicationDelegate::eMenuID_HelpGUIHelp)));
59037caa17caSRaphael Isemann 
59047caa17caSRaphael Isemann     m_app_ap->Initialize();
59057caa17caSRaphael Isemann     WindowSP &main_window_sp = m_app_ap->GetMainWindow();
59067caa17caSRaphael Isemann 
59077caa17caSRaphael Isemann     MenuSP menubar_sp(new Menu(Menu::Type::Bar));
59087caa17caSRaphael Isemann     menubar_sp->AddSubmenu(lldb_menu_sp);
59097caa17caSRaphael Isemann     menubar_sp->AddSubmenu(target_menu_sp);
59107caa17caSRaphael Isemann     menubar_sp->AddSubmenu(process_menu_sp);
59117caa17caSRaphael Isemann     menubar_sp->AddSubmenu(thread_menu_sp);
59127caa17caSRaphael Isemann     menubar_sp->AddSubmenu(view_menu_sp);
59137caa17caSRaphael Isemann     menubar_sp->AddSubmenu(help_menu_sp);
59147caa17caSRaphael Isemann     menubar_sp->SetDelegate(app_menu_delegate_sp);
59157caa17caSRaphael Isemann 
59167caa17caSRaphael Isemann     Rect content_bounds = main_window_sp->GetFrame();
59177caa17caSRaphael Isemann     Rect menubar_bounds = content_bounds.MakeMenuBar();
59187caa17caSRaphael Isemann     Rect status_bounds = content_bounds.MakeStatusBar();
59197caa17caSRaphael Isemann     Rect source_bounds;
59207caa17caSRaphael Isemann     Rect variables_bounds;
59217caa17caSRaphael Isemann     Rect threads_bounds;
59227caa17caSRaphael Isemann     Rect source_variables_bounds;
59237caa17caSRaphael Isemann     content_bounds.VerticalSplitPercentage(0.80, source_variables_bounds,
59247caa17caSRaphael Isemann                                            threads_bounds);
59257caa17caSRaphael Isemann     source_variables_bounds.HorizontalSplitPercentage(0.70, source_bounds,
59267caa17caSRaphael Isemann                                                       variables_bounds);
59277caa17caSRaphael Isemann 
59287caa17caSRaphael Isemann     WindowSP menubar_window_sp =
59297caa17caSRaphael Isemann         main_window_sp->CreateSubWindow("Menubar", menubar_bounds, false);
59307caa17caSRaphael Isemann     // Let the menubar get keys if the active window doesn't handle the keys
59317caa17caSRaphael Isemann     // that are typed so it can respond to menubar key presses.
59327caa17caSRaphael Isemann     menubar_window_sp->SetCanBeActive(
59337caa17caSRaphael Isemann         false); // Don't let the menubar become the active window
59347caa17caSRaphael Isemann     menubar_window_sp->SetDelegate(menubar_sp);
59357caa17caSRaphael Isemann 
59367caa17caSRaphael Isemann     WindowSP source_window_sp(
59377caa17caSRaphael Isemann         main_window_sp->CreateSubWindow("Source", source_bounds, true));
59387caa17caSRaphael Isemann     WindowSP variables_window_sp(
59397caa17caSRaphael Isemann         main_window_sp->CreateSubWindow("Variables", variables_bounds, false));
59407caa17caSRaphael Isemann     WindowSP threads_window_sp(
59417caa17caSRaphael Isemann         main_window_sp->CreateSubWindow("Threads", threads_bounds, false));
59427caa17caSRaphael Isemann     WindowSP status_window_sp(
59437caa17caSRaphael Isemann         main_window_sp->CreateSubWindow("Status", status_bounds, false));
59447caa17caSRaphael Isemann     status_window_sp->SetCanBeActive(
59457caa17caSRaphael Isemann         false); // Don't let the status bar become the active window
59467caa17caSRaphael Isemann     main_window_sp->SetDelegate(
59477caa17caSRaphael Isemann         std::static_pointer_cast<WindowDelegate>(app_delegate_sp));
59487caa17caSRaphael Isemann     source_window_sp->SetDelegate(
59497caa17caSRaphael Isemann         WindowDelegateSP(new SourceFileWindowDelegate(m_debugger)));
59507caa17caSRaphael Isemann     variables_window_sp->SetDelegate(
59517caa17caSRaphael Isemann         WindowDelegateSP(new FrameVariablesWindowDelegate(m_debugger)));
59527caa17caSRaphael Isemann     TreeDelegateSP thread_delegate_sp(new ThreadsTreeDelegate(m_debugger));
59537caa17caSRaphael Isemann     threads_window_sp->SetDelegate(WindowDelegateSP(
59547caa17caSRaphael Isemann         new TreeWindowDelegate(m_debugger, thread_delegate_sp)));
59557caa17caSRaphael Isemann     status_window_sp->SetDelegate(
59567caa17caSRaphael Isemann         WindowDelegateSP(new StatusBarWindowDelegate(m_debugger)));
59577caa17caSRaphael Isemann 
59587caa17caSRaphael Isemann     // Show the main help window once the first time the curses GUI is launched
59597caa17caSRaphael Isemann     static bool g_showed_help = false;
59607caa17caSRaphael Isemann     if (!g_showed_help) {
59617caa17caSRaphael Isemann       g_showed_help = true;
59627caa17caSRaphael Isemann       main_window_sp->CreateHelpSubwindow();
59637caa17caSRaphael Isemann     }
59647caa17caSRaphael Isemann 
596514406ca0SLuboš Luňák     // All colors with black background.
596614406ca0SLuboš Luňák     init_pair(1, COLOR_BLACK, COLOR_BLACK);
596714406ca0SLuboš Luňák     init_pair(2, COLOR_RED, COLOR_BLACK);
596814406ca0SLuboš Luňák     init_pair(3, COLOR_GREEN, COLOR_BLACK);
596914406ca0SLuboš Luňák     init_pair(4, COLOR_YELLOW, COLOR_BLACK);
597014406ca0SLuboš Luňák     init_pair(5, COLOR_BLUE, COLOR_BLACK);
597114406ca0SLuboš Luňák     init_pair(6, COLOR_MAGENTA, COLOR_BLACK);
597214406ca0SLuboš Luňák     init_pair(7, COLOR_CYAN, COLOR_BLACK);
597314406ca0SLuboš Luňák     init_pair(8, COLOR_WHITE, COLOR_BLACK);
597414406ca0SLuboš Luňák     // All colors with blue background.
597514406ca0SLuboš Luňák     init_pair(9, COLOR_BLACK, COLOR_BLUE);
597614406ca0SLuboš Luňák     init_pair(10, COLOR_RED, COLOR_BLUE);
597714406ca0SLuboš Luňák     init_pair(11, COLOR_GREEN, COLOR_BLUE);
597814406ca0SLuboš Luňák     init_pair(12, COLOR_YELLOW, COLOR_BLUE);
597914406ca0SLuboš Luňák     init_pair(13, COLOR_BLUE, COLOR_BLUE);
598014406ca0SLuboš Luňák     init_pair(14, COLOR_MAGENTA, COLOR_BLUE);
598114406ca0SLuboš Luňák     init_pair(15, COLOR_CYAN, COLOR_BLUE);
598214406ca0SLuboš Luňák     init_pair(16, COLOR_WHITE, COLOR_BLUE);
59834a8e4b5cSLuboš Luňák     // These must match the order in the color indexes enum.
59844a8e4b5cSLuboš Luňák     init_pair(17, COLOR_BLACK, COLOR_WHITE);
59854a8e4b5cSLuboš Luňák     init_pair(18, COLOR_MAGENTA, COLOR_WHITE);
59864a8e4b5cSLuboš Luňák     static_assert(LastColorPairIndex == 18, "Color indexes do not match.");
598729cc50e1SOmar Emara 
598829cc50e1SOmar Emara     define_key("\033[Z", KEY_SHIFT_TAB);
59897caa17caSRaphael Isemann   }
59907caa17caSRaphael Isemann }
59917caa17caSRaphael Isemann 
59927caa17caSRaphael Isemann void IOHandlerCursesGUI::Deactivate() { m_app_ap->Terminate(); }
59937caa17caSRaphael Isemann 
59947caa17caSRaphael Isemann void IOHandlerCursesGUI::Run() {
59957caa17caSRaphael Isemann   m_app_ap->Run(m_debugger);
59967caa17caSRaphael Isemann   SetIsDone(true);
59977caa17caSRaphael Isemann }
59987caa17caSRaphael Isemann 
59997caa17caSRaphael Isemann IOHandlerCursesGUI::~IOHandlerCursesGUI() = default;
60007caa17caSRaphael Isemann 
60017caa17caSRaphael Isemann void IOHandlerCursesGUI::Cancel() {}
60027caa17caSRaphael Isemann 
60037caa17caSRaphael Isemann bool IOHandlerCursesGUI::Interrupt() { return false; }
60047caa17caSRaphael Isemann 
60057caa17caSRaphael Isemann void IOHandlerCursesGUI::GotEOF() {}
60067caa17caSRaphael Isemann 
60072f1b24b7SLuboš Luňák void IOHandlerCursesGUI::TerminalSizeChanged() {
60082f1b24b7SLuboš Luňák   m_app_ap->TerminalSizeChanged();
60092f1b24b7SLuboš Luňák }
60102f1b24b7SLuboš Luňák 
6011a4304f96SJonas Devlieghere #endif // LLDB_ENABLE_CURSES
6012