From 56e98aece5fe4122ff2e0db46ae08f997628f056 Mon Sep 17 00:00:00 2001
From: Nathan Sobo <nathan@github.com>
Date: Tue, 10 Apr 2012 19:30:44 -0600
Subject: [PATCH] Add window state getter and setter to RootView

These methods will be used to save and restore split views and editors on a refresh.
---
 spec/app/root-view-spec.coffee | 38 ++++++++++++++++++++++++++++++++++
 src/app/root-view.coffee       | 32 ++++++++++++++++++++++++++++
 2 files changed, 70 insertions(+)

diff --git a/spec/app/root-view-spec.coffee b/spec/app/root-view-spec.coffee
index be3ba4fa8..664fc2699 100644
--- a/spec/app/root-view-spec.coffee
+++ b/spec/app/root-view-spec.coffee
@@ -49,6 +49,44 @@ describe "RootView", ->
       expect(rootView).not.toMatchSelector(':focus')
       expect(rootView.activeEditor().isFocused).toBeTruthy()
 
+  describe "windowState getter and setter", ->
+    [editor1, editor2, editor3, editor4, panesHtml] = []
+
+    beforeEach ->
+      editor1 = rootView.activeEditor()
+      editor2 = editor1.splitRight()
+      editor3 = editor2.splitRight()
+      editor4 = editor2.splitDown()
+      editor2.setBuffer(new Buffer(require.resolve 'fixtures/dir/b'))
+      editor3.setBuffer(new Buffer(require.resolve 'fixtures/sample.js'))
+      editor3.setCursorScreenPosition([2, 3])
+      editor4.setBuffer(new Buffer(require.resolve 'fixtures/sample.txt'))
+      editor4.setCursorScreenPosition([0, 2])
+      panesHtml = rootView.panes.html()
+
+    it "can reconstruct the split pane arrangement from the window state hash returned by getWindowState", ->
+      windowState = rootView.getWindowState()
+
+      editor2.remove()
+      editor3.remove()
+      editor4.remove()
+      expect(rootView.panes.find('.editor').length).toBe 1
+
+      rootView.setWindowState(windowState)
+      expect(rootView.editors.length).toBe 4
+
+      editor1 = rootView.panes.find('.row > .editor:eq(0)').view()
+      editor3 = rootView.panes.find('.row > .editor:eq(1)').view()
+      editor2 = rootView.panes.find('.row > .column > .editor:eq(0)').view()
+      editor4 = rootView.panes.find('.row > .column > .editor:eq(1)').view()
+
+      expect(editor1.buffer.path).toBe require.resolve('fixtures/dir/a')
+      expect(editor2.buffer.path).toBe require.resolve('fixtures/dir/b')
+      expect(editor3.buffer.path).toBe require.resolve('fixtures/sample.js')
+      expect(editor3.getCursorScreenPosition()).toEqual [2, 3]
+      expect(editor4.buffer.path).toBe require.resolve('fixtures/sample.txt')
+      expect(editor4.getCursorScreenPosition()).toEqual [0, 2]
+
   describe "split editor panes", ->
     editor1 = null
 
diff --git a/src/app/root-view.coffee b/src/app/root-view.coffee
index 63e2e9d1a..3c0204b89 100644
--- a/src/app/root-view.coffee
+++ b/src/app/root-view.coffee
@@ -1,4 +1,5 @@
 $ = require 'jquery'
+{$$} = require 'space-pen'
 fs = require 'fs'
 _ = require 'underscore'
 
@@ -78,6 +79,37 @@ class RootView extends View
       editor.appendTo(@panes)
       editor.focus()
 
+  getWindowState: (element = @panes.children(':eq(0)')) ->
+    if element.hasClass('editor')
+      ['editor', element.view().getEditorState()]
+    else if element.hasClass('row')
+      ['row'].concat element.children().toArray().map (elt) =>
+        @getWindowState($(elt))
+    else if element.hasClass('column')
+      ['column'].concat element.children().toArray().map (elt) =>
+        @getWindowState($(elt))
+
+  setWindowState: (windowState, parent) ->
+    unless parent
+      @panes.empty()
+      @editors = []
+      parent = @panes
+
+    switch windowState.shift()
+      when 'editor'
+        editor = new Editor(windowState[0])
+        @editors.push(editor)
+        parent.append(editor)
+      when 'row'
+        row = $$ -> @div class: 'row'
+        parent.append row
+        for child in windowState
+          @setWindowState(child, row)
+      when 'column'
+        column = $$ -> @div class: 'column'
+        parent.append column
+        for child in windowState
+          @setWindowState(child, column)
 
   addPane: (view, sibling, axis, side) ->
     unless sibling.parent().hasClass(axis)