1// Copyright (c) 2012 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
5#ifndef CHROME_BROWSER_BOOKMARKS_BOOKMARK_EXPANDED_STATE_TRACKER_H_
6#define CHROME_BROWSER_BOOKMARKS_BOOKMARK_EXPANDED_STATE_TRACKER_H_
7
8#include <set>
9
10#include "chrome/browser/bookmarks/base_bookmark_model_observer.h"
11
12class BookmarkModel;
13class BookmarkNode;
14class PrefService;
15
16// BookmarkExpandedStateTracker is used to track a set of expanded nodes. The
17// nodes are persisted in preferences. If an expanded node is removed from the
18// model BookmarkExpandedStateTracker removes the node.
19class BookmarkExpandedStateTracker : public BaseBookmarkModelObserver {
20 public:
21  typedef std::set<const BookmarkNode*> Nodes;
22
23  BookmarkExpandedStateTracker(BookmarkModel* bookmark_model,
24                               PrefService* pref_service);
25  virtual ~BookmarkExpandedStateTracker();
26
27  // The set of expanded nodes.
28  void SetExpandedNodes(const Nodes& nodes);
29  Nodes GetExpandedNodes();
30
31 private:
32  // BaseBookmarkModelObserver:
33  virtual void Loaded(BookmarkModel* model, bool ids_reassigned) OVERRIDE;
34  virtual void BookmarkModelChanged() OVERRIDE;
35  virtual void BookmarkModelBeingDeleted(BookmarkModel* model) OVERRIDE;
36  virtual void BookmarkNodeRemoved(BookmarkModel* model,
37                                   const BookmarkNode* parent,
38                                   int old_index,
39                                   const BookmarkNode* node) OVERRIDE;
40  virtual void BookmarkAllNodesRemoved(BookmarkModel* model) OVERRIDE;
41
42  // Updates the value for |prefs::kBookmarkEditorExpandedNodes| from
43  // GetExpandedNodes().
44  void UpdatePrefs(const Nodes& nodes);
45
46  BookmarkModel* bookmark_model_;
47  PrefService* pref_service_;
48
49  DISALLOW_COPY_AND_ASSIGN(BookmarkExpandedStateTracker);
50};
51
52#endif  // CHROME_BROWSER_BOOKMARKS_BOOKMARK_EXPANDED_STATE_TRACKER_H_
53