z0rs
/
chromium
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor executablePath a little bit 

Thanks @aquarius6666 for the initial code!
This commit is contained in:
Sparticuz 2022-12-28 10:27:37 -05:00
parent 4cda3ab393
commit 0562aca09f
1 changed files with 28 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
source/index.ts
View File

@ -1,4 +1,4 @@
import { access, createWriteStream, existsSync, mkdirSync, readdirSync, symlink, unlinkSync } from 'fs'; import { access, createWriteStream, existsSync, mkdirSync, symlink } from 'fs';
import { IncomingMessage } from 'http'; import { IncomingMessage } from 'http';
import LambdaFS from './lambdafs'; import LambdaFS from './lambdafs';
import { join } from 'path'; import { join } from 'path';
@ -176,38 +176,44 @@ class Chromium {
} }
/** /**
* Inflates the current version of Chromium and returns the path to the binary. * Inflates the included version of Chromium
* If not running on AWS Lambda nor Google Cloud Functions, `null` is returned instead. * @param input The location of the `bin` folder
* @returns The path to the `chromium` binary
*/ */
static executablePath(input?: string): Promise<string> { static async executablePath(input?: string): Promise<string> {
if (existsSync('/tmp/chromium') === true && input === undefined) {
for (const file of readdirSync('/tmp')) {
if (file.startsWith('core.chromium') === true) {
unlinkSync(`/tmp/${file}`);
}
}
/**
* If the `chromium` binary already exists in /tmp/chromium, return it.
*/
if (existsSync('/tmp/chromium') === true && input === undefined) {
return Promise.resolve('/tmp/chromium'); return Promise.resolve('/tmp/chromium');
} }
let _input = join(__dirname, '..', 'bin');
if (input !== undefined) { /**
if (existsSync(input)) { * If input is defined, use that as the location of the brotli files,
_input = join(input, 'bin'); * otherwise, the default location is ../bin.
} else { * A custom location is needed for workflows that using custom packaging.
throw new Error(`The input directory "${input}" does not exist.`); */
} input ??= join(__dirname, '..', 'bin');
/**
* If the input directory doesn't exist, throw an error.
*/
if (!existsSync(input)) {
throw new Error(`The input directory "${input}" does not exist.`);
} }
const promises = [ const promises = [
LambdaFS.inflate(`${_input}/chromium.br`), LambdaFS.inflate(`${input}/chromium.br`),
LambdaFS.inflate(`${_input}/swiftshader.tar.br`), LambdaFS.inflate(`${input}/swiftshader.tar.br`),
]; ];
if (/^AWS_Lambda_nodejs(?:10|12|14|16|18)[.]x$/.test(process.env.AWS_EXECUTION_ENV) === true) { if (/^AWS_Lambda_nodejs(?:10|12|14|16|18)[.]x$/.test(process.env.AWS_EXECUTION_ENV) === true) {
promises.push(LambdaFS.inflate(`${_input}/aws.tar.br`)); promises.push(LambdaFS.inflate(`${input}/aws.tar.br`));
} }
return Promise.all(promises).then((result) => result.shift()); const result = await Promise.all(promises);
return result.shift();
} }
/** /**