add storage and accessors for pins

black-sliver · Nov 18, 2023 · 1d4ced7 · 1d4ced7
1 parent 28b72dc
commit 1d4ced7
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 6 deletions.
diff --git a/src/core/tracker.cpp b/src/core/tracker.cpp
@@ -1,4 +1,5 @@
 #include "tracker.h"
+#include <algorithm>
 #include <luaglue/luamethod.h>
 #include <cstring>
 #include <nlohmann/json.hpp>
@@ -426,11 +427,11 @@ const BaseItem& Tracker::getItemByCode(const std::string& code) const
     for (const auto& item: _jsonItems) {
         if (item.canProvideCode(code)) return item;
     }
-    
+
     for (const auto& item: _luaItems) {
         if (item.canProvideCode(code)) return item;
     }
-    
+
     return blankItem;
 }
 BaseItem& Tracker::getItemById(const std::string& id)
@@ -498,6 +499,29 @@ LocationSection& Tracker::getLocationSection(const std::string& id)
     return blankLocationSection;
 }
 
+bool Tracker::isPinned(const std::string& id) const
+{
+    return std::find(_pins.begin(), _pins.end(), id) != _pins.end();
+}
+
+void Tracker::setPinned(const std::string& id, bool pinned)
+{
+    auto it = std::find(_pins.begin(), _pins.end(), id);
+    if (pinned && it == _pins.end()) {
+        _pins.push_back(id);
+        onPinChanged.emit(this, id, pinned);
+    }
+    else if (!pinned && it != _pins.end()) {
+        onPinChanged.emit(this, id, pinned);
+        _pins.erase(it);
+    }
+}
+
+const std::list<std::string>& Tracker::getPins() const
+{
+    return _pins;
+}
+
 const Pack* Tracker::getPack() const
 {
     return _pack;
@@ -831,16 +855,17 @@ json Tracker::saveState() const
             jSections[id] = sec.save();
         }
     }
-    
+
     json state = { {
         "tracker", {
             {"format_version", 1},
             {"json_items", jJsonItems},
             {"lua_items", jLuaItems},
-            {"sections", jSections}
+            {"sections", jSections},
+            {"pins", _pins},
         }
     } };
-    
+
     return state;
 }
 
@@ -849,6 +874,10 @@ bool Tracker::loadState(nlohmann::json& state)
     _reachableCache.clear();
     _providerCountCache.clear();
     _bulkItemUpdates.clear();
+    for (const auto& pin: _pins)
+        onPinChanged.emit(this, pin, false);
+    _pins.clear();
+
     if (state.type() != json::value_t::object) return false;
     auto& j = state["tracker"]; // state's tracker data
     if (j["format_version"] != 1) return false; // incompatible state format
@@ -885,6 +914,17 @@ bool Tracker::loadState(nlohmann::json& state)
             }
         }
     }
+    auto& jPins = j["pins"];
+    if (jPins.type() == json::value_t::array) {
+        try {
+            _pins = jPins.get<std::list<std::string>>();
+        } catch (std::exception& ex) {
+            fprintf(stderr, "Could not load pins: %s\n", ex.what());
+        }
+    }
+    for (const auto& pin: _pins)
+        onPinChanged.emit(this, pin, true);
+
     for (const auto& id: _bulkItemUpdates)
         onStateChanged.emit(this, id);
     _bulkItemUpdates.clear();

diff --git a/src/core/tracker.h b/src/core/tracker.h
@@ -62,7 +62,8 @@ class Tracker final : public LuaInterface<Tracker> {
     Signal<const std::string&> onStateChanged;
     Signal<const std::string&> onDisplayChanged; // changed display of an item
     Signal<const std::string&, const std::string&> onUiHint;
-
+    Signal<const std::string&, bool> onPinChanged;
+
     const LayoutNode& getLayout(const std::string& name) const;
     bool hasLayout(const std::string& name) const;
     const BaseItem& getItemByCode(const std::string& code) const;
@@ -83,6 +84,10 @@ class Tracker final : public LuaInterface<Tracker> {
     bool isVisible(const Location& location);
     bool isVisible(const Location::MapLocation& mapLoc);
 
+    bool isPinned(const std::string& id) const;
+    void setPinned(const std::string& id, bool pinned);
+    const std::list<std::string>& getPins() const;
+
     const Pack* getPack() const;
 
     bool changeItemState(const std::string& id, BaseItem::Action action);
@@ -102,6 +107,7 @@ class Tracker final : public LuaInterface<Tracker> {
     std::map<std::string, int> _providerCountCache;
     std::list<std::string> _bulkItemUpdates;
     bool _bulkUpdate = false;
+    std::list<std::string> _pins;
 
     std::list<std::string>* _parents = nullptr;