1 files changed, 19 insertions, 13 deletions

diff --git a/src/ws/node.js b/src/ws/node.js
index 424aa99..fd4a9d7 100644
--- a/src/ws/node.js
+++ b/src/ws/node.js
@@ -218,19 +218,8 @@ Node.prototype.create = function()
 	// oberveid from the node id alone.
 	node.id = createId(this.subscribeid, this.id);
 
-	{
-		var collapsed = localStorage.getItem(node.id+"_collapsed") == "true";
-		if(collapsed)
-		{
-			this.element.classList.add('collapsed');
-			this.collapse_button.innerHTML = "▼";
-		}
-		else
-		{
-			this.element.classList.remove('collapsed');
-			this.collapse_button.innerHTML = "▲";
-		}
-	}
+	var collapsed = localStorage.getItem(node.id+"_collapsed") == "true";
+	this.setCollapsed(collapsed);
 };
 
 Node.prototype.updateAnnotations = function()
@@ -250,6 +239,23 @@ Node.prototype.updateAnnotations = function()
 	this.children_indicator_element.innerHTML = ann;
 };
 
+Node.prototype.setCollapsed = function(collapsed)
+{
+	if(collapsed)
+	{
+		this.children_indicator_element.classList.add('collapsed');
+		this.element.classList.add('collapsed');
+		this.collapse_button.innerHTML = "▼";
+	}
+	else
+	{
+		this.children_indicator_element.classList.remove('collapsed');
+		this.element.classList.remove('collapsed');
+		this.collapse_button.innerHTML = "▲";
+	}
+};
+
+
 //! Recursively set draggable property from node until root node.
 Node.prototype.setDraggable = function(value)
 {