57 lines
2.0 KiB
TypeScript
57 lines
2.0 KiB
TypeScript
/**
|
|
* Copyright 2020 Google Inc. All rights reserved.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
import { PuppeteerNode } from './node/Puppeteer.js';
|
|
import { PUPPETEER_REVISIONS } from './revisions.js';
|
|
import { sync } from 'pkg-dir';
|
|
import { dirname } from 'path';
|
|
import { Product } from './common/Product.js';
|
|
|
|
function resolvePuppeteerRootDirectory(): string | undefined {
|
|
try {
|
|
// In some environments, like esbuild, this will throw an error.
|
|
// We suppress the error since the bundled binary is not expected
|
|
// to be used or installed in this case and, therefore, the
|
|
// root directory does not have to be known.
|
|
return sync(dirname(require.resolve('./initialize-node')));
|
|
} catch (error) {
|
|
// Fallback to __dirname.
|
|
return sync(__dirname);
|
|
}
|
|
}
|
|
|
|
export const initializePuppeteerNode = (packageName: string): PuppeteerNode => {
|
|
const puppeteerRootDirectory = resolvePuppeteerRootDirectory();
|
|
let preferredRevision = PUPPETEER_REVISIONS.chromium;
|
|
const isPuppeteerCore = packageName === 'puppeteer-core';
|
|
// puppeteer-core ignores environment variables
|
|
const productName = isPuppeteerCore
|
|
? undefined
|
|
: process.env.PUPPETEER_PRODUCT ||
|
|
process.env.npm_config_puppeteer_product ||
|
|
process.env.npm_package_config_puppeteer_product;
|
|
|
|
if (!isPuppeteerCore && productName === 'firefox')
|
|
preferredRevision = PUPPETEER_REVISIONS.firefox;
|
|
|
|
return new PuppeteerNode({
|
|
projectRoot: puppeteerRootDirectory,
|
|
preferredRevision,
|
|
isPuppeteerCore,
|
|
productName: productName as Product,
|
|
});
|
|
};
|