add opened tabs to state

4 files changed, 36 insertions, 0 deletions

diff --git a/src/components/channeltabswitcherhandy.cpp b/src/components/channeltabswitcherhandy.cpp
index a9d2aef..bb28272 100644
--- a/src/components/channeltabswitcherhandy.cpp
+++ b/src/components/channeltabswitcherhandy.cpp
@@ -81,6 +81,28 @@ void ChannelTabSwitcherHandy::ReplaceActiveTab(Snowflake id) {
     }
 }
 
+TabsState ChannelTabSwitcherHandy::GetTabsState() {
+    TabsState state;
+
+    const gint num_pages = hdy_tab_view_get_n_pages(m_tab_view);
+    for (gint i = 0; i < num_pages; i++) {
+        auto *page = hdy_tab_view_get_nth_page(m_tab_view, i);
+        if (page != nullptr) {
+            if (const auto it = m_pages_rev.find(page); it != m_pages_rev.end()) {
+                state.Channels.push_back(it->second);
+            }
+        }
+    }
+
+    return state;
+}
+
+void ChannelTabSwitcherHandy::UseTabsState(const TabsState &state) {
+    for (auto id : state.Channels) {
+        AddChannelTab(id);
+    }
+}
+
 void ChannelTabSwitcherHandy::CheckUnread(Snowflake id) {
     if (auto it = m_pages.find(id); it != m_pages.end()) {
         hdy_tab_page_set_needs_attention(it->second, Abaddon::Get().GetDiscordClient().GetUnreadStateForChannel(id) > -1);
diff --git a/src/components/channeltabswitcherhandy.hpp b/src/components/channeltabswitcherhandy.hpp
index 6a2dbff..41c9c92 100644
--- a/src/components/channeltabswitcherhandy.hpp
+++ b/src/components/channeltabswitcherhandy.hpp
@@ -5,6 +5,7 @@
     #include <unordered_map>
     #include <handy.h>
     #include "discord/snowflake.hpp"
+    #include "state.hpp"
 
 class ChannelData;
 
@@ -18,6 +19,8 @@ public:
     void AddChannelTab(Snowflake id);
     // switches to existing tab if it exists
     void ReplaceActiveTab(Snowflake id);
+    TabsState GetTabsState();
+    void UseTabsState(const TabsState &state);
 
 private:
     void CheckUnread(Snowflake id);
diff --git a/src/components/chatwindow.cpp b/src/components/chatwindow.cpp
index 99ec8a0..f51b58f 100644
--- a/src/components/chatwindow.cpp
+++ b/src/components/chatwindow.cpp
@@ -172,6 +172,14 @@ void ChatWindow::SetTopic(const std::string &text) {
 void ChatWindow::OpenNewTab(Snowflake id) {
     m_tab_switcher->AddChannelTab(id);
 }
+
+TabsState ChatWindow::GetTabsState() {
+    return m_tab_switcher->GetTabsState();
+}
+
+void ChatWindow::UseTabsState(const TabsState &state) {
+    m_tab_switcher->UseTabsState(state);
+}
 #endif
 
 Snowflake ChatWindow::GetActiveChannel() const {
diff --git a/src/components/chatwindow.hpp b/src/components/chatwindow.hpp
index d77afec..d09a87b 100644
--- a/src/components/chatwindow.hpp
+++ b/src/components/chatwindow.hpp
@@ -4,6 +4,7 @@
 #include <set>
 #include "discord/discord.hpp"
 #include "completer.hpp"
+#include "state.hpp"
 
 #ifdef WITH_LIBHANDY
 class ChannelTabSwitcherHandy;
@@ -36,6 +37,8 @@ public:
 
 #ifdef WITH_LIBHANDY
     void OpenNewTab(Snowflake id);
+    TabsState GetTabsState();
+    void UseTabsState(const TabsState &state);
 #endif
 
 protected: