Back DisplayBuffer with a telepath document

src/app/display-buffer.coffee

@@ -1,6 +1,7 @@
 _ = require 'underscore'
 guid = require 'guid'
-{Point, Range} = require 'telepath'
+telepath = require 'telepath'
+{Point, Range} = telepath
 TokenizedBuffer = require 'tokenized-buffer'
 RowMap = require 'row-map'
 EventEmitter = require 'event-emitter'
@@ -11,30 +12,47 @@ Subscriber = require 'subscriber'
 
 module.exports =
 class DisplayBuffer
-  @idCounter: 1
   screenLines: null
   rowMap: null
   tokenizedBuffer: null
   markers: null
   foldsByMarkerId: null
 
-  ### Internal ###
+  @acceptsDocuments: true
+  registerDeserializer(this)
+
+  @deserialize: (state) ->
+    new DisplayBuffer(state)
+
+  constructor: (optionsOrState) ->
+    if optionsOrState instanceof telepath.Document
+      @state = optionsOrState
+      @id = @state.get('id')
+      @tokenizedBuffer = deserialize(@state.get('tokenizedBuffer'))
+      @buffer = @tokenizedBuffer.buffer
+    else
+      {@buffer, softWrapColumn} = optionsOrState
+      @id = guid.create().toString()
+      @tokenizedBuffer = new TokenizedBuffer(optionsOrState)
+      @state = telepath.create
+        deserializer: @constructor.name
+        id: @id
+        tokenizedBuffer: @tokenizedBuffer.getState()
+        softWrapColumn: softWrapColumn ? Infinity
 
-  constructor: (@buffer, options={}) ->
-    @id = guid.create().toString()
-    @tokenizedBuffer = new TokenizedBuffer(_.defaults({@buffer}, options))
-    @softWrapColumn = options.softWrapColumn ? Infinity
     @markers = {}
     @foldsByMarkerId = {}
     @updateAllScreenLines()
-
     @tokenizedBuffer.on 'grammar-changed', (grammar) => @trigger 'grammar-changed', grammar
     @tokenizedBuffer.on 'changed', @handleTokenizedBufferChange
     @subscribe @buffer, 'markers-updated', @handleMarkersUpdated
     @subscribe @buffer, 'marker-created', @handleMarkerCreated
 
+  serialize: -> @state.clone()
+  getState: -> @state
+
   copy: ->
-    newDisplayBuffer = new DisplayBuffer(@buffer, tabLength: @getTabLength())
+    newDisplayBuffer = new DisplayBuffer({@buffer, tabLength: @getTabLength()})
     for marker in @findMarkers(displayBufferId: @id)
       marker.copy(displayBufferId: newDisplayBuffer.id)
     newDisplayBuffer
@@ -62,7 +80,8 @@ class DisplayBuffer
   # Defines the limit at which the buffer begins to soft wrap text.
   #
   # softWrapColumn - A {Number} defining the soft wrap limit.
-  setSoftWrapColumn: (@softWrapColumn) ->
+  setSoftWrapColumn: (softWrapColumn) ->
+    @state.set('softWrapColumn', softWrapColumn)
     start = 0
     end = @getLastRow()
     @updateAllScreenLines()
@@ -70,6 +89,9 @@ class DisplayBuffer
     bufferDelta = 0
     @triggerChanged({ start, end, screenDelta, bufferDelta })
 
+  getSoftWrapColumn: ->
+    @state.get('softWrapColumn')
+
   # Gets the screen line for the given screen row.
   #
   # screenRow - A {Number} indicating the screen row.
@@ -379,7 +401,7 @@ class DisplayBuffer
   #
   # Returns a {Number} representing the `line` position where the wrap would take place.
   # Returns `null` if a wrap wouldn't occur.
-  findWrapColumn: (line, softWrapColumn=@softWrapColumn) ->
+  findWrapColumn: (line, softWrapColumn=@getSoftWrapColumn()) ->
     return unless line.length > softWrapColumn
 
     if /\s/.test(line[softWrapColumn])

src/app/edit-session.coffee

@@ -45,7 +45,7 @@ class EditSession
       @state = optionsOrState
       {@id, tabLength, softTabs, @softWrap} = @state.toObject()
       @setBuffer(project.bufferForId(@state.get('bufferId')))
-      @setDisplayBuffer(new DisplayBuffer(@buffer, { tabLength }))
+      @setDisplayBuffer(new DisplayBuffer({@buffer, tabLength}))
       @addSelection(marker) for marker in @findMarkers(@getSelectionMarkerAttributes())
       @setScrollTop(@state.get('scrollTop'))
       @setScrollLeft(@state.get('scrollLeft'))
@@ -59,7 +59,7 @@ class EditSession
         scrollTop: 0
         scrollLeft: 0
       @setBuffer(buffer)
-      @setDisplayBuffer(displayBuffer ? new DisplayBuffer(@buffer, { tabLength }))
+      @setDisplayBuffer(displayBuffer ? new DisplayBuffer({@buffer, tabLength}))
       @addCursorAtScreenPosition([0, 0]) unless suppressCursorCreation
 
     @languageMode = new LanguageMode(this, @buffer.getExtension())