MD Bookmarks: Remove deleted nodes from state tree

chrome/browser/resources/md_bookmarks/util.js

Index: chrome/browser/resources/md_bookmarks/util.js
diff --git a/chrome/browser/resources/md_bookmarks/util.js b/chrome/browser/resources/md_bookmarks/util.js
index bf20a151c1246529fcb4ca5c327e84df876066ce..2678e798f81730592fbbec5ead2badfd4542896a 100644
--- a/chrome/browser/resources/md_bookmarks/util.js
+++ b/chrome/browser/resources/md_bookmarks/util.js
@@ -67,9 +67,56 @@ cr.define('bookmarks.util', function() {
     };
   }
 
+  /**
+   * Get all descendants of a node, including the node itself.
+   * @param {NodeList} nodes
+   * @param {string} baseId
+   * @return {!Set<string>}
+   */
+  function getDescendants(nodes, baseId) {
+    var descendants = new Set();
+    var stack = [];
+    stack.push(baseId);
+
+    while (stack.length > 0) {
+      var id = stack.pop();
+      var node = nodes[id];
+
+      if (!node)
+        continue;
+
+      descendants.add(id);
+
+      if (!node.children)
+        continue;
+
+      node.children.forEach(function(childId) {
+        stack.push(childId);
+      });
+    }
+
+    return descendants;
+  }
+
+  /**
+   * @param {!Object<string, T>} map
+   * @param {!Set<string>} nodes
+   * @return {!Object<string, T>}
+   * @template T
+   */
+  function removeNodesFromMap(map, nodes) {

[calamity, 2017/04/03 06:54:24]

Maybe genericize further into removeKeysFromMap? At the moment this is
overloading Nodes in a confusing way. Another alternative is removeIdsFromMap.

[tsergeant, 2017/04/04 04:42:28]

Done, with removeIdsFromMap (and removeIdsFromSet)

+    var newMap = Object.assign({}, map);
+    nodes.forEach(function(id) {
+      delete newMap[id];
+    });
+    return newMap;
+  };
+
   return {
     createEmptyState: createEmptyState,
+    getDescendants: getDescendants,
     getDisplayedList: getDisplayedList,
     normalizeNodes: normalizeNodes,
+    removeNodesFromMap: removeNodesFromMap,
   };
 });