Update setPath to work with custom files

Contributed under CC0.

Author: hansonw

spec/text-buffer-io-spec.js

@@ -19,6 +19,10 @@ class SimpleFile {
     return this.path
   }
 
+  setPath (path) {
+    this.path = path
+  }
+
   createReadStream () {
     return fs.createReadStream(this.path)
   }
@@ -149,9 +153,10 @@ describe('TextBuffer IO', () => {
 
   describe('.save', () => {
     let filePath
+    let tempDir
 
     beforeEach(() => {
-      const tempDir = temp.mkdirSync()
+      tempDir = temp.mkdirSync()
       filePath = path.join(tempDir, 'temp.txt')
       fs.writeFileSync(filePath, '')
       buffer = new TextBuffer()
@@ -257,6 +262,17 @@ describe('TextBuffer IO', () => {
           done()
         })
       })
+
+      it('passes setPath to the custom File object', (done) => {
+        const newPath = path.join(tempDir, 'temp2.txt')
+        buffer.setPath(newPath)
+        buffer.setText('test')
+        buffer.save().then(() => {
+          expect(fs.readFileSync(newPath, 'utf8')).toEqual(buffer.getText())
+          expect(file.createWriteStream).toHaveBeenCalled()
+          done()
+        })
+      })
     })
   })
 

src/text-buffer.coffee

@@ -585,6 +585,10 @@ class TextBuffer
   # * `filePath` A {String} representing the new file path
   setPath: (filePath) ->
     return if filePath is @getPath()
+    if @file? and filePath
+      @file.setPath filePath
+      @updateFilePath()
+      return
     @setFile(new File(filePath) if filePath)
 
   # Experimental: Set a custom {File} object as the buffer's backing store.
@@ -607,8 +611,11 @@ class TextBuffer
   setFile: (file) ->
     return if file?.getPath() is @getPath()
     @file = file
+    @file?.setEncoding?(@getEncoding())
+    @updateFilePath()
+
+  updateFilePath: ->
     if @file?
-      @file.setEncoding?(@getEncoding())
       @subscribeToFile()
     @emitter.emit 'did-change-path', @getPath()