restore collapse status

Author: sorax

assets/js/hooks2/index.ts

@@ -4,6 +4,7 @@ import {
 } from "./tree";
 
 import { toggleCollapse } from "./node";
+import { restoreCollapsedStatus } from "./store";
 
 import { DomContainer } from "./types";
 
@@ -15,6 +16,7 @@ export const Hooks = {
       container.addEventListener("toggle_collapse", toggleCollapse);
 
       moveHtmlChildNodesToDataPosition(container);
+      restoreCollapsedStatus(container);
       setCursorToEndOfFirstChildNode(container);
     },
     //updated() {},

assets/js/hooks2/node/index.ts

@@ -35,7 +35,7 @@ function getDomNodeById(uuid: UUID | undefined) {
   return document.getElementById(`nodes-form-${uuid}`) as DomNode | null;
 }
 
-function getDataNodeFromDomNode(node: DomNode): DataNode {
+export function getDataNodeFromDomNode(node: DomNode): DataNode {
   const uuid = getUUID(node);
   const parent_id = getAttribute(node, "parent");
   const prev_id = getAttribute(node, "prev");

assets/js/hooks2/store.ts

@@ -1,4 +1,14 @@
-import { UUID } from "./types";
+import { getDataNodeFromDomNode } from "./node";
+import { DomContainer, DomNode, UUID } from "./types";
+
+export function restoreCollapsedStatus(container: DomContainer) {
+  const status = loadCollapsedStatus(container.id);
+
+  container.querySelectorAll(".node").forEach((node) => {
+    const { uuid } = getDataNodeFromDomNode(node as DomNode);
+    status[uuid] && node.classList.add("collapsed");
+  });
+}
 
 export function saveCollapseStatus(id: string, uuid: UUID, collapsed: boolean) {
   const status = loadCollapsedStatus(id);

assets/js/hooks2/tree/index.ts

@@ -8,4 +8,3 @@ export function moveHtmlChildNodesToDataPosition(container: DomContainer) {
     moveDomNodeToDataPosition(node as DomNode);
   });
 }
-