Index: chrome/browser/resources/md_bookmarks/store.js |
diff --git a/chrome/browser/resources/md_bookmarks/store.js b/chrome/browser/resources/md_bookmarks/store.js |
index 7a9817817b0991fe590a6f28add51eabb1c1b0b1..0ced082371d31fb9105abb196c8f4a00e91825ab 100644 |
--- a/chrome/browser/resources/md_bookmarks/store.js |
+++ b/chrome/browser/resources/md_bookmarks/store.js |
@@ -55,6 +55,7 @@ Polymer({ |
chrome.bookmarks.getTree(function(results) { |
this.setupStore_(results[0]); |
}.bind(this)); |
+ chrome.bookmarks.onRemoved.addListener(this.onBookmarkRemoved_.bind(this)); |
}, |
/** |
@@ -65,6 +66,7 @@ Polymer({ |
this.rootNode = rootNode; |
this.idToNodeMap_ = {}; |
this.rootNode.path = 'rootNode'; |
+ this.generatePaths_(rootNode, 0); |
this.initNodes_(this.rootNode); |
this.fire('selected-folder-changed', this.rootNode.children[0].id); |
}, |
@@ -115,10 +117,26 @@ Polymer({ |
}, |
/** |
+ * Callback for when a bookmark node is removed. |
+ * If a folder is selected or is an ancestor of a selected folder, the parent |
+ * of the removed folder will be selected. |
+ * @param {string} id The id of the removed bookmark node. |
+ * @param {!Object} removeInfo |
+ */ |
+ onBookmarkRemoved_: function(id, removeInfo) { |
+ if (this.isAncestorOfSelected_(this.idToNodeMap_[id])) |
+ this.fire('selected-folder-changed', removeInfo.parentId); |
+ |
+ var parentNode = this.idToNodeMap_[removeInfo.parentId]; |
+ this.splice(parentNode.path + '.children', removeInfo.index, 1); |
+ this.removeDescendantsFromMap_(id); |
+ this.generatePaths_(parentNode, Number(removeInfo.index)); |
tsergeant
2017/01/04 23:30:36
Is this Number() call only here to satisfy Closure
jiaxi
2017/01/05 05:28:01
Done.
|
+ }, |
+ |
+ /** |
* Initializes the nodes in the bookmarks tree as follows: |
* - Populates |idToNodeMap_| with a mapping of all node ids to their |
* corresponding BookmarkTreeNode. |
- * - Stores the path from the store to a node inside the node. |
* - Sets all the nodes to not selected and open by default. |
* @param {BookmarkTreeNode} bookmarkNode |
* @private |
@@ -131,8 +149,41 @@ Polymer({ |
bookmarkNode.isSelected = false; |
bookmarkNode.isOpen = true; |
for (var i = 0; i < bookmarkNode.children.length; i++) { |
- bookmarkNode.children[i].path = bookmarkNode.path + '.children.' + i; |
+ bookmarkNode.children[i].isSelected = false; |
+ bookmarkNode.isOpen = true; |
this.initNodes_(bookmarkNode.children[i]); |
} |
}, |
+ |
+ /** |
+ * Stores the path from the store to a node inside the node. |
+ * @param {BookmarkTreeNode} bookmarkNode |
+ * @param {number} startIndex |
+ * @private |
+ */ |
+ generatePaths_: function(bookmarkNode, startIndex) { |
+ if (bookmarkNode.children) { |
+ for (var i = startIndex; i < bookmarkNode.children.length; i++) { |
+ bookmarkNode.children[i].path = bookmarkNode.path + '.children.#' + i; |
+ this.generatePaths_(bookmarkNode.children[i], 0); |
+ } |
+ } |
+ }, |
+ |
+ /** |
+ * Remove all descendants of a given node from the map. |
+ * @param {string} id |
+ * @private |
+ */ |
+ removeDescendantsFromMap_: function(id) { |
+ var node = this.idToNodeMap_[id]; |
+ if (!node) |
+ return; |
+ |
+ if (node.children) { |
+ for (var i = 0; i < node.children.length; i++) |
+ this.removeDescendantsFromMap_(node.children[i].id); |
+ } |
+ delete this.idToNodeMap_[id]; |
+ } |
}); |