bookmark_expanded_state_tracker.h 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. // Copyright 2014 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef COMPONENTS_BOOKMARKS_BROWSER_BOOKMARK_EXPANDED_STATE_TRACKER_H_
  5. #define COMPONENTS_BOOKMARKS_BROWSER_BOOKMARK_EXPANDED_STATE_TRACKER_H_
  6. #include <set>
  7. #include "base/memory/raw_ptr.h"
  8. #include "components/bookmarks/browser/base_bookmark_model_observer.h"
  9. class PrefService;
  10. namespace bookmarks {
  11. class BookmarkModel;
  12. class BookmarkNode;
  13. // BookmarkExpandedStateTracker is used to track a set of expanded nodes. The
  14. // nodes are persisted in preferences. If an expanded node is removed from the
  15. // model BookmarkExpandedStateTracker removes the node.
  16. class BookmarkExpandedStateTracker : public BaseBookmarkModelObserver {
  17. public:
  18. typedef std::set<const BookmarkNode*> Nodes;
  19. BookmarkExpandedStateTracker(BookmarkModel* bookmark_model,
  20. PrefService* pref_service);
  21. BookmarkExpandedStateTracker(const BookmarkExpandedStateTracker&) = delete;
  22. BookmarkExpandedStateTracker& operator=(const BookmarkExpandedStateTracker&) =
  23. delete;
  24. ~BookmarkExpandedStateTracker() override;
  25. // The set of expanded nodes.
  26. void SetExpandedNodes(const Nodes& nodes);
  27. Nodes GetExpandedNodes();
  28. private:
  29. // BaseBookmarkModelObserver:
  30. void BookmarkModelLoaded(BookmarkModel* model, bool ids_reassigned) override;
  31. void BookmarkModelChanged() override;
  32. void BookmarkModelBeingDeleted(BookmarkModel* model) override;
  33. void BookmarkNodeRemoved(BookmarkModel* model,
  34. const BookmarkNode* parent,
  35. size_t old_index,
  36. const BookmarkNode* node,
  37. const std::set<GURL>& removed_urls) override;
  38. void BookmarkAllUserNodesRemoved(BookmarkModel* model,
  39. const std::set<GURL>& removed_urls) override;
  40. // Updates the value for |prefs::kBookmarkEditorExpandedNodes| from
  41. // GetExpandedNodes().
  42. void UpdatePrefs(const Nodes& nodes);
  43. raw_ptr<BookmarkModel> bookmark_model_;
  44. raw_ptr<PrefService> pref_service_;
  45. };
  46. } // namespace bookmarks
  47. #endif // COMPONENTS_BOOKMARKS_BROWSER_BOOKMARK_EXPANDED_STATE_TRACKER_H_