add a createWriteStream method and expose the stream constructors on the FQ prototype

Author: treygriffith

lib/methods.js

@@ -1,7 +1,7 @@
 // All of the fs methods we want exposed on fq
 // This should pass the Grep Test (http://jamie-wong.com/2013/07/12/grep-test/)
- 
-var FQReadStream = require('./stream');
+
+var stream = require('./stream');
 
 module.exports = function (FQ) {
 
@@ -165,9 +165,11 @@ module.exports = function (FQ) {
 		this.execQueue();
 	};
 
+	FQ.prototype.ReadStream = stream.ReadStream;
+
 	FQ.prototype.createReadStream = function(path, options) {
 
-		var readStream = new FQReadStream(path, options);
+		var readStream = new this.ReadStream(path, options);
 
 		this.addToQueue(function(fs, cb) {
 
@@ -181,4 +183,22 @@ module.exports = function (FQ) {
 		return readStream;
 	};
 
+	FQ.prototype.WriteStream = stream.WriteStream;
+
+	FQ.prototype.createWriteStream = function(path, options) {
+
+		var writeStream = new this.WriteStream(path, options);
+
+		this.addToQueue(function(fs, cb) {
+
+			writeStream.on('close', cb);
+
+			writeStream.open();
+		});
+
+		this.execQueue();
+
+		return writeStream;
+	};
+
 };

lib/stream.js

@@ -1,7 +1,10 @@
 var fs = require('fs');
 var util = require('util');
 
+var fq = exports;
+
 util.inherits(FQReadStream, fs.ReadStream);
+fq.ReadStream = FQReadStream;
 
 function FQReadStream(path, options) {
 
@@ -16,4 +19,18 @@ function FQReadStream(path, options) {
 	return this;
 }
 
-module.exports = FQReadStream;
+util.inherits(FQWriteStream, fs.WriteStream);
+fq.WriteStream = FQWriteStream;
+
+function FQWriteStream(path, options) {
+
+	// temporarily change our #open method so the ReadStream initializer doesn't open the file
+	var open = this.open;
+	this.open = function() {};
+
+	fs.WriteStream.call(this, path, options);
+
+	this.open = open;
+
+	return this;
+}