Merge commit 'be3e8236086165e5e45a5a10783823874b3f3ebd' as 'lib/vscode'

lib/vscode/extensions/emmet/src/imageSizeHelper.ts

@@ -0,0 +1,113 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+// Based on @sergeche's work on the emmet plugin for atom
+// TODO: Move to https://github.com/emmetio/image-size
+
+import * as path from 'path';
+import * as http from 'http';
+import * as https from 'https';
+import { parse as parseUrl } from 'url';
+import * as sizeOf from 'image-size';
+
+const reUrl = /^https?:/;
+
+/**
+ * Get size of given image file. Supports files from local filesystem,
+ * as well as URLs
+ */
+export function getImageSize(file: string) {
+	file = file.replace(/^file:\/\//, '');
+	return reUrl.test(file) ? getImageSizeFromURL(file) : getImageSizeFromFile(file);
+}
+
+/**
+ * Get image size from file on local file system
+ */
+function getImageSizeFromFile(file: string) {
+	return new Promise((resolve, reject) => {
+		const isDataUrl = file.match(/^data:.+?;base64,/);
+
+		if (isDataUrl) {
+			// NB should use sync version of `sizeOf()` for buffers
+			try {
+				const data = Buffer.from(file.slice(isDataUrl[0].length), 'base64');
+				return resolve(sizeForFileName('', sizeOf(data)));
+			} catch (err) {
+				return reject(err);
+			}
+		}
+
+		sizeOf(file, (err: any, size: any) => {
+			if (err) {
+				reject(err);
+			} else {
+				resolve(sizeForFileName(path.basename(file), size));
+			}
+		});
+	});
+}
+
+/**
+ * Get image size from given remove URL
+ */
+function getImageSizeFromURL(urlStr: string) {
+	return new Promise((resolve, reject) => {
+		const url = parseUrl(urlStr);
+		const getTransport = url.protocol === 'https:' ? https.get : http.get;
+
+		if (!url.pathname) {
+			return reject('Given url doesnt have pathname property');
+		}
+		const urlPath: string = url.pathname;
+
+		getTransport(url as any, resp => {
+			const chunks: Buffer[] = [];
+			let bufSize = 0;
+
+			const trySize = (chunks: Buffer[]) => {
+				try {
+					const size = sizeOf(Buffer.concat(chunks, bufSize));
+					resp.removeListener('data', onData);
+					resp.destroy(); // no need to read further
+					resolve(sizeForFileName(path.basename(urlPath), size));
+				} catch (err) {
+					// might not have enough data, skip error
+				}
+			};
+
+			const onData = (chunk: Buffer) => {
+				bufSize += chunk.length;
+				chunks.push(chunk);
+				trySize(chunks);
+			};
+
+			resp
+				.on('data', onData)
+				.on('end', () => trySize(chunks))
+				.once('error', err => {
+					resp.removeListener('data', onData);
+					reject(err);
+				});
+		})
+			.once('error', reject);
+	});
+}
+
+/**
+ * Returns size object for given file name. If file name contains `@Nx` token,
+ * the final dimentions will be downscaled by N
+ */
+function sizeForFileName(fileName: string, size: any) {
+	const m = fileName.match(/@(\d+)x\./);
+	const scale = m ? +m[1] : 1;
+
+	return {
+		realWidth: size.width,
+		realHeight: size.height,
+		width: Math.floor(size.width / scale),
+		height: Math.floor(size.height / scale)
+	};
+}