Merge branch 'master' into heartbeat

1 files changed, 73 insertions, 0 deletions

diff --git a/src/client/util/request-image-size.ts b/src/client/util/request-image-size.ts
new file mode 100644
index 000000000..57e8516ac
--- /dev/null
+++ b/src/client/util/request-image-size.ts
@@ -0,0 +1,73 @@
+/**
+ * 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: any) {
+    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: any) => {
+            if (res.statusCode >= 400) {
+                return reject(new HttpError(res.statusCode, res.statusMessage));
+            }
+
+            let buffer = Buffer.from([]);
+            let size: any;
+
+            res.on('data', (chunk: any) => {
+                buffer = Buffer.concat([buffer, chunk]);
+
+                try {
+                    size = imageSize(buffer);
+                    if (size) {
+                        resolve(size);
+                        return req.abort();
+                    }
+                } catch (err) {}
+            });
+
+            res.on('error', reject);
+
+            res.on('end', () => {
+                if (!size) {
+                    return reject(new Error('Image has no size'));
+                }
+
+                size.downloaded = buffer.length;
+                return resolve(size);
+            });
+        });
+
+        req.on('error', reject);
+    });
+};