1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_BOOKMARKS_BOOKMARK_FOLDER_TREE_MODEL_H_
#define CHROME_BROWSER_BOOKMARKS_BOOKMARK_FOLDER_TREE_MODEL_H_
#include "chrome/browser/bookmarks/bookmark_model.h"
#include "chrome/views/tree_node_model.h"
// The type of nodes created by BookmarkFolderTreeModel.
typedef views::TreeNodeWithValue<BookmarkNode*> FolderNode;
// TreeModel implementation that shows the folders from the BookmarkModel.
// The root node contains the following nodes:
// bookmark bar, other folders, recently bookmarked and search.
class BookmarkFolderTreeModel : public views::TreeNodeModel<FolderNode>,
public BookmarkModelObserver {
public:
// Type of the node.
enum NodeType {
// Represents an entry from the BookmarkModel.
BOOKMARK,
RECENTLY_BOOKMARKED,
SEARCH,
NONE // Used for no selection.
};
explicit BookmarkFolderTreeModel(BookmarkModel* model);
~BookmarkFolderTreeModel();
// The tree is not editable.
virtual void SetTitle(views::TreeModelNode* node, const std::wstring& title) {
NOTREACHED();
}
// Returns the type of the specified node.
NodeType GetNodeType(views::TreeModelNode* node);
// Returns the FolderNode for the specified BookmarkNode.
FolderNode* GetFolderNodeForBookmarkNode(BookmarkNode* node);
// Converts the tree node into a BookmarkNode. Returns NULL if |node| is NULL
// or not of NodeType::BOOKMARK.
BookmarkNode* TreeNodeAsBookmarkNode(views::TreeModelNode* node);
// Returns the search node.
FolderNode* search_node() const { return search_node_; }
// BookmarkModelObserver implementation.
virtual void Loaded(BookmarkModel* model);
virtual void BookmarkModelBeingDeleted(BookmarkModel* model);
virtual void BookmarkNodeMoved(BookmarkModel* model,
BookmarkNode* old_parent,
int old_index,
BookmarkNode* new_parent,
int new_index);
virtual void BookmarkNodeAdded(BookmarkModel* model,
BookmarkNode* parent,
int index);
virtual void BookmarkNodeRemoved(BookmarkModel* model,
BookmarkNode* parent,
int index,
BookmarkNode* node);
virtual void BookmarkNodeChanged(BookmarkModel* model,
BookmarkNode* node);
// Folders don't have favicons, so we ignore this.
virtual void BookmarkNodeFavIconLoaded(BookmarkModel* model,
BookmarkNode* node) {}
// The following are overriden to return custom icons for the recently
// bookmarked and search nodes.
virtual void GetIcons(std::vector<SkBitmap>* icons);
virtual int GetIconIndex(views::TreeModelNode* node);
private:
// Invoked from the constructor to create the children of the root node.
void AddRootChildren();
// Implementation of GetFolderNodeForBookmarkNode. If the |folder_node|
// represents |node|, |folder_node| is returned, otherwise this recurses
// through the children.
FolderNode* GetFolderNodeForBookmarkNode(FolderNode* folder_node,
BookmarkNode* node);
// Creates a new folder node for |node| and all its children.
FolderNode* CreateFolderNode(BookmarkNode* node);
// Returns the number of folders that precede |node| in |node|s parent.
// The returned value is the index of the folder node representing |node|
// in its parent.
// This is used when new bookmarks are created to determine where the
// corresponding folder node should be created.
int CalculateIndexForChild(BookmarkNode* node);
// The model we're getting data from. Owned by the Profile.
BookmarkModel* model_;
// The two special nodes. These are owned by the root tree node owned by
// TreeNodeModel.
FolderNode* recently_bookmarked_node_;
FolderNode* search_node_;
DISALLOW_COPY_AND_ASSIGN(BookmarkFolderTreeModel);
};
#endif // CHROME_BROWSER_BOOKMARKS_BOOKMARK_FOLDER_TREE_MODEL_H_
|