From 21773ab52dced6e42a151111e9e1e5382e173409 Mon Sep 17 00:00:00 2001
From: hppeng <hppeng>
Date: Sun, 26 Jun 2022 05:18:23 -0700
Subject: [PATCH] Fix set initial load

---
 js/display_atree.js | 6 ++++++
 js/load.js          | 7 ++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/js/display_atree.js b/js/display_atree.js
index 888383e..205a228 100644
--- a/js/display_atree.js
+++ b/js/display_atree.js
@@ -1,4 +1,5 @@
 let atree_map;
+let atree_head;
 let atree_connectors_map;
 function construct_AT(elem, tree) {
     console.log("constructing ability tree UI");
@@ -25,6 +26,11 @@ function construct_AT(elem, tree) {
         // create rows if not exist
         let missing_rows = [node.display.row];
 
+        if (node.parents.length == 0) {
+            // Assuming there is only one head.
+            atree_head = node;
+        }
+
         for (let parent of node.parents) {
             missing_rows.push(tree.find(object => {return object.display_name === parent;}).display.row);
         }
diff --git a/js/load.js b/js/load.js
index 521765e..678e012 100644
--- a/js/load.js
+++ b/js/load.js
@@ -103,7 +103,7 @@ async function load() {
     let url = baseUrl + "/compress.json?"+new Date();
     let result = await (await fetch(url)).json();
     items = result.items;
-    sets = result.sets;
+    let sets_ = result.sets;
     
     let add_tx = db.transaction(['item_db', 'set_db'], 'readwrite');
     add_tx.onabort = function(e) {
@@ -121,8 +121,9 @@ async function load() {
         add_promises.push(req);
     }
     let sets_store = add_tx.objectStore('set_db');
-    for (const set in sets) {
-        add_promises.push(sets_store.add(sets[set], set));
+    for (const set in sets_) {
+        add_promises.push(sets_store.add(sets_[set], set));
+        sets.set(set, sets_[set]);
     }
     add_promises.push(add_tx.complete);