156 lines
4.7 KiB
JavaScript
156 lines
4.7 KiB
JavaScript
/**
|
|
* @license
|
|
* Copyright 2023 Google Inc.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
import { spawnSync, spawn } from 'node:child_process';
|
|
import { createReadStream } from 'node:fs';
|
|
import { mkdir, readdir } from 'node:fs/promises';
|
|
import * as path from 'node:path';
|
|
import { Stream } from 'node:stream';
|
|
import debug from 'debug';
|
|
const debugFileUtil = debug('puppeteer:browsers:fileUtil');
|
|
/**
|
|
* @internal
|
|
*/
|
|
export async function unpackArchive(archivePath, folderPath) {
|
|
if (!path.isAbsolute(folderPath)) {
|
|
folderPath = path.resolve(process.cwd(), folderPath);
|
|
}
|
|
if (archivePath.endsWith('.zip')) {
|
|
const extractZip = await import('extract-zip');
|
|
await extractZip.default(archivePath, { dir: folderPath });
|
|
}
|
|
else if (archivePath.endsWith('.tar.bz2')) {
|
|
await extractTar(archivePath, folderPath, 'bzip2');
|
|
}
|
|
else if (archivePath.endsWith('.dmg')) {
|
|
await mkdir(folderPath);
|
|
await installDMG(archivePath, folderPath);
|
|
}
|
|
else if (archivePath.endsWith('.exe')) {
|
|
// Firefox on Windows.
|
|
const result = spawnSync(archivePath, [`/ExtractDir=${folderPath}`], {
|
|
env: {
|
|
__compat_layer: 'RunAsInvoker',
|
|
},
|
|
});
|
|
if (result.status !== 0) {
|
|
throw new Error(`Failed to extract ${archivePath} to ${folderPath}: ${result.output}`);
|
|
}
|
|
}
|
|
else if (archivePath.endsWith('.tar.xz')) {
|
|
await extractTar(archivePath, folderPath, 'xz');
|
|
}
|
|
else {
|
|
throw new Error(`Unsupported archive format: ${archivePath}`);
|
|
}
|
|
}
|
|
function createTransformStream(child) {
|
|
const stream = new Stream.Transform({
|
|
transform(chunk, encoding, callback) {
|
|
if (!child.stdin.write(chunk, encoding)) {
|
|
child.stdin.once('drain', callback);
|
|
}
|
|
else {
|
|
callback();
|
|
}
|
|
},
|
|
flush(callback) {
|
|
if (child.stdout.destroyed) {
|
|
callback();
|
|
}
|
|
else {
|
|
child.stdin.end();
|
|
child.stdout.on('close', callback);
|
|
}
|
|
},
|
|
});
|
|
child.stdin.on('error', e => {
|
|
if ('code' in e && e.code === 'EPIPE') {
|
|
// finished before reading the file finished (i.e. head)
|
|
stream.emit('end');
|
|
}
|
|
else {
|
|
stream.destroy(e);
|
|
}
|
|
});
|
|
child.stdout
|
|
.on('data', data => {
|
|
return stream.push(data);
|
|
})
|
|
.on('error', e => {
|
|
return stream.destroy(e);
|
|
});
|
|
child.once('close', () => {
|
|
return stream.end();
|
|
});
|
|
return stream;
|
|
}
|
|
/**
|
|
* @internal
|
|
*/
|
|
export const internalConstantsForTesting = {
|
|
xz: 'xz',
|
|
bzip2: 'bzip2',
|
|
};
|
|
/**
|
|
* @internal
|
|
*/
|
|
async function extractTar(tarPath, folderPath, decompressUtilityName) {
|
|
const tarFs = await import('tar-fs');
|
|
return await new Promise((fulfill, reject) => {
|
|
function handleError(utilityName) {
|
|
return (error) => {
|
|
if ('code' in error && error.code === 'ENOENT') {
|
|
error = new Error(`\`${utilityName}\` utility is required to unpack this archive`, {
|
|
cause: error,
|
|
});
|
|
}
|
|
reject(error);
|
|
};
|
|
}
|
|
const unpack = spawn(internalConstantsForTesting[decompressUtilityName], ['-d'], {
|
|
stdio: ['pipe', 'pipe', 'inherit'],
|
|
})
|
|
.once('error', handleError(decompressUtilityName))
|
|
.once('exit', code => {
|
|
debugFileUtil(`${decompressUtilityName} exited, code=${code}`);
|
|
});
|
|
const tar = tarFs.extract(folderPath);
|
|
tar.once('error', handleError('tar'));
|
|
tar.once('finish', fulfill);
|
|
createReadStream(tarPath).pipe(createTransformStream(unpack)).pipe(tar);
|
|
});
|
|
}
|
|
/**
|
|
* @internal
|
|
*/
|
|
async function installDMG(dmgPath, folderPath) {
|
|
const { stdout } = spawnSync(`hdiutil`, [
|
|
'attach',
|
|
'-nobrowse',
|
|
'-noautoopen',
|
|
dmgPath,
|
|
]);
|
|
const volumes = stdout.toString('utf8').match(/\/Volumes\/(.*)/m);
|
|
if (!volumes) {
|
|
throw new Error(`Could not find volume path in ${stdout}`);
|
|
}
|
|
const mountPath = volumes[0];
|
|
try {
|
|
const fileNames = await readdir(mountPath);
|
|
const appName = fileNames.find(item => {
|
|
return typeof item === 'string' && item.endsWith('.app');
|
|
});
|
|
if (!appName) {
|
|
throw new Error(`Cannot find app in ${mountPath}`);
|
|
}
|
|
const mountedPath = path.join(mountPath, appName);
|
|
spawnSync('cp', ['-R', mountedPath, folderPath]);
|
|
}
|
|
finally {
|
|
spawnSync('hdiutil', ['detach', mountPath, '-quiet']);
|
|
}
|
|
}
|
|
//# sourceMappingURL=fileUtil.js.map
|