114 lines
3.0 KiB
TypeScript
114 lines
3.0 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* 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)
|
|
};
|
|
}
|