Update tree.nodeAttrKeys during ADD_EXTRA_METADATA action

Ensures that new metadata fields added to the tree are included as
new tip labels.

Fixes #1844

Author: joverlee521

src/reducers/tree.js

@@ -1,5 +1,6 @@
 import { countTraitsAcrossTree } from "../util/treeCountingHelpers";
 import { addNodeAttrs } from "../util/treeMiscHelpers";
+import { processNodes } from "../util/treeJsonProcessing";
 import * as types from "../actions/types";
 
 /* A version increase (i.e. props.version !== nextProps.version) necessarily implies
@@ -68,13 +69,16 @@ const Tree = (state = getDefaultTreeState(), action) => {
     case types.ADD_EXTRA_METADATA:
       // add data into `nodes` in-place, so no redux update will be triggered if you only listen to `nodes`
       addNodeAttrs(state.nodes, action.newNodeAttrs);
+      // add the new nodeAttrKeys to ensure tip labels get updated
+      const { nodeAttrKeys } = processNodes(state.nodes)
       // add the new colorings to totalStateCounts so that they can function as filters
       return {
         ...state,
         totalStateCounts: {
           ...state.totalStateCounts,
           ...countTraitsAcrossTree(state.nodes, Object.keys(action.newColorings), false, true)
-        }
+        },
+        nodeAttrKeys
       };
     default:
       return state;

src/util/treeJsonProcessing.js

@@ -16,7 +16,7 @@ const pseudoRandomName = () => (Math.random()*1e32).toString(36).slice(0, 6);
  *
  * side-effects: node.hasChildren (bool) and node.arrayIdx (INT) for each node in nodes
  */
-const processNodes = (nodes) => {
+export const processNodes = (nodes) => {
   const nodeNamesSeen = new Set();
   const nodeAttrKeys = new Set();
   calcFullTipCounts(nodes[0]); /* recursive. Uses d.children */