From 03edea82575743d6e098252db8f75da64642ca47 Mon Sep 17 00:00:00 2001
From: Myk Taylor <myk002@yahoo.com>
Date: Mon, 7 Aug 2023 02:50:37 -0700
Subject: [PATCH] add focus string support for legends mode

---
 library/modules/Gui.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/library/modules/Gui.cpp b/library/modules/Gui.cpp
index 102fef9d6..a8941fee7 100644
--- a/library/modules/Gui.cpp
+++ b/library/modules/Gui.cpp
@@ -69,6 +69,7 @@ using namespace DFHack;
 #include "df/item_corpsepiecest.h"
 #include "df/item_corpsest.h"
 #include "df/job.h"
+#include "df/legend_pagest.h"
 #include "df/occupation.h"
 #include "df/plant.h"
 #include "df/popup_message.h"
@@ -84,6 +85,7 @@ using namespace DFHack;
 #include "df/unit.h"
 #include "df/unit_inventory_item.h"
 #include "df/viewscreen_dwarfmodest.h"
+#include "df/viewscreen_legendsst.h"
 #include "df/viewscreen_new_regionst.h"
 #include "df/viewscreen_titlest.h"
 #include "df/world.h"
@@ -170,6 +172,16 @@ DEFINE_GET_FOCUS_STRING_HANDLER(new_region)
         focusStrings.push_back(baseFocus);
 }
 
+DEFINE_GET_FOCUS_STRING_HANDLER(legends)
+{
+    if (screen->init_stage != -1)
+        focusStrings.push_back(baseFocus + "/Loading");
+    else if (screen->page.size() <= 1)
+        focusStrings.push_back(baseFocus + "/Default");
+    else
+        focusStrings.push_back(baseFocus + '/' + screen->page[screen->active_page_index]->header);
+}
+
 DEFINE_GET_FOCUS_STRING_HANDLER(dwarfmode)
 {
     std::string newFocusString;