mirror of
https://github.com/balena-io/etcher.git
synced 2025-04-23 14:57:18 +00:00
fc46958ac0
This is a regression caused by https://github.com/resin-io/etcher/pull/1257, which introduced a new way to detect mime types by using the `mime-types` module. This module, contrary to `file-type`, will detect certain ISO files as `application/x-iso9660-image`, which Etcher doesn't know how to handle, and will therefore should at the user that the format is unsupported. Change-Type: patch Changelog-Entry: Don't interpret certain ISO images as unsupported. Signed-off-by: Juan Cruz Viotti <jviotti@openmailbox.org>
100 lines
2.6 KiB
JavaScript
100 lines
2.6 KiB
JavaScript
/*
|
|
* Copyright 2016 resin.io
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
const _ = require('lodash');
|
|
const Bluebird = require('bluebird');
|
|
const fileType = require('file-type');
|
|
const mime = require('mime-types');
|
|
const fs = require('fs');
|
|
|
|
/**
|
|
* @summary Get archive mime type
|
|
* @function
|
|
* @public
|
|
*
|
|
* @param {String} filename - file path
|
|
* @fulfil {String} - mime type
|
|
* @returns {Promise}
|
|
*
|
|
* @example
|
|
* utils.getArchiveMimeType('path/to/raspberrypi.img.gz').then((mimeType) => {
|
|
* console.log(mimeType);
|
|
* });
|
|
*/
|
|
exports.getArchiveMimeType = (filename) => {
|
|
const MIME_TYPE_RAW_IMAGE = 'application/octet-stream';
|
|
const mimeType = mime.lookup(filename);
|
|
|
|
if (mimeType) {
|
|
if (mimeType === 'application/x-iso9660-image') {
|
|
return Bluebird.resolve(MIME_TYPE_RAW_IMAGE);
|
|
}
|
|
|
|
return Bluebird.resolve(mimeType);
|
|
}
|
|
|
|
const FILE_TYPE_ID_BYTES = 262;
|
|
|
|
return Bluebird.using(fs.openAsync(filename, 'r').disposer((fileDescriptor) => {
|
|
return fs.closeAsync(fileDescriptor);
|
|
}), (fileDescriptor) => {
|
|
const BUFFER_START = 0;
|
|
const buffer = Buffer.alloc(FILE_TYPE_ID_BYTES);
|
|
|
|
return fs.readAsync(fileDescriptor, buffer, BUFFER_START, FILE_TYPE_ID_BYTES, null).then(() => {
|
|
return _.get(fileType(buffer), [ 'mime' ], MIME_TYPE_RAW_IMAGE);
|
|
});
|
|
});
|
|
|
|
};
|
|
|
|
/**
|
|
* @summary Extract the data of a readable stream
|
|
* @function
|
|
* @public
|
|
*
|
|
* @description
|
|
* You should be careful when using this function, since you can only
|
|
* extract files that are not bigger than the available computer memory.
|
|
*
|
|
* @param {StreamReadable} stream - stream
|
|
* @fulfil {Buffer} - data
|
|
* @returns {Promise}
|
|
*
|
|
* @example
|
|
* const stream = fs.createReadStream('./foo/bar');
|
|
*
|
|
* utils.extractStream(stream).then((data) => {
|
|
* console.log(data.toString());
|
|
* });
|
|
*/
|
|
exports.extractStream = (stream) => {
|
|
return new Bluebird((resolve, reject) => {
|
|
const chunks = [];
|
|
|
|
stream.on('data', (chunk) => {
|
|
chunks.push(chunk);
|
|
});
|
|
|
|
stream.on('error', reject);
|
|
stream.on('end', () => {
|
|
resolve(Buffer.concat(chunks));
|
|
});
|
|
});
|
|
};
|