blob: 27605d1679661c50edb3a6599ee71fc473307013 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/**
* request-image-size: Detect image dimensions via request.
* Licensed under the MIT license.
*
* https://github.com/FdezRomero/request-image-size
* © 2017 Rodrigo Fernández Romero
*
* Based on the work of Johannes J. Schmidt
* https://github.com/jo/http-image-size
*/
const request = require('request');
const imageSize = require('image-size');
const HttpError = require('standard-http-error');
module.exports = function requestImageSize(options) {
let opts = {
encoding: null
};
if (options && typeof options === 'object') {
opts = Object.assign(options, opts);
} else if (options && typeof options === 'string') {
opts = Object.assign({
uri: options
}, opts);
} else {
return Promise.reject(new Error('You should provide an URI string or a "request" options object.'));
}
opts.encoding = null;
return new Promise((resolve, reject) => {
const req = request(opts);
req.on('response', res => {
if (res.statusCode >= 400) {
return reject(new HttpError(res.statusCode, res.statusMessage));
}
let buffer = new Buffer([]);
let size;
let imageSizeError;
res.on('data', chunk => {
buffer = Buffer.concat([buffer, chunk]);
try {
size = imageSize(buffer);
} catch (err) {
imageSizeError = err;
return;
}
if (size) {
resolve(size);
return req.abort();
}
});
res.on('error', err => reject(err));
res.on('end', () => {
if (!size) {
return reject(imageSizeError);
}
size.downloaded = buffer.length;
return resolve(size);
});
});
req.on('error', err => reject(err));
});
};
|
