2014-02-12 12:27:43 -05:00
|
|
|
request = require("request").defaults(jar: false)
|
|
|
|
fs = require("fs")
|
2015-04-29 10:55:58 -04:00
|
|
|
logger = require "logger-sharelatex"
|
2014-02-12 12:27:43 -05:00
|
|
|
|
|
|
|
module.exports = UrlFetcher =
|
|
|
|
pipeUrlToFile: (url, filePath, _callback = (error) ->) ->
|
|
|
|
callbackOnce = (error) ->
|
2015-04-29 10:55:58 -04:00
|
|
|
cleanUp error, (error) ->
|
|
|
|
_callback(error)
|
|
|
|
_callback = () ->
|
|
|
|
|
|
|
|
cleanUp = (error, callback) ->
|
|
|
|
if error?
|
|
|
|
logger.log filePath: filePath, "deleting file from cache due to error"
|
|
|
|
fs.unlink filePath, (err) ->
|
|
|
|
if err?
|
|
|
|
logger.err err: err, filePath: filePath, "error deleting file from cache"
|
|
|
|
callback(error)
|
|
|
|
else
|
|
|
|
callback()
|
2014-02-12 12:27:43 -05:00
|
|
|
|
|
|
|
fileStream = fs.createWriteStream(filePath)
|
2015-04-29 10:55:58 -04:00
|
|
|
fileStream.on 'error', (error) ->
|
|
|
|
logger.error err: error, url:url, filePath: filePath, "error writing file into cache"
|
|
|
|
callbackOnce(error)
|
2014-02-12 12:27:43 -05:00
|
|
|
|
2015-04-29 10:55:58 -04:00
|
|
|
logger.log url:url, filePath: filePath, "downloading url to cache"
|
|
|
|
urlStream = request.get(url)
|
2014-02-12 12:27:43 -05:00
|
|
|
urlStream.on "response", (res) ->
|
|
|
|
if res.statusCode >= 200 and res.statusCode < 300
|
|
|
|
urlStream.pipe(fileStream)
|
|
|
|
else
|
2014-06-10 09:09:36 -04:00
|
|
|
callbackOnce(new Error("URL returned non-success status code: #{res.statusCode} #{url}"))
|
2014-02-12 12:27:43 -05:00
|
|
|
|
|
|
|
urlStream.on "error", (error) ->
|
2015-04-29 10:55:58 -04:00
|
|
|
logger.error err: error, url:url, filePath: filePath, "error downloading url"
|
2014-06-10 09:09:36 -04:00
|
|
|
callbackOnce(error or new Error("Something went wrong downloading the URL #{url}"))
|
2014-02-12 12:27:43 -05:00
|
|
|
|
|
|
|
urlStream.on "end", () ->
|
2015-04-29 10:55:58 -04:00
|
|
|
# FIXME: what if we get an error writing the file? Maybe we
|
|
|
|
# should be using the fileStream end event as the point of
|
|
|
|
# callback.
|
2014-02-12 12:27:43 -05:00
|
|
|
callbackOnce()
|