initial commit

node_modules/stream-length/lib/stream-length.coffee

@@ -0,0 +1,82 @@
+Promise = require "bluebird"
+fs = Promise.promisifyAll(require "fs")
+
+nodeifyWrapper = (callback, func) ->
+	func().nodeify(callback)
+
+createRetrieverPromise = (stream, retriever) ->
+	new Promise (resolve, reject) ->
+		retriever stream, (result) ->
+			if result?
+				if result instanceof Error
+					reject result
+				else
+					resolve result
+			else
+				reject new Error("Could not find a length using this lengthRetriever.")
+
+retrieveBuffer = (stream, callback) ->
+	if stream instanceof Buffer
+		callback stream.length
+	else
+		callback null
+
+retrieveFilesystemStream = (stream, callback) ->
+	if stream.hasOwnProperty "fd"
+		# FIXME: https://github.com/joyent/node/issues/7819
+		if stream.end != undefined and stream.end != Infinity and stream.start != undefined
+			# A stream start and end were defined, we can calculate the size just from that information.
+			callback(stream.end + 1 - (stream.start ? 0))
+		else
+			# We have the start offset at most, stat the file and work off the filesize.
+			Promise.try ->
+				fs.statAsync stream.path
+			.then (stat) ->
+				callback(stat.size - (stream.start ? 0))
+			.catch (err) ->
+				callback err
+	else
+		callback null
+
+retrieveCoreHttpStream = (stream, callback) ->
+	if stream.hasOwnProperty("httpVersion") and stream.headers["content-length"]?
+		callback parseInt(stream.headers["content-length"])
+	else
+		callback null
+
+retrieveRequestHttpStream = (stream, callback) ->
+	if stream.hasOwnProperty "httpModule"
+		stream.on "response", (response) ->
+			if response.headers["content-length"]?
+				callback parseInt(response.headers["content-length"])
+			else
+				callback null
+	else
+		callback null
+
+retrieveCombinedStream = (stream, callback) ->
+	if stream.getCombinedStreamLength?
+		stream.getCombinedStreamLength()
+			.then (length) -> callback(length)
+			.catch (err) -> callback(err)
+	else
+		callback null
+
+
+module.exports = (stream, options = {}, callback) ->
+	nodeifyWrapper callback, ->
+		retrieverPromises = []
+
+		if options.lengthRetrievers?
+			# First, the custom length retrievers, if any.
+			for retriever in options.lengthRetrievers
+				retrieverPromises.push createRetrieverPromise(stream, retriever)
+
+		# Then, the standard ones.
+		for retriever in [retrieveBuffer, retrieveFilesystemStream, retrieveCoreHttpStream, retrieveRequestHttpStream, retrieveCombinedStream]
+			retrieverPromises.push createRetrieverPromise(stream, retriever)
+
+		Promise.any retrieverPromises
+
+
+