| module.exports = FileWriter |
| |
| var fs = require('graceful-fs') |
| var Writer = require('./writer.js') |
| var inherits = require('inherits') |
| var EOF = {} |
| |
| inherits(FileWriter, Writer) |
| |
| function FileWriter (props) { |
| var self = this |
| if (!(self instanceof FileWriter)) { |
| throw new Error('FileWriter must be called as constructor.') |
| } |
| |
| // should already be established as a File type |
| if (props.type !== 'File' || !props.File) { |
| throw new Error('Non-file type ' + props.type) |
| } |
| |
| self._buffer = [] |
| self._bytesWritten = 0 |
| |
| Writer.call(this, props) |
| } |
| |
| FileWriter.prototype._create = function () { |
| var self = this |
| if (self._stream) return |
| |
| var so = {} |
| if (self.props.flags) so.flags = self.props.flags |
| so.mode = Writer.filemode |
| if (self._old && self._old.blksize) so.bufferSize = self._old.blksize |
| |
| self._stream = fs.createWriteStream(self._path, so) |
| |
| self._stream.on('open', function () { |
| // console.error("FW open", self._buffer, self._path) |
| self.ready = true |
| self._buffer.forEach(function (c) { |
| if (c === EOF) self._stream.end() |
| else self._stream.write(c) |
| }) |
| self.emit('ready') |
| // give this a kick just in case it needs it. |
| self.emit('drain') |
| }) |
| |
| self._stream.on('error', function (er) { self.emit('error', er) }) |
| |
| self._stream.on('drain', function () { self.emit('drain') }) |
| |
| self._stream.on('close', function () { |
| // console.error('\n\nFW Stream Close', self._path, self.size) |
| self._finish() |
| }) |
| } |
| |
| FileWriter.prototype.write = function (c) { |
| var self = this |
| |
| self._bytesWritten += c.length |
| |
| if (!self.ready) { |
| if (!Buffer.isBuffer(c) && typeof c !== 'string') { |
| throw new Error('invalid write data') |
| } |
| self._buffer.push(c) |
| return false |
| } |
| |
| var ret = self._stream.write(c) |
| // console.error('\t-- fw wrote, _stream says', ret, self._stream._queue.length) |
| |
| // allow 2 buffered writes, because otherwise there's just too |
| // much stop and go bs. |
| if (ret === false && self._stream._queue) { |
| return self._stream._queue.length <= 2 |
| } else { |
| return ret |
| } |
| } |
| |
| FileWriter.prototype.end = function (c) { |
| var self = this |
| |
| if (c) self.write(c) |
| |
| if (!self.ready) { |
| self._buffer.push(EOF) |
| return false |
| } |
| |
| return self._stream.end() |
| } |
| |
| FileWriter.prototype._finish = function () { |
| var self = this |
| if (typeof self.size === 'number' && self._bytesWritten !== self.size) { |
| self.error( |
| 'Did not get expected byte count.\n' + |
| 'expect: ' + self.size + '\n' + |
| 'actual: ' + self._bytesWritten) |
| } |
| Writer.prototype._finish.call(self) |
| } |