1 files changed, 45 insertions, 45 deletions
diff --git a/examples/demos/src/main/scala/scrollmenu/Controller.scala b/examples/demos/src/main/scala/scrollmenu/Controller.scala
index 4480445..fe35be8 100644
--- a/examples/demos/src/main/scala/scrollmenu/Controller.scala
+++ b/examples/demos/src/main/scala/scrollmenu/Controller.scala
@@ -28,53 +28,53 @@ object Controller{
     val snippets = dom.document.getElementsByClassName("highlight-me")
 
     snippets.foreach(js.Dynamic.global.hljs.highlightBlock(_))
-    def rest() = {
-      val scrollSpy = new ScrollSpy(structure, main)
-      val list = ul(cls := "menu-item-list collapsed")(
-        scrollSpy.domTrees.map(_.value.frag)
+
+    val scrollSpy = new ScrollSpy(structure, main)
+    val list = ul(cls := "menu-item-list collapsed")(
+      scrollSpy.domTrees.value.frag
+    ).render
+
+    def updateScroll() = scrollSpy()
+    val expandIcon = i(cls := "fa fa-caret-down").render
+    val expandLink =
+      a(
+        expandIcon,
+        href := "javascript:",
+        marginLeft := "0px",
+        paddingLeft := "15px",
+        paddingRight := "15px",
+        position.absolute,
+        top := "0px",
+        right := "0px",
+        cls := "pure-menu-selected",
+        onclick := { (e: dom.Event) =>
+          expandIcon.classList.toggle("fa-caret-down")
+          expandIcon.classList.toggle("fa-caret-up")
+          list.classList.toggle("collapsed")
+          list.classList.toggle("expanded")
+          scrollSpy.toggleOpen()
+//          updateScroll()
+        }
+      ).render
+
+
+    menu.appendChild(
+      div(cls := "pure-menu  pure-menu-open")(
+        a(cls := "pure-menu-heading")(
+          "Contents", expandLink
+        ),
+        list
       ).render
+    )
 
-      def updateScroll() = scrollSpy()
-      val expandIcon = i(cls := "fa fa-caret-down").render
-      val expandLink =
-        a(
-          expandIcon,
-          href := "javascript:",
-          marginLeft := "0px",
-          paddingLeft := "15px",
-          paddingRight := "15px",
-          position.absolute,
-          top := "0px",
-          right := "0px",
-          cls := "pure-menu-selected",
-          onclick := { (e: dom.Event) =>
-            expandIcon.classList.toggle("fa-caret-down")
-            expandIcon.classList.toggle("fa-caret-up")
-            list.classList.toggle("collapsed")
-            scrollSpy.clean = !scrollSpy.clean
-            updateScroll()
-          }
-        ).render
-
-
-      menu.appendChild(
-        div(cls := "pure-menu  pure-menu-open")(
-          a(cls := "pure-menu-heading")(
-            "Contents", expandLink
-          ),
-          list
-        ).render
-      )
-
-      menuLink.onclick = (e: dom.MouseEvent) => {
-        layout.classList.toggle("active")
-        menu.classList.toggle("active")
-        menuLink.classList.toggle("active")
-      }
-
-      main.onscroll = (e: dom.UIEvent) => updateScroll()
-      updateScroll()
+    menuLink.onclick = (e: dom.MouseEvent) => {
+      layout.classList.toggle("active")
+      menu.classList.toggle("active")
+      menuLink.classList.toggle("active")
     }
-    dom.setTimeout(rest _, 10)
+
+    main.onscroll = (e: dom.UIEvent) => updateScroll()
+    updateScroll()
+
   }
 }