diff --git a/wget.js b/wget.js index fade3e4..1830683 100644 --- a/wget.js +++ b/wget.js @@ -9,9 +9,19 @@ function wget(uri, callback) { var options = url.parse(uri); var paths = options.pathname.split('/'); var filename = paths[paths.length - 1]; + var proxy; + var proto = options.protocol.slice(0,-1); + if(process.env[proto+'_proxy']) { + proxy = url.parse(process.env[proto+'_proxy']); + } + if(proxy) { + proxy.path = proxy.pathname = options.protocol+'//'+options.host+options.pathname; + options = proxy; + proto = options.protocol.slice(0,-1); + } console.log(filename); - var http = require(uri.indexOf('https') === 0 ? 'https' : 'http'); + var http = require(proto); var req = http.get(options, function (res) { if (res.statusCode === 302 || res.statusCode === 301) {